diff --git a/src/database/__init__.py b/src/database/__init__.py index 4c8fbb8..0d430c9 100644 --- a/src/database/__init__.py +++ b/src/database/__init__.py @@ -1,4 +1,4 @@ -__all__ = ["create_session", "create_db", "drop_db"] +__all__ = ["get_session", "create_db", "drop_db"] -from .session import create_db, create_session, drop_db +from .session import create_db, drop_db, get_session diff --git a/src/database/session.py b/src/database/session.py index 4fd389c..6cd6c06 100644 --- a/src/database/session.py +++ b/src/database/session.py @@ -11,7 +11,7 @@ from .models import Base engine: AsyncEngine -def create_session( +def get_session( url: str, engine_echo: bool = True ) -> async_sessionmaker[AsyncSession]: global engine diff --git a/src/main.py b/src/main.py index 17de15b..a786e00 100644 --- a/src/main.py +++ b/src/main.py @@ -2,7 +2,7 @@ import asyncio from bot import start_bot from config import configure_logger, settings -from database import create_db, create_session +from database import create_db, get_session from redis_client import get_redis_client @@ -17,7 +17,7 @@ def main() -> None: listen_logging=settings.listen_logging, ) - session = create_session( + database_session = get_session( url=settings.database_url, engine_echo=True ) @@ -33,7 +33,7 @@ def main() -> None: start_bot( bot_token=settings.bot_token, redis_client=redis_client, - database_session=session, + database_session=database_session, ) )