checktheroads-hyperglass/hyperglass/api/error_handlers.py

"""API Error Handlers."""

# Third Party
from starlette.responses import JSONResponse

# Project
from hyperglass.state import use_state


async def default_handler(request, exc):
    """Handle uncaught errors."""
    state = use_state()
    return JSONResponse(
        {"output": state.params.messages.general, "level": "danger", "keywords": []},
        status_code=500,
    )


async def http_handler(request, exc):
    """Handle web server errors."""

    return JSONResponse(
        {"output": exc.detail, "level": "danger", "keywords": []}, status_code=exc.status_code,
    )


async def app_handler(request, exc):
    """Handle application errors."""
    return JSONResponse(
        {"output": exc.message, "level": exc.level, "keywords": exc.keywords},
        status_code=exc.status_code,
    )


async def validation_handler(request, exc):
    """Handle Pydantic validation errors raised by FastAPI."""
    error = exc.errors()[0]
    return JSONResponse(
        {"output": error["msg"], "level": "error", "keywords": error["loc"]}, status_code=422,
    )