|
@@ -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
|