File size: 2,623 Bytes
ef1ad9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
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 # pylint: disable=protected-access
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},
)
|