from contextlib import asynccontextmanager from fastapi import FastAPI from app.database import connect, disconnect from app.routers import authors, books @asynccontextmanager async def lifespan(app: FastAPI): await connect() yield await disconnect() app = FastAPI(title="Database API", lifespan=lifespan) for router in [authors.router, books.router]: app.include_router(router)