From 85712548d82feaa88823c7e1dd1a0ff9e7d9c6c3 Mon Sep 17 00:00:00 2001 From: matv864 Date: Fri, 14 Mar 2025 04:23:47 +1000 Subject: [PATCH] small makeup --- alembic/env.py | 2 +- src/database_adapter/__init__.py | 3 +++ src/database_adapter/engine.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) create mode 100644 src/database_adapter/__init__.py 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)