chenjunkai 6 жил өмнө
parent
commit
7dbc63d657

+ 3 - 1
Controller/EquipmentManager.py

@@ -222,6 +222,8 @@ def deleteInterface(request):
         return response.json(tko.code)
     userID = tko.userID
     # 主用户删除设备全部删除
+    if not userID:
+        return response.json(309)
     try:
         dv_qs = Device_Info.objects.filter(userID_id=userID, id=id)
         uid = dv_qs[0].UID
@@ -299,7 +301,7 @@ def showAllUserEquipmentInterface(request, *callback_args, **callback_kwargs):
         return response.json(tko.code)
     userID = tko.userID
     if not userID:
-        return response.json(104)
+        return response.json(309)
     own_perm = ModelService.check_perm(userID=userID, permID=30)
     if not own_perm:
         return response.json(404)

+ 6 - 4
Object/TokenObject.py

@@ -53,6 +53,8 @@ class TokenObject:
             self.code = 309
         else:
             self.code = 0
+            if not self.userID:
+                self.code = 309
             return res
 
     def generate(self, data={}):
@@ -96,8 +98,8 @@ class TokenObject:
             print(repr(e))
         else:
             self.code = 0
-            userID = res.get('userID','')
-            lang = res.get('lang','')
-            user = res.get('user','')
-            refreshRes = self.generate(data={'userID': userID, 'lang': lang,'user':user})
+            userID = res.get('userID', '')
+            lang = res.get('lang', '')
+            user = res.get('user', '')
+            refreshRes = self.generate(data={'userID': userID, 'lang': lang, 'user': user})
             return refreshRes