thatmattlove-hyperglass/hyperglass/api/error_handlers.py

"""API Error Handlers."""

# Third Party
from litestar import Request, Response

# Project
from hyperglass.log import log
from hyperglass.state import use_state

__all__ = (
    "default_handler",
    "http_handler",
    "app_handler",
    "validation_handler",
)


def default_handler(request: Request, exc: BaseException) -> Response:
    """Handle uncaught errors."""
    state = use_state()
    log.critical(
        "{method} {path} {detail!s}", method=request.method, path=request.url.path, detail=exc
    )
    return Response(
        {"output": state.params.messages.general, "level": "danger", "keywords": []},
        status_code=500,
    )


def http_handler(request: Request, exc: BaseException) -> Response:
    """Handle web server errors."""
    log.critical(
        "{method} {path} {detail}", method=request.method, path=request.url.path, detail=exc.detail
    )
    return Response(
        {"output": exc.detail, "level": "danger", "keywords": []},
        status_code=exc.status_code,
    )


def app_handler(request: Request, exc: BaseException) -> Response:
    """Handle application errors."""
    log.critical(
        "{method} {path} {detail}", method=request.method, path=request.url.path, detail=exc.message
    )
    return Response(
        {"output": exc.message, "level": exc.level, "keywords": exc.keywords},
        status_code=exc.status_code,
    )


def validation_handler(request: Request, exc: BaseException) -> Response:
    """Handle Pydantic validation errors raised by FastAPI."""
    error = exc.errors()[0]
    log.critical(
        "{method} {path} {detail}",
        method=request.method,
        path=request.url.path,
        detail=error["msg"],
    )
    return Response(
        {"output": error["msg"], "level": "error", "keywords": error["loc"]},
        status_code=422,
    )