浏览代码

token放置头文件;优化编辑接口

guanhailong 2 年之前
父节点
当前提交
1a580173ff
共有 1 个文件被更改,包括 8 次插入10 次删除
  1. 8 10
      Controller/SuperPasswordTool.py

+ 8 - 10
Controller/SuperPasswordTool.py

@@ -35,15 +35,12 @@ class SuperPasswordView(View):
         return self.validation(request_dict, request, operation)
 
     def validation(self, request_dict, request, operation):
-        token = request_dict.get('token', None)
-        response = ResponseObject()
-        tko = TokenObject(token)
+        tko = TokenObject(
+            request.META.get('HTTP_AUTHORIZATION'))
+        lang = request_dict.get('lang', tko.lang)
+        response = ResponseObject(lang)
         if tko.code != 0:
-            tko = TokenObject(
-                request.META.get('HTTP_AUTHORIZATION'))
-            if tko.code != 0:
-                return response.json(tko.code)
-            response.lang = tko.lang
+            return response.json(tko.code)
         userID = tko.userID
         if operation == 'getAuthorizationCode':  # 用户提交请求
             return self.getAuthorizationCode(request_dict, response, userID)
@@ -104,17 +101,18 @@ class SuperPasswordView(View):
         @param hint:提示内容
         @return:
         """
+        ID = request_dict.get('ID', None)
         userID = request_dict.get('userID', None)
         uid = request_dict.get('uid', None)
         status = request_dict.get('status', None)
         hint = request_dict.get('hint', None)
         lang = request_dict.get('lang', 'en')
-        if not all({uid, userID}):
+        if not all({ID, uid, userID}):
             return response.json(444)
         now = int(time.time())
         try:
             with transaction.atomic():
-                device_super_password_qs = DeviceSuperPassword.objects.filter(uid=uid, userID=userID)
+                device_super_password_qs = DeviceSuperPassword.objects.filter(id=ID, uid=uid, userID=userID)
                 if not device_super_password_qs.exists():
                     return response.json(173)
                 status = int(status)