Prechádzať zdrojové kódy

判断userID是否存在

chenjunkai 6 rokov pred
rodič
commit
7dbc63d657
2 zmenil súbory, kde vykonal 9 pridanie a 5 odobranie
  1. 3 1
      Controller/EquipmentManager.py
  2. 6 4
      Object/TokenObject.py

+ 3 - 1
Controller/EquipmentManager.py

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

+ 6 - 4
Object/TokenObject.py

@@ -53,6 +53,8 @@ class TokenObject:
             self.code = 309
             self.code = 309
         else:
         else:
             self.code = 0
             self.code = 0
+            if not self.userID:
+                self.code = 309
             return res
             return res
 
 
     def generate(self, data={}):
     def generate(self, data={}):
@@ -96,8 +98,8 @@ class TokenObject:
             print(repr(e))
             print(repr(e))
         else:
         else:
             self.code = 0
             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
             return refreshRes