77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from sqlalchemy import func, select
|
|
|
|
from app.config import settings
|
|
from app.database import async_session, engine
|
|
from app.models import Base, Program
|
|
from app.user_models import User
|
|
from app.api import programs, events, tiers, auth
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
# Create tables on startup
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
|
|
# Auto-seed if database is empty
|
|
async with async_session() as session:
|
|
result = await session.execute(select(func.count(Program.id)))
|
|
count = result.scalar()
|
|
if count == 0:
|
|
print(" → Database is empty — running seed...")
|
|
from seed import seed as run_seed
|
|
await run_seed()
|
|
|
|
# Bootstrap admin user if configured
|
|
if settings.ADMIN_USERNAME and settings.ADMIN_PASSWORD:
|
|
async with async_session() as session:
|
|
result = await session.execute(
|
|
select(User).where(
|
|
User.email == f"{settings.ADMIN_USERNAME}@local",
|
|
User.auth_provider == "local",
|
|
)
|
|
)
|
|
if result.scalar_one_or_none() is None:
|
|
admin_user = User(
|
|
email=f"{settings.ADMIN_USERNAME}@local",
|
|
display_name=settings.ADMIN_USERNAME,
|
|
auth_provider="local",
|
|
is_admin=True,
|
|
)
|
|
session.add(admin_user)
|
|
await session.commit()
|
|
print(f" → Bootstrap admin created: {settings.ADMIN_USERNAME}")
|
|
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="KMountain Flower Radio API",
|
|
description="Backend API for the KMountain Flower Radio Station website.",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.CORS_ORIGINS,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Register routers
|
|
app.include_router(auth.router, prefix="/api/auth", tags=["auth"])
|
|
app.include_router(programs.router, prefix="/api/programs", tags=["programs"])
|
|
app.include_router(events.router, prefix="/api/events", tags=["events"])
|
|
app.include_router(tiers.router, prefix="/api/tiers", tags=["tiers"])
|
|
|
|
# Dev-only admin router (disabled in production)
|
|
if settings.ENV != "production":
|
|
from app.api import admin
|
|
app.include_router(admin.router, prefix="/api/admin", tags=["admin"])
|