|
|
import sys
|
|
|
|
|
|
from typing import Union
|
|
|
|
|
|
from fastapi import Request
|
|
|
from fastapi.exceptions import RequestValidationError, HTTPException
|
|
|
from fastapi.exception_handlers import http_exception_handler as _http_exception_handler
|
|
|
from fastapi.exception_handlers import (
|
|
|
request_validation_exception_handler as _request_validation_exception_handler,
|
|
|
)
|
|
|
from fastapi.responses import JSONResponse
|
|
|
from fastapi.responses import PlainTextResponse
|
|
|
from fastapi.responses import Response
|
|
|
|
|
|
from app.utils.app_logger.logger import logger
|
|
|
|
|
|
async def request_validation_exception_handler(request: Request, exc: RequestValidationError) -> JSONResponse:
|
|
|
"""
|
|
|
This is a wrapper to the default RequestValidationException handler of FastAPI.
|
|
|
This function will be called when client input is not valid.
|
|
|
"""
|
|
|
logger.debug("Our custom request_validation_exception_handler was called")
|
|
|
body = await request.body()
|
|
|
query_params = request.query_params._dict
|
|
|
detail = {"errors": exc.errors(), "body": body.decode(), "query_params": query_params}
|
|
|
logger.info(detail)
|
|
|
return await _request_validation_exception_handler(request, exc)
|
|
|
|
|
|
|
|
|
async def http_exception_handler(request: Request, exc: HTTPException) -> Union[JSONResponse, Response]:
|
|
|
"""
|
|
|
This is a wrapper to the default HTTPException handler of FastAPI.
|
|
|
This function will be called when a HTTPException is explicitly raised.
|
|
|
"""
|
|
|
logger.debug("Our custom http_exception_handler was called")
|
|
|
return await _http_exception_handler(request, exc)
|
|
|
|
|
|
|
|
|
async def unhandled_exception_handler(request: Request, exc: Exception) -> PlainTextResponse:
|
|
|
"""
|
|
|
This middleware will log all unhandled exceptions.
|
|
|
Unhandled exceptions are all exceptions that are not HTTPExceptions or RequestValidationErrors.
|
|
|
"""
|
|
|
logger.debug("Our custom unhandled_exception_handler was called")
|
|
|
host = getattr(getattr(request, "client", None), "host", None)
|
|
|
port = getattr(getattr(request, "client", None), "port", None)
|
|
|
url = f"{request.url.path}?{request.query_params}" if request.query_params else request.url.path
|
|
|
exception_type, exception_value, exception_traceback = sys.exc_info()
|
|
|
exception_name = getattr(exception_type, "__name__", None)
|
|
|
logger.error(
|
|
|
f'{host}:{port} - "{request.method} {url}" 500 Internal Server Error <{exception_name}: {exception_value}>'
|
|
|
)
|
|
|
return JSONResponse(
|
|
|
status_code=500,
|
|
|
content={"status": 500,
|
|
|
"message": str(exc), "success": False},
|
|
|
)
|
|
|
|
