浏览代码

修改接口的状态,加多一个参数all=1就是标记全部为已读。不传这个参数就按照之前的传id一个个来标记已读,

pengzhibo168 5 年之前
父节点
当前提交
d6653e5f12
共有 1 个文件被更改,包括 27 次插入18 次删除
  1. 27 18
      Controller/EquipmentInfo.py

+ 27 - 18
Controller/EquipmentInfo.py

@@ -149,25 +149,34 @@ class EquipmentInfo(View):
 
 
     def update_info(self, request_dict, userID, response):
-        id_list = request_dict.getlist('id[]', None)
-        if id_list is None or len(id_list) < 1:
-            id_list = request_dict.getlist('id', None)
-        param_flag = CommonService.get_param_flag(data=[id_list])
-        if param_flag is True:
-            count = 0
-            for id in id_list:
-                try:
-                    eq = Equipment_Info.objects.filter(id=int(id))
-                    if eq.exists():
-                        own_dev = ModelService.check_own_device(userID, eq[0].devUid)
-                        if own_dev is True:
-                            count += 1
-                            eq.update(status=1)
-                except Exception as e:
-                    print(repr(e))
-            return response.json(0, {'update_success': count})
+        id_all = request_dict.getlist('all', None)
+        if id_all == 1:
+            try:
+                eq = Equipment_Info.objects.filter(userID=userID)
+                eq.update(status=1)
+            except Exception as e:
+                print(repr(e))
+            return response.json(0, {'update_success': 'all'})
         else:
-            return response.json(444)
+            id_list = request_dict.getlist('id[]', None)
+            if id_list is None or len(id_list) < 1:
+                id_list = request_dict.getlist('id', None)
+            param_flag = CommonService.get_param_flag(data=[id_list])
+            if param_flag is True:
+                count = 0
+                for id in id_list:
+                    try:
+                        eq = Equipment_Info.objects.filter(id=int(id))
+                        if eq.exists():
+                            own_dev = ModelService.check_own_device(userID, eq[0].devUid)
+                            if own_dev is True:
+                                count += 1
+                                eq.update(status=1)
+                    except Exception as e:
+                        print(repr(e))
+                return response.json(0, {'update_success': count})
+            else:
+                return response.json(444)
 
     def delete_info(self, request_dict, userID, response):
         id_list = request_dict.getlist('id[]', None)