from contextlib import asynccontextmanager
from fastapi import FastAPI

from fastapi_demo.app.infrastructure.db.session import dispose_engine
from fastapi_demo.app.api.schemas.api_router import api_router


@asynccontextmanager
async def lifespan(app: FastAPI):
    yield
    dispose_engine()


app = FastAPI(title="FASTAPI_DEMO", version="0.1.0", lifespan=lifespan)
app.include_router(api_router)