项目基础模块代码

app/common/jwt_check.py

@@ -2,6 +2,7 @@ import datetime
 
 import jwt
 from app.common.redis_cli import redis_conn
+from fastapi import Request
 
 # 过期时间，单位S
 exp = 6000
@@ -48,3 +49,11 @@ def check_token(token: str):
         raise jwt.ExpiredSignatureError("Expired Token")
     except jwt.InvalidTokenError:
         raise jwt.InvalidTokenError("Invalid Token")
+
+
+def get_user_id(request: Request):
+    """根据Request请求获取token"""
+    token = request.headers.get("Authorization")
+    decoded_payload = check_token(token)
+    user_id = decoded_payload['user_id']
+    return user_id