56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
from fastapi.exceptions import (
|
||
HTTPException,
|
||
RequestValidationError,
|
||
ResponseValidationError,
|
||
)
|
||
from fastapi.requests import Request
|
||
from fastapi.responses import JSONResponse
|
||
from tortoise.exceptions import DoesNotExist, IntegrityError
|
||
from app.log import logger
|
||
|
||
|
||
class SettingNotFound(Exception):
|
||
pass
|
||
|
||
|
||
async def DoesNotExistHandle(req: Request, exc: DoesNotExist) -> JSONResponse:
|
||
content = dict(
|
||
code=404,
|
||
msg=f"Object has not found, exc: {exc}, query_params: {req.query_params}",
|
||
)
|
||
return JSONResponse(content=content, status_code=404)
|
||
|
||
|
||
async def IntegrityHandle(_: Request, exc: IntegrityError) -> JSONResponse:
|
||
content = dict(
|
||
code=500,
|
||
msg=f"IntegrityError，{exc}",
|
||
)
|
||
return JSONResponse(content=content, status_code=500)
|
||
|
||
|
||
async def HttpExcHandle(_: Request, exc: HTTPException) -> JSONResponse:
|
||
content = dict(code=exc.status_code, msg=exc.detail, data=None)
|
||
return JSONResponse(content=content, status_code=exc.status_code)
|
||
|
||
|
||
async def RequestValidationHandle(req: Request, exc: RequestValidationError) -> JSONResponse:
|
||
try:
|
||
body_bytes = await req.body()
|
||
body_text = body_bytes.decode("utf-8", errors="replace")
|
||
if len(body_text) > 10000:
|
||
body_text = body_text[:10000] + "..."
|
||
logger.error(
|
||
f"422 RequestValidationError path={req.url.path} errors={exc.errors()} body={body_text}"
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"422 RequestValidationError logging failed: {e}")
|
||
|
||
content = dict(code=422, msg=f"RequestValidationError, {exc}")
|
||
return JSONResponse(content=content, status_code=422)
|
||
|
||
|
||
async def ResponseValidationHandle(_: Request, exc: ResponseValidationError) -> JSONResponse:
|
||
content = dict(code=500, msg=f"ResponseValidationError, {exc}")
|
||
return JSONResponse(content=content, status_code=500)
|