generated from Luis/nextjs-python-web-template
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
# Imports
|
|
import logging
|
|
|
|
# Import FastAPI components and SQLAlchemy related modules
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.routing import APIRoute
|
|
from fastapi.staticfiles import StaticFiles
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
|
|
# Import configs
|
|
from ..config import cors_ports, cors_url
|
|
|
|
# Import custom modules and classes
|
|
from ..errors import ClanError
|
|
from . import sql_models
|
|
from .assets import asset_path
|
|
from .error_handlers import clan_error_handler, sql_error_handler
|
|
from .routers import endpoints, health, root
|
|
from .sql_db import engine
|
|
from .tags import tags_metadata
|
|
|
|
cors_whitelist = []
|
|
for u in cors_url:
|
|
for p in cors_ports:
|
|
cors_whitelist.append(f"{u}:{p}")
|
|
|
|
# Logging setup
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
# Function to set up and configure the FastAPI application
|
|
def setup_app() -> FastAPI:
|
|
# Uncomment the following line to drop existing tables during startup (if needed)
|
|
# sql_models.Base.metadata.drop_all(engine)
|
|
|
|
# Create tables in the database using SQLAlchemy
|
|
sql_models.Base.metadata.create_all(bind=engine)
|
|
|
|
# Initialize FastAPI application with lifespan management
|
|
app = FastAPI(swagger_ui_parameters={"tryItOutEnabled": True})
|
|
|
|
# Configure CORS middleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=cors_whitelist,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include routers for various endpoints and components
|
|
app.include_router(health.router)
|
|
app.include_router(endpoints.router)
|
|
|
|
# Needs to be last in registration due to wildcard route
|
|
app.include_router(root.router)
|
|
|
|
# Add custom exception handlers
|
|
app.add_exception_handler(ClanError, clan_error_handler) # type: ignore
|
|
app.add_exception_handler(SQLAlchemyError, sql_error_handler) # type: ignore
|
|
|
|
# Mount the "static" route for serving static files
|
|
app.mount("/static", StaticFiles(directory=asset_path()), name="static")
|
|
|
|
# Add tag descriptions to the OpenAPI schema
|
|
app.openapi_tags = tags_metadata
|
|
|
|
# Assign operation IDs to API routes
|
|
for route in app.routes:
|
|
if isinstance(route, APIRoute):
|
|
route.operation_id = route.name # in this case, 'read_items'
|
|
log.debug(f"Registered route: {route}")
|
|
|
|
# Log registered exception handlers
|
|
for i in app.exception_handlers.items():
|
|
log.debug(f"Registered exception handler: {i}")
|
|
|
|
return app
|
|
|
|
|
|
# Create an instance of the FastAPI application
|
|
app = setup_app()
|