style: 格式调整
This commit is contained in:
parent
640d1a1bce
commit
e78511b556
|
@ -9,63 +9,57 @@ from rest_framework.exceptions import ValidationError
|
|||
class EcateSerializer(CustomModelSerializer):
|
||||
class Meta:
|
||||
model = Ecate
|
||||
fields = '__all__'
|
||||
fields = "__all__"
|
||||
read_only_fields = EXCLUDE_FIELDS_BASE
|
||||
|
||||
|
||||
class EquipmentSerializer(CustomModelSerializer):
|
||||
belong_dept = serializers.PrimaryKeyRelatedField(
|
||||
label='责任部门', queryset=Dept.objects.all())
|
||||
keeper_name = serializers.CharField(source='keeper.name', read_only=True)
|
||||
belong_dept_name = serializers.CharField(
|
||||
source='belong_dept.name', read_only=True)
|
||||
mgroup_name = serializers.CharField(source='mgroup.name', read_only=True)
|
||||
cate_name = serializers.CharField(source='cate.name', read_only=True)
|
||||
cate_code = serializers.CharField(source='cate.code', read_only=True)
|
||||
belong_dept = serializers.PrimaryKeyRelatedField(label="责任部门", queryset=Dept.objects.all())
|
||||
keeper_name = serializers.CharField(source="keeper.name", read_only=True)
|
||||
belong_dept_name = serializers.CharField(source="belong_dept.name", read_only=True)
|
||||
mgroup_name = serializers.CharField(source="mgroup.name", read_only=True)
|
||||
cate_name = serializers.CharField(source="cate.name", read_only=True)
|
||||
cate_code = serializers.CharField(source="cate.code", read_only=True)
|
||||
full_name = serializers.SerializerMethodField()
|
||||
|
||||
def validate(self, attrs):
|
||||
mgroup = attrs.get('mgroup', None)
|
||||
mgroup = attrs.get("mgroup", None)
|
||||
if mgroup:
|
||||
attrs['belong_dept'] = mgroup.belong_dept
|
||||
cate = attrs.get('cate', None)
|
||||
attrs["belong_dept"] = mgroup.belong_dept
|
||||
cate = attrs.get("cate", None)
|
||||
if cate:
|
||||
attrs['type'] = cate.type
|
||||
attrs["type"] = cate.type
|
||||
return super().validate(attrs)
|
||||
|
||||
def get_full_name(self, obj):
|
||||
return f'{obj.number}|{obj.name}|{obj.model}'
|
||||
return f"{obj.number}|{obj.name}|{obj.model}"
|
||||
|
||||
class Meta:
|
||||
model = Equipment
|
||||
fields = '__all__'
|
||||
read_only_fields = EXCLUDE_FIELDS + \
|
||||
['check_date', 'next_check_date', 'keeper_name', 'belong_dept_name']
|
||||
fields = "__all__"
|
||||
read_only_fields = EXCLUDE_FIELDS + ["check_date", "next_check_date", "keeper_name", "belong_dept_name", "running_state"]
|
||||
|
||||
|
||||
class EcheckRecordSerializer(CustomModelSerializer):
|
||||
equipment_name = serializers.StringRelatedField(
|
||||
source='equipment', read_only=True)
|
||||
equipment_name = serializers.StringRelatedField(source="equipment", read_only=True)
|
||||
|
||||
class Meta:
|
||||
model = EcheckRecord
|
||||
fields = '__all__'
|
||||
fields = "__all__"
|
||||
read_only_fields = EXCLUDE_FIELDS
|
||||
|
||||
def validate(self, attrs):
|
||||
check_date = attrs['check_date']
|
||||
if EcheckRecord.objects.filter(check_date__gte=check_date, equipment=attrs['equipment']).exists():
|
||||
raise ValidationError('检定日期小于历史记录')
|
||||
check_date = attrs["check_date"]
|
||||
if EcheckRecord.objects.filter(check_date__gte=check_date, equipment=attrs["equipment"]).exists():
|
||||
raise ValidationError("检定日期小于历史记录")
|
||||
return attrs
|
||||
|
||||
|
||||
class EInspectSerializer(CustomModelSerializer):
|
||||
equipment_name = serializers.StringRelatedField(
|
||||
source='equipment', read_only=True)
|
||||
inspect_user_name = serializers.CharField(
|
||||
source='inspect_user.name', read_only=True)
|
||||
equipment_name = serializers.StringRelatedField(source="equipment", read_only=True)
|
||||
inspect_user_name = serializers.CharField(source="inspect_user.name", read_only=True)
|
||||
|
||||
class Meta:
|
||||
model = EInspect
|
||||
fields = '__all__'
|
||||
fields = "__all__"
|
||||
read_only_fields = EXCLUDE_FIELDS
|
||||
|
|
Loading…
Reference in New Issue