ソースを参照

获取令牌时保存用户地区信息

locky 1 年間 前
コミット
f11a2b1bff
1 ファイル変更5 行追加3 行削除
  1. 5 3
      controller/AppToApp.py

+ 5 - 3
controller/AppToApp.py

@@ -37,7 +37,8 @@ class Oa2View(View):
     @staticmethod
     def get_auth_code(request_dict, response):
         user_id = request_dict.get('user_id', None)
-        if not user_id:
+        region_code = request_dict.get('region_code', None)
+        if not all([user_id, region_code]):
             return response.json(10, 'error params')
 
         now_time = int(time.time())
@@ -48,9 +49,10 @@ class Oa2View(View):
         # 用户不存在则创建
         if not user_qs.exists():
             UserModel.objects.create(userID=user_id, access_token=access_token, refresh_token=refresh_token,
-                                     user_authorization_code=user_authorization_code, addTime=now_time, updTime=now_time)
+                                     user_authorization_code=user_authorization_code, region_code=region_code,
+                                     addTime=now_time, updTime=now_time)
         else:
-            user_qs.update(access_token=access_token, refresh_token=refresh_token,
+            user_qs.update(access_token=access_token, refresh_token=refresh_token, region_code=region_code,
                            user_authorization_code=user_authorization_code, updTime=now_time)
         res = {
             'user_authorization_code': user_authorization_code