Sergey Vanyushkin
66ce5ae746
Co-authored-by: Sergey Vanyushkin <pi3c@yandex.ru> Co-committed-by: Sergey Vanyushkin <pi3c@yandex.ru>
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from datetime import datetime
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.responses import JSONResponse
|
|
from pydantic import BaseModel
|
|
from starlette.exceptions import HTTPException
|
|
|
|
from app.core.exceptions import AppException
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
status_code: int
|
|
message: str
|
|
details: dict | None = None
|
|
timestamp: str
|
|
|
|
|
|
async def app_exception_handler(request: Request, exc: AppException) -> JSONResponse:
|
|
return JSONResponse(
|
|
status_code=exc.status_code,
|
|
content={
|
|
"status_code": exc.status_code,
|
|
"message": exc.message,
|
|
"timestamp": datetime.utcnow().isoformat(),
|
|
},
|
|
)
|
|
|
|
|
|
async def http_exception_handler(request: Request, exc: HTTPException) -> JSONResponse:
|
|
return JSONResponse(
|
|
status_code=exc.status_code,
|
|
content={
|
|
"status_code": exc.status_code,
|
|
"message": str(exc.detail),
|
|
"timestamp": datetime.utcnow().isoformat(),
|
|
},
|
|
)
|
|
|
|
|
|
def register_exception_handlers(app: FastAPI):
|
|
app.add_exception_handler(
|
|
AppException,
|
|
app_exception_handler, # type: ignore[arg-type]
|
|
)
|
|
app.add_exception_handler(
|
|
HTTPException,
|
|
http_exception_handler, # type: ignore[arg-type]
|
|
)
|