lint and format

This commit is contained in:
grillazz
2025-08-24 15:40:06 +02:00
parent 9c7db17da8
commit 978041c6ee
7 changed files with 37 additions and 106 deletions

View File

@@ -1,3 +1,3 @@
from app.exception_handlers.registry import register_exception_handlers
__all__ = ["register_exception_handlers"]
__all__ = ["register_exception_handlers"]

View File

@@ -1,8 +1,7 @@
import orjson
from attrs import define, field
from fastapi import Request
from rotoger import AppStructLogger
from attrs import define, field
logger = AppStructLogger().get_logger()
@@ -10,6 +9,7 @@ logger = AppStructLogger().get_logger()
@define(slots=True)
class RequestInfo:
"""Contains extracted request information."""
path: str = field()
body: dict = field(default=None)
@@ -39,5 +39,5 @@ class BaseExceptionHandler:
message,
request_url=request_info.path,
request_body=request_info.body,
**kwargs
)
**kwargs,
)

View File

@@ -1,6 +1,7 @@
from fastapi import Request
from fastapi.responses import JSONResponse
from sqlalchemy.exc import SQLAlchemyError
from app.exception_handlers.base import BaseExceptionHandler
@@ -8,16 +9,16 @@ class SQLAlchemyExceptionHandler(BaseExceptionHandler):
"""Handles SQLAlchemy database exceptions."""
@classmethod
async def handle_exception(cls, request: Request, exc: SQLAlchemyError) -> JSONResponse:
async def handle_exception(
cls, request: Request, exc: SQLAlchemyError
) -> JSONResponse:
request_info = await cls.extract_request_info(request)
await cls.log_error(
"Database error occurred",
request_info,
sql_error=repr(exc)
"Database error occurred", request_info, sql_error=repr(exc)
)
return JSONResponse(
status_code=500,
content={"message": "A database error occurred. Please try again later."}
)
content={"message": "A database error occurred. Please try again later."},
)

View File

@@ -1,11 +1,16 @@
from fastapi import FastAPI
from sqlalchemy.exc import SQLAlchemyError
from fastapi.exceptions import ResponseValidationError
from sqlalchemy.exc import SQLAlchemyError
from app.exception_handlers.database import SQLAlchemyExceptionHandler
from app.exception_handlers.validation import ResponseValidationExceptionHandler
def register_exception_handlers(app: FastAPI) -> None:
"""Register all exception handlers with the FastAPI app."""
app.add_exception_handler(SQLAlchemyError, SQLAlchemyExceptionHandler.handle_exception)
app.add_exception_handler(ResponseValidationError, ResponseValidationExceptionHandler.handle_exception)
app.add_exception_handler(
SQLAlchemyError, SQLAlchemyExceptionHandler.handle_exception
)
app.add_exception_handler(
ResponseValidationError, ResponseValidationExceptionHandler.handle_exception
)

View File

@@ -9,14 +9,18 @@ class ResponseValidationExceptionHandler(BaseExceptionHandler):
"""Handles response validation exceptions."""
@classmethod
async def handle_exception(cls, request: Request, exc: ResponseValidationError) -> JSONResponse:
async def handle_exception(
cls, request: Request, exc: ResponseValidationError
) -> JSONResponse:
request_info = await cls.extract_request_info(request)
errors = exc.errors()
# Check if this is a None/null response case
is_none_response = False
for error in errors:
if error.get("input") is None and "valid dictionary" in error.get("msg", ""):
if error.get("input") is None and "valid dictionary" in error.get(
"msg", ""
):
is_none_response = True
break
@@ -24,16 +28,15 @@ class ResponseValidationExceptionHandler(BaseExceptionHandler):
"Response validation error occurred",
request_info,
validation_errors=errors,
is_none_response=is_none_response
is_none_response=is_none_response,
)
if is_none_response:
return JSONResponse(
status_code=404,
content={"no_response": "The requested resource was not found"}
content={"no_response": "The requested resource was not found"},
)
else:
return JSONResponse(
status_code=422,
content={"response_format_error": errors}
)
status_code=422, content={"response_format_error": errors}
)