Kaynağa Gözat

优化加密token

peng 10 ay önce
ebeveyn
işleme
b6bce9b423
2 değiştirilmiş dosya ile 3 ekleme ve 3 silme
  1. 2 2
      Controller/PctestController.py
  2. 1 1
      Object/TokenObject.py

+ 2 - 2
Controller/PctestController.py

@@ -49,7 +49,7 @@ class PcTest(View):
                 if tko.code != 0:
                     return response.json(tko.code)
                 # 获取访问者的id和岗位
-                userID = tko.userID
+                userID = tko.id
                 jobID = tko.job
                 # if jobID == 1:
                 #     # 管理员可访问的接口
@@ -119,7 +119,7 @@ class PcTest(View):
         tko = TokenObject()
         # 加密
         res = tko.generate(
-            data={'userID': users['id'], 'username': users['username'], 'password': users['password'], 'jobid': users['job__id'], 'job': users['job__jobcode']})
+            data={'id': users['id'], 'username': users['username'], 'password': users['password'], 'jobid': users['job__id'], 'job': users['job__jobcode']})
         res_qs = {
             'res': res,
             'id': users['id'],

+ 1 - 1
Object/TokenObject.py

@@ -118,7 +118,7 @@ class TokenObject:
                 }
             app_bundle_id = data.get('appBundleId', None)
             token_version = data.get('tokenVersion', None)
-            LOGGER.info('{}生成新token,参数:{}'.format(data['userID'], data))
+            LOGGER.info('{}生成新token,参数:{}'.format(data.get('userID', 'pctest'), data))
             if app_bundle_id and token_version == 'V2':
                 redis_obj = RedisObject()
                 key = 'token_user_{}_{}'.format(data['userID'], app_bundle_id)