i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-01-03 00:02:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(auth): make get_current_user and get_current_user_from_header async functions

Browse Source 
将 get_current_user 和 get_current_user_from_header 函数声明为 async，
并更新 AuthService.verify_token 的调用为 await，以正确处理异步 Token 验证。
This commit is contained in:
fawney19
2025-12-16 13:42:26 +08:00
parent 33265b4b13
commit edce43d45f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/utils/auth_utils.py
View File

@@ -19,7 +19,7 @@ from ..models.database import User, UserRole
security = HTTPBearer() security = HTTPBearer()
def get_current_user( async def get_current_user(
credentials: HTTPAuthorizationCredentials = Depends(security), db: Session = Depends(get_db) credentials: HTTPAuthorizationCredentials = Depends(security), db: Session = Depends(get_db)
) -> User: ) -> User:
""" """
@@ -41,7 +41,7 @@ def get_current_user(
try: try:
# 验证Token格式和签名 # 验证Token格式和签名
try: try:
payload = AuthService.verify_token(token) payload = await AuthService.verify_token(token)
except HTTPException as token_error: except HTTPException as token_error:
# 保持原始的HTTP状态码（如401 Unauthorized），不要转换为403 # 保持原始的HTTP状态码（如401 Unauthorized），不要转换为403
logger.error(f"Token验证失败: {token_error.status_code}: {token_error.detail}, Token前10位: {token[:10]}...") logger.error(f"Token验证失败: {token_error.status_code}: {token_error.detail}, Token前10位: {token[:10]}...")
@@ -122,7 +122,7 @@ def get_current_user(
) )
def get_current_user_from_header( async def get_current_user_from_header(
authorization: Optional[str] = Header(None), db: Session = Depends(get_db) authorization: Optional[str] = Header(None), db: Session = Depends(get_db)
) -> User: ) -> User:
""" """
@@ -144,7 +144,7 @@ def get_current_user_from_header(
token = authorization.replace("Bearer ", "") token = authorization.replace("Bearer ", "")
try: try:
payload = AuthService.verify_token(token) payload = await AuthService.verify_token(token)
user_id = payload.get("user_id") user_id = payload.get("user_id")
if not user_id: if not user_id: