优化auth_api.py

This commit is contained in:
jingrow 2025-11-05 06:05:35 +08:00
parent 02c09fa975
commit 0ba07c4b82

View File

@ -83,13 +83,11 @@ async def login_route(login_data: LoginRequest):
user_info_result = get_user_info(session_cookie) user_info_result = get_user_info(session_cookie)
user_info = user_info_result.get("user_info") if user_info_result.get("success") else None user_info = user_info_result.get("user_info") if user_info_result.get("success") else None
response_data = { return create_response_with_cookie({
"success": True, "success": True,
"message": result.get("message", "Logged In"), "message": result.get("message", "Logged In"),
"user": user_info "user": user_info
} }, session_cookie)
return create_response_with_cookie(response_data, session_cookie)
except HTTPException: except HTTPException:
raise raise
except Exception as e: except Exception as e:
@ -102,13 +100,10 @@ async def logout_route(session_cookie: Optional[str] = Depends(get_session_cooki
try: try:
result = logout(session_cookie) result = logout(session_cookie)
handle_auth_result(result, error_status_code=400) handle_auth_result(result, error_status_code=400)
return create_response_clear_cookies({
response_data = {
"success": True, "success": True,
"message": result.get("message", "登出成功") "message": result.get("message", "登出成功")
} })
return create_response_clear_cookies(response_data)
except HTTPException: except HTTPException:
raise raise
except Exception as e: except Exception as e:
@ -121,7 +116,6 @@ async def get_user_info_route(session_cookie: Optional[str] = Depends(get_sessio
try: try:
result = get_user_info(session_cookie) result = get_user_info(session_cookie)
handle_auth_result(result, error_status_code=401) handle_auth_result(result, error_status_code=401)
return { return {
"success": True, "success": True,
"user_info": result.get("user_info") "user_info": result.get("user_info")