doc: drain view swagger 变动3

This commit is contained in:
caoqianming 2024-08-08 15:07:05 +08:00
parent fff6c76cc6
commit a2b52e95a5
1 changed files with 2 additions and 2 deletions

View File

@ -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(