69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
|
from fastapi import Request, status
|
|||
|
|
from fastapi.responses import JSONResponse
|
|||
|
|
from fastapi.exceptions import HTTPException, RequestValidationError
|
|||
|
|
from mysql.connector import Error as MySQLError
|
|||
|
|
from jose import JWTError
|
|||
|
|
|
|||
|
|
from schema.response_schema import APIResponse
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def global_exception_handler(request: Request, exc: Exception):
|
|||
|
|
"""全局异常处理器：所有未捕获的异常都会在这里统一处理"""
|
|||
|
|
# 1. 请求参数验证错误（Pydantic 校验失败）
|
|||
|
|
if isinstance(exc, RequestValidationError):
|
|||
|
|
error_details = []
|
|||
|
|
for err in exc.errors():
|
|||
|
|
error_details.append(f"{err['loc'][1]}: {err['msg']}")
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|||
|
|
content=APIResponse(
|
|||
|
|
code=400,
|
|||
|
|
message=f"请求参数错误：{'; '.join(error_details)}",
|
|||
|
|
data=None
|
|||
|
|
).model_dump()
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 2. HTTP 异常（主动抛出的业务错误、如 401/404）
|
|||
|
|
if isinstance(exc, HTTPException):
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=exc.status_code,
|
|||
|
|
content=APIResponse(
|
|||
|
|
code=exc.status_code,
|
|||
|
|
message=exc.detail,
|
|||
|
|
data=None
|
|||
|
|
).model_dump()
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 3. JWT 相关错误（Token 无效/过期）
|
|||
|
|
if isinstance(exc, JWTError):
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|||
|
|
content=APIResponse(
|
|||
|
|
code=401,
|
|||
|
|
message="Token 无效或已过期",
|
|||
|
|
data=None
|
|||
|
|
).model_dump(),
|
|||
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 4. MySQL 数据库错误
|
|||
|
|
if isinstance(exc, MySQLError):
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|||
|
|
content=APIResponse(
|
|||
|
|
code=500,
|
|||
|
|
message=f"数据库错误：{str(exc)}",
|
|||
|
|
data=None
|
|||
|
|
).model_dump()
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 5. 其他未知错误（兜底处理）
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|||
|
|
content=APIResponse(
|
|||
|
|
code=500,
|
|||
|
|
message=f"服务器内部错误：{str(exc)}",
|
|||
|
|
data=None
|
|||
|
|
).model_dump()
|
|||
|
|
)
|