优化auth_api.py
This commit is contained in:
parent
02c09fa975
commit
0ba07c4b82
@ -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")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user