From acc1efe2c803e362fe6bfb3283e978be9a715504 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Thu, 22 Sep 2022 10:58:10 +0800
Subject: [PATCH] =?UTF-8?q?blt=20all=20=E5=A2=9E=E5=8A=A0depts=E7=AD=9B?=
 =?UTF-8?q?=E9=80=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/third/serializers.py | 2 ++
 apps/third/views_d.py     | 2 ++
 2 files changed, 4 insertions(+)

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: