55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import asyncio
|
|
from contextlib import asynccontextmanager, suppress
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from app.api.routes import api_router
|
|
from app.config import get_settings
|
|
from app.db.base import init_db
|
|
from app.fitness.watcher import fitness_watcher_loop
|
|
from app.homelab.watcher import homelab_watcher_loop
|
|
from app.pomodoro.watcher import pomodoro_watcher_loop
|
|
from app.reminders.watcher import reminders_watcher_loop
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_: FastAPI):
|
|
init_db()
|
|
pomodoro_task = asyncio.create_task(pomodoro_watcher_loop())
|
|
fitness_task = asyncio.create_task(fitness_watcher_loop())
|
|
homelab_task = asyncio.create_task(homelab_watcher_loop())
|
|
reminders_task = asyncio.create_task(reminders_watcher_loop())
|
|
yield
|
|
pomodoro_task.cancel()
|
|
fitness_task.cancel()
|
|
homelab_task.cancel()
|
|
reminders_task.cancel()
|
|
with suppress(asyncio.CancelledError):
|
|
await pomodoro_task
|
|
with suppress(asyncio.CancelledError):
|
|
await fitness_task
|
|
with suppress(asyncio.CancelledError):
|
|
await homelab_task
|
|
with suppress(asyncio.CancelledError):
|
|
await reminders_task
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
settings = get_settings()
|
|
app = FastAPI(title="Home AI Assistant", lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.cors_origins_list,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(api_router)
|
|
return app
|
|
|
|
|
|
app = create_app()
|