From a2b52e95a5eaac9dda7593ed3c54a7fcf942f07f Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Thu, 8 Aug 2024 15:07:05 +0800
Subject: [PATCH] =?UTF-8?q?doc:=20drain=20view=20swagger=20=E5=8F=98?=
 =?UTF-8?q?=E5=8A=A83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/enp/views.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/apps/enp/views.py b/apps/enp/views.py
index f862c60f..225f8ee1 100644
--- a/apps/enp/views.py
+++ b/apps/enp/views.py
@@ -30,9 +30,9 @@ class DrainViewSet(CustomModelViewSet):
         equip_data = self.request.query_params.get("equip_data", "no")
         if self.request.method == "GET":
             if equip_data == "base":
-                return DrainWithEquipEnpSerializer
-            elif equip_data == "enp":
                 return DrainWithEquipBaseSerializer
+            elif equip_data == "enp":
+                return DrainWithEquipEnpSerializer
         return super().get_serializer_class()
 
     @swagger_auto_schema(