Implementation Engineer
80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
"""FastAPI application assembly with CORS, routers, and APScheduler."""
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
|
|
from app.config import get_settings
|
|
from app.database import engine, init_db
|
|
from app.cache import cache
|
|
from app.routers import equipment, health, admin, rental_requests, contact
|
|
from app.services.sync_service import SyncService
|
|
from app.database import async_session
|
|
|
|
logger = logging.getLogger(__name__)
|
|
settings = get_settings()
|
|
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
|
|
async def run_equipment_sync() -> None:
|
|
"""Scheduled job: run equipment sync every 6 hours."""
|
|
logger.info("Starting scheduled equipment sync")
|
|
async with async_session() as db:
|
|
sync_service = SyncService(db)
|
|
result = await sync_service.run_sync()
|
|
logger.info("Scheduled sync complete: %s", result)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Application lifespan: init DB, start scheduler, connect cache."""
|
|
# Create tables (for dev/testing; production uses Alembic)
|
|
await init_db()
|
|
|
|
# Connect Redis cache
|
|
await cache.connect()
|
|
|
|
# Start APScheduler
|
|
scheduler.add_job(
|
|
run_equipment_sync,
|
|
trigger="cron",
|
|
hour="*/6",
|
|
id="equipment_sync",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.start()
|
|
logger.info("APScheduler started with equipment_sync job (every 6h)")
|
|
|
|
yield
|
|
|
|
# Shutdown
|
|
scheduler.shutdown(wait=False)
|
|
await cache.disconnect()
|
|
await engine.dispose()
|
|
|
|
|
|
app = FastAPI(
|
|
title="HMS Licht & Ton API",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# CORS
|
|
origins = [o.strip() for o in settings.cors_origins.split(",") if o.strip()]
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=origins,
|
|
allow_methods=["GET", "POST"],
|
|
allow_headers=["*"],
|
|
allow_credentials=True,
|
|
)
|
|
|
|
# Routers
|
|
app.include_router(equipment.router)
|
|
app.include_router(health.router)
|
|
app.include_router(admin.router)
|
|
app.include_router(rental_requests.router)
|
|
app.include_router(contact.router)
|