diff --git a/apps/enp/serializers.py b/apps/enp/serializers.py index e1359156..496923f5 100644 --- a/apps/enp/serializers.py +++ b/apps/enp/serializers.py @@ -24,6 +24,7 @@ class DrainSerializer(CustomModelSerializer): class Drain2Serializer(CustomModelSerializer): equip_data = serializers.SerializerMethodField() + mgroup_name = serializers.CharField(source='mgroup.name', read_only=True) class Meta: model = Drain diff --git a/apps/enp/views.py b/apps/enp/views.py index c8cdb3b7..eed9f4f9 100644 --- a/apps/enp/views.py +++ b/apps/enp/views.py @@ -21,6 +21,7 @@ class DrainViewSet(CustomModelViewSet): queryset = Drain.objects.all() serializer_class = DrainSerializer filterset_fields = ['type', 'cate', 'mgroup'] + select_related_fields = ['mgroup'] def get_serializer_class(self): has_equipdata = self.request.query_params.get('has_equipdata', 'no')