Răsfoiți Sursa

测试服和国内服token过期时间改为60天

locky 1 an în urmă
părinte
comite
e09ce1a660
1 a modificat fișierele cu 15 adăugiri și 5 ștergeri
  1. 15 5
      Object/TokenObject.py

+ 15 - 5
Object/TokenObject.py

@@ -1,7 +1,9 @@
+import datetime
+
 import jwt
 import time
 from Ansjer.config import OAUTH_ACCESS_TOKEN_SECRET, OAUTH_REFRESH_TOKEN_SECRET, OAUTH_ACCESS_TOKEN_TIME, \
-    OAUTH_REFRESH_TOKEN_TIME
+    OAUTH_REFRESH_TOKEN_TIME, CONFIG_INFO, CONFIG_TEST, CONFIG_CN
 
 
 class TokenObject:
@@ -58,8 +60,12 @@ class TokenObject:
         if data is None:
             data = {}
         try:
-            access_expire = int(OAUTH_ACCESS_TOKEN_TIME.total_seconds())
-            refresh_expire = int(OAUTH_REFRESH_TOKEN_TIME.total_seconds())
+            if CONFIG_INFO == CONFIG_TEST or CONFIG_INFO == CONFIG_CN:
+                access_expire = int(datetime.timedelta(days=60).total_seconds())
+                refresh_expire = int(datetime.timedelta(days=60).total_seconds())
+            else:
+                access_expire = int(OAUTH_ACCESS_TOKEN_TIME.total_seconds())
+                refresh_expire = int(OAUTH_REFRESH_TOKEN_TIME.total_seconds())
             now_stamp = int(time.time())
             access_data = data
             refresh_data = data
@@ -99,8 +105,12 @@ class TokenObject:
         if data is None:
             data = {}
         try:
-            access_expire = int(OAUTH_ACCESS_TOKEN_TIME.total_seconds())
-            refresh_expire = int(OAUTH_REFRESH_TOKEN_TIME.total_seconds())
+            if CONFIG_INFO == CONFIG_TEST or CONFIG_INFO == CONFIG_CN:
+                access_expire = int(datetime.timedelta(days=60).total_seconds())
+                refresh_expire = int(datetime.timedelta(days=60).total_seconds())
+            else:
+                access_expire = int(OAUTH_ACCESS_TOKEN_TIME.total_seconds())
+                refresh_expire = int(OAUTH_REFRESH_TOKEN_TIME.total_seconds())
             now_stamp = int(time.time())
             access_data = data
             refresh_data = data