diff --git a/alembic/env.py b/alembic/env.py index 10e3f91..1a0e0a3 100644 --- a/alembic/env.py +++ b/alembic/env.py @@ -7,7 +7,7 @@ from sqlalchemy.ext.asyncio import async_engine_from_config from alembic import context -from src.database_adapter.model import Base +from src.database_adapter import Base from src.settings import settings # this is the Alembic Config object, which provides diff --git a/src/database_adapter/__init__.py b/src/database_adapter/__init__.py new file mode 100644 index 0000000..666320d --- /dev/null +++ b/src/database_adapter/__init__.py @@ -0,0 +1,3 @@ +from .engine import async_session_maker as async_session_maker +from .model import Base as Base +from .model import Log as Log diff --git a/src/database_adapter/engine.py b/src/database_adapter/engine.py index cbcc622..d2c6bea 100644 --- a/src/database_adapter/engine.py +++ b/src/database_adapter/engine.py @@ -3,4 +3,4 @@ from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine from src.settings import settings engine = create_async_engine(settings.postgres_url) -async_session_maker_db1 = async_sessionmaker(engine, expire_on_commit=False) +async_session_maker = async_sessionmaker(engine, expire_on_commit=False)