统计设备关联产品方案

Controller/IncomeProductsController.py

@@ -22,6 +22,8 @@ class IncomeProductsView(View):
         token_code, user_id, response = CommonService.verify_token_get_user_id(request_dict, request)
         if operation == 'income':  # 设备关联产品方案
             return self.income_device_scheme(request_dict, response)
+        elif operation == 'statistics':
+            return self.statistics_device_scheme(request_dict, response)
         else:
             return response.json(414)
 
@@ -65,6 +67,24 @@ class IncomeProductsView(View):
 
             return response.json(0)
 
+        except Exception as e:
+            print(e)
+            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
+
+    @staticmethod
+    def statistics_device_scheme(request_dict, response):
+        """
+        统计设备关联产品方案
+        @param request_dict:
+        @param response:
+        @return:
+        """
+        try:
+            storage_code = request_dict.get("storageCode", None)
+            if not storage_code:
+                return response.json(444)
+            device_num = DeviceScheme.objects.filter(storage_code=storage_code).count()
+            return response.json(0, {"storageCode": storage_code,"deviceNum": device_num})
         except Exception as e:
             print(e)
             return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))