from apps.em.serializers import EquipmentSerializer, EquipmentSimpleSerializer
from rest_framework import serializers
from rest_framework.exceptions import ValidationError
from .models import InputMaterial, Material, OutputMaterial, Process, ProductProcess, Step, UsedStep
from apps.system.serializers import FileSimpleSerializer


class MaterialSerializer(serializers.ModelSerializer):
    processes = serializers.ListField(child=serializers.IntegerField(min_value=1))
    class Meta:
        model = Material
        fields = '__all__'

class MaterialDetailSerializer(serializers.ModelSerializer):
    processes_ = serializers.SerializerMethodField()
    class Meta:
        model = Material
        fields = '__all__'
    
    def get_processes_(self, obj):
        objs = Process.objects.filter(id__in=obj.processes)
        return ProcessSimpleSerializer(instance=objs, many=True).data


class MaterialSimpleSerializer(serializers.ModelSerializer):
    class Meta:
        model = Material
        fields = ['id', 'name', 'number', 'unit']

class ProcessSerializer(serializers.ModelSerializer):
    instruction_ = FileSimpleSerializer(source='instruction', read_only=True)
    class Meta:
        model = Process
        fields = '__all__'

class ProcessSimpleSerializer(serializers.ModelSerializer):
    class Meta:
        model = Process
        fields = ['id', 'name', 'number']

class StepSerializer(serializers.ModelSerializer):
    class Meta:
        model = Step
        fields = '__all__'

class StepSimpleSerializer(serializers.ModelSerializer):
    class Meta:
        model = Step
        fields = ['id', 'name', 'sort']

class StepDetailSerializer(serializers.ModelSerializer):
    equipments_ = EquipmentSimpleSerializer(source='equipments_', many=True)
    class Meta:
        model = Step
        fields = '__all__'
    
    @staticmethod
    def setup_eager_loading(queryset):
        queryset = queryset.prefetch_related('equipments')
        return queryset

class ProductProcessListSerializer(serializers.ModelSerializer):
    process_ = ProcessSimpleSerializer(source='process', read_only=True)
    product_ = MaterialSimpleSerializer(source='product', read_only=True)
    class Meta:
        model = ProductProcess
        fields = '__all__'

class ProductProcessUpdateSerializer(serializers.ModelSerializer):
    class Meta:
        model = ProductProcess
        fields = ['sort']

class InputMaterialListSerializer(serializers.ModelSerializer):
    material_ = MaterialSimpleSerializer(source='material', read_only=True)
    class Meta:
        model = InputMaterial
        fields = '__all__'

class OutputMaterialListSerializer(serializers.ModelSerializer):
    material_ = MaterialSimpleSerializer(source='material', read_only=True)
    class Meta:
        model = OutputMaterial
        fields = '__all__'


class InputMaterialSerializer(serializers.ModelSerializer):
    class Meta:
        model = InputMaterial
        fields = ['count', 'sort', 'material', 'product', 'process']
    
    def create(self, validated_data):
        if InputMaterial.objects.filter(material=validated_data['material'], product=validated_data['product'], process=validated_data['process'], is_deleted=False).exists():
            raise ValidationError('该物料已存在')
        return super().create(validated_data)

class InputMaterialUpdateSerializer(serializers.ModelSerializer):
    class Meta:
        model = InputMaterial
        fields = ['count', 'sort']

class OutputMaterialSerializer(serializers.ModelSerializer):
    class Meta:
        model = OutputMaterial
        fields = ['count', 'sort', 'material', 'product', 'process']
    
    def create(self, validated_data):
        if OutputMaterial.objects.filter(material=validated_data['material'], product=validated_data['product'], process=validated_data['process'], is_deleted=False).exists():
            raise ValidationError('该物料已存在')
        return super().create(validated_data)

class OutputMaterialUpdateSerializer(serializers.ModelSerializer):
    class Meta:
        model = OutputMaterial
        fields = ['count', 'sort']

class UsedStepCreateSerializer(serializers.ModelSerializer):
    """
    产品生产子工序创建
    """
    class Meta:
        model = UsedStep
        fields =  ['step', 'product', 'process']

class UsedStepListSerializer(serializers.ModelSerializer):
    """
    产品生产子工序序列化
    """
    step_ = StepSimpleSerializer(source='step', read_only=True)
    class Meta:
        model = UsedStep
        fields = '__all__'
    
    @staticmethod
    def setup_eager_loading(queryset):
        """ Perform necessary eager loading of data. """
        queryset = queryset.select_related('step')
        return queryset