128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
from apps.utils.constants import EXCLUDE_FIELDS
|
|
from apps.utils.serializers import CustomModelSerializer
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from .models import SfLog, StLog, SfLogExp, WMaterial, Mlog
|
|
from apps.system.models import Dictionary
|
|
from apps.pm.models import Mtask
|
|
from apps.wpm.tasks import cal_enstat_when_pcoal_heat_change, cal_enstat_when_team_change
|
|
from apps.mtm.serializers import MaterialSerializer
|
|
|
|
|
|
class StLogSerializer(CustomModelSerializer):
|
|
mgroup_name = serializers.CharField(source='mgroup.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = StLog
|
|
fields = '__all__'
|
|
|
|
|
|
class SfLogSerializer(CustomModelSerializer):
|
|
mgroup_name = serializers.CharField(source='mgroup.name', read_only=True)
|
|
team_name = serializers.CharField(source='team.name', read_only=True)
|
|
shift_name = serializers.CharField(source='shift.name', read_only=True)
|
|
leader_name = serializers.CharField(source='leader.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = SfLog
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + \
|
|
['mgroup', 'start_time', 'end_time', 'belong_dept']
|
|
extra_kwargs = {
|
|
'team': {'required': True},
|
|
'leader': {'required': True}
|
|
}
|
|
|
|
def update(self, instance, validated_data):
|
|
old_pcoal_heat = instance.pcoal_heat
|
|
old_team = instance.team
|
|
instance = super().update(instance, validated_data)
|
|
new_pcoal_heat = instance.pcoal_heat
|
|
new_team = instance.team
|
|
if new_pcoal_heat != old_pcoal_heat:
|
|
cal_enstat_when_pcoal_heat_change.delay(instance.id)
|
|
if new_team != old_team:
|
|
cal_enstat_when_team_change.delay(instance.id)
|
|
return instance
|
|
# def to_internal_value(self, data):
|
|
# if hasattr(self.Meta, 'update_fields') and self.context['request'].method in ['PUT', 'PATCH']:
|
|
# u_fields = self.Meta.update_fields
|
|
# new_data = {key: data[key] for key in u_fields if key in data}
|
|
# return super().to_internal_value(new_data)
|
|
# return super().to_internal_value(data)
|
|
|
|
|
|
class SflogExpSerializer(CustomModelSerializer):
|
|
mgroup = serializers.CharField(source='sflog.mgroup.id', read_only=True)
|
|
mgroup_name = serializers.CharField(
|
|
source='sflog.mgroup.name', read_only=True)
|
|
stlog_ = StLogSerializer(source='stlog', read_only=True)
|
|
happen_time = serializers.DateTimeField(required=True, label='发生时间')
|
|
cate = serializers.CharField(required=True, label='原因类别')
|
|
|
|
class Meta:
|
|
model = SfLogExp
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + ['stlog', 'is_current_down']
|
|
|
|
|
|
class WMaterialSerializer(CustomModelSerializer):
|
|
material_ = MaterialSerializer(source='material', read_only=True)
|
|
belong_dept_name = serializers.CharField(
|
|
source='belong_dept.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = WMaterial
|
|
fields = '__all__'
|
|
|
|
|
|
class MlogSerializer(CustomModelSerializer):
|
|
mgroup_name = serializers.CharField(
|
|
source='mtask.mgroup.name', read_only=True)
|
|
mtask_count = serializers.IntegerField(
|
|
source='mtask.count', read_only=True)
|
|
create_by_name = serializers.CharField(
|
|
source='create_by.name', read_only=True)
|
|
update_by_name = serializers.CharField(
|
|
source='update_by.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Mlog
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
def create(self, validated_data):
|
|
mtask = validated_data['mtask']
|
|
batch = validated_data['batch']
|
|
handle_date = validated_data['handle_date']
|
|
handle_user = validated_data['handle_user']
|
|
if not WMaterial.objects.filter(batch=batch).exists():
|
|
raise ValidationError('批次号不存在')
|
|
if Mlog.objects.filter(mtask=mtask, batch=batch, handle_date=handle_date, handle_user=handle_user).exists():
|
|
raise ValidationError('存在相同的日志的记录')
|
|
return super().create(validated_data)
|
|
|
|
def update(self, instance, validated_data):
|
|
validated_data.pop('mtask', None)
|
|
validated_data.pop('batch', None)
|
|
validated_data.pop('handle_date', None)
|
|
validated_data.pop('handle_user', None)
|
|
return super().update(instance, validated_data)
|
|
|
|
def validate(self, attrs):
|
|
mtask = attrs['mtask']
|
|
handle_date = attrs['handle_date']
|
|
if handle_date >= mtask.start_date and handle_date <= mtask.end_date:
|
|
pass
|
|
else:
|
|
raise ValidationError('操作日期错误')
|
|
if mtask.state != Mtask.MTASK_ASSGINED:
|
|
raise ValidationError(f'{mtask.mgroup.name}_不可提交日志!')
|
|
return attrs
|
|
|
|
|
|
class MlogRelatedSerializer(serializers.Serializer):
|
|
mtask = serializers.PrimaryKeyRelatedField(
|
|
label='小任务ID', queryset=Mtask.objects.all())
|