File tree Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -43,22 +43,21 @@ def password_verify(plain_password: str, hashed_password: str) -> bool:
43
43
return pwd_context .verify (plain_password , hashed_password )
44
44
45
45
46
- async def create_access_token (sub : int | Any , data : dict , expires_delta : timedelta | None = None ) -> str :
46
+ async def create_access_token (sub : int | Any , expires_delta : timedelta | None = None , ** kwargs ) -> str :
47
47
"""
48
48
Generate encryption token
49
49
50
50
:param sub: The subject/userid of the JWT
51
- :param data: Data transferred to the token
52
51
:param expires_delta: Increased expiry time
53
52
:return:
54
53
"""
55
54
if expires_delta :
56
55
expires = datetime .utcnow () + expires_delta
57
56
expire_seconds = expires_delta .total_seconds ()
58
57
else :
59
- expires = datetime .utcnow () + timedelta (seconds = settings .TOKEN_EXPIRE_MINUTES )
58
+ expires = datetime .utcnow () + timedelta (seconds = settings .TOKEN_EXPIRE_SECONDS )
60
59
expire_seconds = settings .TOKEN_EXPIRE_SECONDS
61
- to_encode = {'exp' : expires , 'sub' : str (sub ), ** data }
60
+ to_encode = {'exp' : expires , 'sub' : str (sub ), ** kwargs }
62
61
token = jwt .encode (to_encode , settings .TOKEN_SECRET_KEY , settings .TOKEN_ALGORITHM )
63
62
if sub not in settings .TOKEN_WHITE_LIST :
64
63
await redis_client .delete (f'token:{ sub } :*' )
Original file line number Diff line number Diff line change @@ -32,7 +32,7 @@ async def swagger_login(form_data: OAuth2PasswordRequestForm):
32
32
# 获取最新用户信息
33
33
user = await UserDao .get_user_by_id (db , current_user .id )
34
34
# 创建token
35
- access_token = await jwt .create_access_token (user .id , { ' role_ids' : user_role_ids } )
35
+ access_token = await jwt .create_access_token (user .id , role_ids = user_role_ids )
36
36
return access_token , user
37
37
38
38
@staticmethod
@@ -48,7 +48,7 @@ async def login(obj: Auth):
48
48
await UserDao .update_user_login_time (db , obj .username )
49
49
user_role_ids = await UserDao .get_user_role_ids (db , current_user .id )
50
50
user = await UserDao .get_user_by_id (db , current_user .id )
51
- access_token = await jwt .create_access_token (user .id , { ' role_ids' : user_role_ids } )
51
+ access_token = await jwt .create_access_token (user .id , role_ids = user_role_ids )
52
52
return access_token , user
53
53
54
54
@staticmethod
You can’t perform that action at this time.
0 commit comments