From 209a3fc5cb4a879dac41a06110c229718b3bd912 Mon Sep 17 00:00:00 2001
From: youhuo <sunfeiyu.sfy@alibaba-inc.com>
Date: Wed, 24 Aug 2022 17:10:37 +0800
Subject: [PATCH] fix: jwt lack user role info

---
 services/auth_service.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/services/auth_service.py b/services/auth_service.py
index 7a2bee0..e9e6a4b 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
-- 
Gitee