138 lines
5.5 KiB
Python
138 lines
5.5 KiB
Python
from apps.utils.serializers import CustomModelSerializer
|
|
from apps.mtm.models import Shift, Material, Mgroup, Team, Goal, Process, Route
|
|
from apps.utils.constants import EXCLUDE_FIELDS
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import ValidationError
|
|
from apps.system.models import Dept
|
|
from django.db import transaction
|
|
|
|
|
|
class ShiftSerializer(CustomModelSerializer):
|
|
class Meta:
|
|
model = Shift
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class MaterialSerializer(CustomModelSerializer):
|
|
process_name = serializers.CharField(source='process.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Material
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class MgroupSerializer(CustomModelSerializer):
|
|
belong_dept = serializers.PrimaryKeyRelatedField(
|
|
label="所属部门", queryset=Dept.objects.all(), required=True)
|
|
belong_dept_name = serializers.CharField(
|
|
source='belong_dept.name', read_only=True)
|
|
process_name = serializers.CharField(source='process.name', read_only=True)
|
|
process_cate = serializers.CharField(source='process.cate', read_only=True)
|
|
|
|
class Meta:
|
|
model = Mgroup
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class TeamSerializer(CustomModelSerializer):
|
|
belong_dept = serializers.PrimaryKeyRelatedField(
|
|
label="所属部门", queryset=Dept.objects.all(), required=True)
|
|
leader_name = serializers.CharField(source='leader.name', read_only=True)
|
|
belong_dept_name = serializers.CharField(
|
|
source='belong_dept.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Team
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class GoalSerializer(CustomModelSerializer):
|
|
goal_cate_name = serializers.CharField(
|
|
source='goal_cate.name', read_only=True)
|
|
mgroup_name = serializers.CharField(source='mgroup.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Goal
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class MgroupGoalYearSerializer(serializers.Serializer):
|
|
mgroup = serializers.CharField(label='ID或名称')
|
|
year = serializers.IntegerField(label='年')
|
|
|
|
|
|
class ProcessSerializer(CustomModelSerializer):
|
|
belong_dept_name = serializers.CharField(
|
|
source='belong_dept.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Process
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class RouteSerializer(CustomModelSerializer):
|
|
material_ = MaterialSerializer(source='material', read_only=True)
|
|
process_name = serializers.CharField(source='process.name', read_only=True)
|
|
process_cate = serializers.CharField(source='process.cate', read_only=True)
|
|
material_in_name = serializers.CharField(
|
|
source='material_in.name', read_only=True)
|
|
material_out_name = serializers.CharField(
|
|
source='material_out.name', read_only=True)
|
|
material_out_type = serializers.CharField(
|
|
source='material_out.type', read_only=True)
|
|
material_out_is_hidden = serializers.BooleanField(
|
|
source='material_out.is_hidden', read_only=True)
|
|
|
|
class Meta:
|
|
model = Route
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
# def validate(self, attrs):
|
|
# material = attrs['material']
|
|
# if material.type != Material.MA_TYPE_GOOD:
|
|
# raise ValidationError('请选择最终产品')
|
|
# return super().validate(attrs)
|
|
|
|
def gen_material_out(self, instance):
|
|
name = f'{instance.material.name}-中'
|
|
instance.material_out, _ = Material.objects.get_or_create(type=Material.MA_TYPE_HALFGOOD, parent=instance.material, process=instance.process,
|
|
defaults={'parent': instance.material, 'process': instance.process,
|
|
'is_hidden': True, 'name': name,
|
|
'number': instance.material.number,
|
|
'specification': instance.material.specification,
|
|
'type': Material.MA_TYPE_HALFGOOD,
|
|
'create_by': self.request.user,
|
|
'update_by': self.request.user,
|
|
})
|
|
instance.save()
|
|
|
|
def create(self, validated_data):
|
|
process = validated_data['process']
|
|
if Route.objects.filter(material=validated_data['material'], process=process).exists():
|
|
raise ValidationError('已选择该工序')
|
|
with transaction.atomic():
|
|
instance = super().create(validated_data)
|
|
if 'material_out' in validated_data and validated_data['material_out']:
|
|
pass
|
|
else:
|
|
self.gen_material_out(instance)
|
|
return instance
|
|
|
|
def update(self, instance, validated_data):
|
|
validated_data.pop('material', None)
|
|
validated_data.pop('process', None)
|
|
with transaction.atomic():
|
|
instance = super().update(instance, validated_data)
|
|
if 'material_out' in validated_data and validated_data['material_out']:
|
|
pass
|
|
else:
|
|
self.gen_material_out(instance)
|
|
return instance
|