40 lines
941 B
Python
40 lines
941 B
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.pomodoro.watcher import pomodoro_watcher_loop
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_: FastAPI):
|
|
init_db()
|
|
watcher_task = asyncio.create_task(pomodoro_watcher_loop())
|
|
yield
|
|
watcher_task.cancel()
|
|
with suppress(asyncio.CancelledError):
|
|
await watcher_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()
|