counterweight
53aa54d6c9
Issue #6: Multiple routers per file made main.py verbose. Changes: - Add 'routers' list export to booking.py and invites.py - Update main.py to iterate over router lists for multi-router modules - Keep explicit registration for single-router modules - Cleaner separation between simple and complex route modules
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
"""FastAPI application entry point."""
|
|
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from database import Base, engine
|
|
from jobs import close_job_pool
|
|
from routes import audit as audit_routes
|
|
from routes import auth as auth_routes
|
|
from routes import availability as availability_routes
|
|
from routes import booking as booking_routes
|
|
from routes import counter as counter_routes
|
|
from routes import invites as invites_routes
|
|
from routes import meta as meta_routes
|
|
from routes import profile as profile_routes
|
|
from routes import sum as sum_routes
|
|
from validate_constants import validate_shared_constants
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Create database tables on startup and validate constants."""
|
|
# Validate shared constants match backend definitions
|
|
validate_shared_constants()
|
|
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
yield
|
|
# Cleanup on shutdown
|
|
await close_job_pool()
|
|
|
|
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["http://localhost:3000"],
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
allow_credentials=True,
|
|
)
|
|
|
|
# Include routers - modules with single router
|
|
app.include_router(auth_routes.router)
|
|
app.include_router(sum_routes.router)
|
|
app.include_router(counter_routes.router)
|
|
app.include_router(audit_routes.router)
|
|
app.include_router(profile_routes.router)
|
|
app.include_router(availability_routes.router)
|
|
app.include_router(meta_routes.router)
|
|
|
|
# Include routers - modules with multiple routers
|
|
for r in invites_routes.routers:
|
|
app.include_router(r)
|
|
for r in booking_routes.routers:
|
|
app.include_router(r)
|