139 lines
4.7 KiB
Python
139 lines
4.7 KiB
Python
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
|