diff --git a/services/auth_service.py b/services/auth_service.py index 7a2bee0d33048d8a095b985033b66a2bc8600a34..e9e6a4bd866278a235163088c8f4afb1b54c1d84 100644 --- a/services/auth_service.py +++ b/services/auth_service.py @@ -29,7 +29,7 @@ async def check_user(data): user = await find_user_exists(login_name) if not user or user.password != password: return ERROR_WRONG_USER_INFO, False - token = create_jwt_token(user.nick_name) + token = create_jwt_token(user.nick_name, user.role) key = get_user_token_key(user.nick_name) await redis.conn.hset('testlib', key, token) return token, True @@ -168,10 +168,11 @@ def create_private_secret(): return ''.join(random.sample(string.ascii_letters + string.digits, 32)) -def create_jwt_token(name): +def create_jwt_token(name, role): current_time = datetime.datetime.now() payload = { "user_name": name, + "role": role, "exp": current_time + datetime.timedelta(hours=1), 'iat': current_time, 'iss': name