update

smartfarming/views/counts_views.py

@@ -269,6 +269,7 @@ class APPAlarmAPIView(APIView):
         # 气象站与测报预警
         request_data = request.data 
         type_id = request_data.get("type_id")
+        device_id = request_data.get("device_id")
         page_num = int(request_data.get("page", 1)) if request_data.get("page") else 1
         page_size = int(request_data.get("page_size", 10)) if request_data.get("page_size") else 10
         data = {}
@@ -276,6 +277,8 @@ class APPAlarmAPIView(APIView):
         if type_id == "3":
             # 虫情
             queryset = MongoCBDPestWarning.objects.all().order_by("-upltime")
+            if device_id:
+                queryset = queryset.filter(device_id=device_id)
             total_obj = queryset.count()
             paginator = Paginator(queryset, page_size)
             page_obj = paginator.get_page(page_num)
@@ -283,6 +286,9 @@ class APPAlarmAPIView(APIView):
             data = pest_serializers.data 
         elif type_id == "5":
             queryset = MongoQXZ_Alarm_Log_New.objects.filter(warning_name="5").order_by("-upl_time")
+            if device_id:
+                d_id = MongoDevice.objects.get(device_id=device_id).id
+                queryset = queryset.filter(device_id=d_id)
             total_obj = queryset.count()
             paginator = Paginator(queryset, page_size)
             page_obj = paginator.get_page(page_num)
@@ -290,6 +296,9 @@ class APPAlarmAPIView(APIView):
             data = qxz_serializers.data
         elif type_id == "8":
             queryset = MongoQXZ_Alarm_Log_New.objects.filter(warning_name="8").order_by("-upl_time")
+            if device_id:
+                d_id = MongoDevice.objects.get(device_id=device_id).id
+                queryset = queryset.filter(device_id=d_id)
             total_obj = queryset.count()
             paginator = Paginator(queryset, page_size)
             page_obj = paginator.get_page(page_num)