31 lines
945 B
Python
31 lines
945 B
Python
from apps.utils.serializers import CustomModelSerializer
|
|
from apps.utils.constants import EXCLUDE_FIELDS_BASE, EXCLUDE_FIELDS_DEPT
|
|
from .models import Drain, DrainEquip
|
|
from apps.em.serializers import EquipmentSerializer
|
|
from rest_framework import serializers
|
|
|
|
|
|
class DrainSerializer(CustomModelSerializer):
|
|
"""Serializer for Drain model"""
|
|
|
|
class Meta:
|
|
model = Drain
|
|
fields = "__all__"
|
|
read_only_fields = EXCLUDE_FIELDS_DEPT
|
|
|
|
def validate(self, attrs):
|
|
attrs['belong_dept'] = attrs['mgroup'].belong_dept
|
|
return super().validate(attrs)
|
|
|
|
|
|
class DrainEquipSerializer(CustomModelSerializer):
|
|
equipment_type = serializers.CharField(
|
|
source='equipment.type', read_only=True)
|
|
equipment_name = serializers.CharField(
|
|
source='equipment.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = DrainEquip
|
|
fields = "__all__"
|
|
read_only_fields = EXCLUDE_FIELDS_BASE
|