config di lifespan
parent
c809f14fdc
commit
7eaa082b41
|
@ -2,8 +2,13 @@ from collections.abc import AsyncGenerator
|
|||
from contextlib import asynccontextmanager
|
||||
|
||||
from fastapi import FastAPI
|
||||
from sqlalchemy.ext.asyncio import AsyncEngine
|
||||
|
||||
from api.app_builder.dependencies import init_dependencies
|
||||
from api.infrastructure.dependencies.adapters import create_engine
|
||||
from api.infrastructure.dependencies.configs import app_settings
|
||||
from api.infrastructure.persistence.models import Base
|
||||
from api.infrastructure.settings import Settings
|
||||
|
||||
from .routers import init_routers
|
||||
|
||||
|
@ -11,6 +16,12 @@ from .routers import init_routers
|
|||
@asynccontextmanager
|
||||
async def lifespan(app: FastAPI) -> AsyncGenerator:
|
||||
print("init lifespan")
|
||||
app.dependency_overrides[Settings] = app_settings
|
||||
app.dependency_overrides[AsyncEngine] = create_engine
|
||||
engine = app.dependency_overrides[AsyncEngine](app.dependency_overrides[Settings]())
|
||||
|
||||
async with engine.begin() as conn:
|
||||
await conn.run_sync(Base.metadata.create_all)
|
||||
yield
|
||||
|
||||
|
||||
|
|
|
@ -0,0 +1,7 @@
|
|||
from .base import Base
|
||||
from .user import UserModel
|
||||
|
||||
__all__ = (
|
||||
"Base",
|
||||
"UserModel",
|
||||
)
|
Loading…
Reference in New Issue