main.py

from http import HTTPStatus

import sentry_sdk
from fastapi import FastAPI
from fastapi.exceptions import RequestValidationError
from fastapi.middleware.cors import CORSMiddleware

from domains.health.delivery import HealthRestHandler
from modules.config.environment import get_environment_variables
from modules.datadog.datadog import init_datadog_tracer
from modules.logger.logger import Logger
from modules.middleware.exception import ExceptionHandlerMiddleware
from modules.trace.middleware import RequestIDMiddleware
from schemas.custom_error import CustomError

logger = Logger(name=__name__)


app_config = get_environment_variables()
sentry_sdk.init(
    dsn=app_config.SENTRY_DSN,
    enable_tracing=True,
    environment=app_config.SENTRY_ENV,
)
app = FastAPI(
    title=app_config.APP_NAME,
    version=app_config.API_VERSION,
)


@app.exception_handler(RequestValidationError)
async def custom_validation_exception_handler(request, exc: RequestValidationError):
    message = "invalid request"
    validation_errors = exc.errors()
    for validation_error in validation_errors:
        msg = validation_error.get("msg")
        loc = ":".join(map(str, validation_error.get("loc")))
        if msg and loc:
            message = f"{msg} at {loc}"
            break
    return CustomError(
        message=message,
        code=HTTPStatus.BAD_REQUEST,
        http_code=HTTPStatus.BAD_REQUEST,
        error_code="INVALID_REQUEST_ERR",
    ).to_json_response()


app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.add_middleware(RequestIDMiddleware)
app.add_middleware(ExceptionHandlerMiddleware)

app.include_router(HealthRestHandler)

# setup datadog
init_datadog_tracer(app)