From 44c4a34c69819cd801e750e4e9756391475e72eb Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Tue, 27 Feb 2024 14:46:28 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0drain2serializer?=
 =?UTF-8?q?=E8=BF=94=E5=9B=9E=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/enp/serializers.py | 1 +
 apps/enp/views.py       | 1 +
 2 files changed, 2 insertions(+)

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')