diff --git a/apps/third/serializers.py b/apps/third/serializers.py index cb61a2a3..e4cf48a0 100755 --- a/apps/third/serializers.py +++ b/apps/third/serializers.py @@ -86,6 +86,8 @@ class BltSerializer(serializers.ModelSerializer): class BltQuerySerializer(serializers.ModelSerializer): + depts = serializers.ListField(child=serializers.CharField(), label="部门ID列表", required=False, allow_blank=True) + class Meta: model = TDevice fields = ['area'] diff --git a/apps/third/views_d.py b/apps/third/views_d.py index 04f40e66..a883c606 100644 --- a/apps/third/views_d.py +++ b/apps/third/views_d.py @@ -99,6 +99,8 @@ class BltViewSet(CustomGenericViewSet): i['mac'] = i['userId'] macs.append(i['mac']) qs = self.queryset.filter(code__in=macs).exclude(employee=None) + if data.get('depts', []): + qs = qs.filter(employee__belong_dept=data['depts']) qs_data = BltSerializer(instance=qs, many=True).data qs_dict = {} for i in qs_data: