feat: 增加drain2serializer返回数据

This commit is contained in:
caoqianming 2024-02-27 14:46:28 +08:00
parent 666d317f01
commit 44c4a34c69
2 changed files with 2 additions and 0 deletions

View File

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

View File

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