浏览代码

新增获取AI开关状态接口

locky 3 年之前
父节点
当前提交
ea2e2e9ba4
共有 1 个文件被更改,包括 19 次插入0 次删除
  1. 19 0
      Controller/AiController.py

+ 19 - 0
Controller/AiController.py

@@ -100,6 +100,8 @@ class AiView(View):
                 return self.do_create_pay_order(request_dict, userID, ip, response)
             elif operation == 'changeaistatus':  # 修改AI开关状态
                 return self.do_change_ai_status(userID, request_dict, response)
+            elif operation == 'getAiStatus':  # 获取AI开关状态
+                return self.getAiStatus(userID, request_dict, response)
             elif operation == 'commoditylist':  # 修改云存状态,传送两个url,即getsignsts接口和storeplaylist接口
                 return self.do_commodity_list(userID, request_dict, response)
             elif operation == 'identification':  # ai识别
@@ -266,6 +268,23 @@ class AiView(View):
         else:
             return response.json(14)
 
+    def getAiStatus(self, userID, request_dict, response):
+        uid = request_dict.get('uid', None)
+
+        if not uid:
+            return response.json(444)
+        try:
+            ai_server_qs = AiService.objects.filter(uid=uid).values('detect_status', 'detect_group')
+            if not ai_server_qs.exists():
+                return response,json(173)
+            res = {
+                'detect_status': ai_server_qs[0]['detect_status'],
+                'detect_group': ai_server_qs[0]['detect_group'],
+            }
+            return response.json(0, {'data': res})
+        except Exception as e:
+            return response.json(500, repr(e))
+
     def requestPublishMqtt(self, thing_name, topic_name, msg):
         # 通用发布MQTT主题通知
         if not all([msg, thing_name, topic_name]):