Refactor [database]: update function for connect to db name

This commit is contained in:
Kirill Samoylenkov 2025-10-30 18:39:49 +05:00
parent 619b2be6ea
commit 43a2002889
3 changed files with 6 additions and 6 deletions

View file

@ -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

View file

@ -11,7 +11,7 @@ from .models import Base
engine: AsyncEngine engine: AsyncEngine
def create_session( def get_session(
url: str, engine_echo: bool = True url: str, engine_echo: bool = True
) -> async_sessionmaker[AsyncSession]: ) -> async_sessionmaker[AsyncSession]:
global engine global engine

View file

@ -2,7 +2,7 @@ import asyncio
from bot import start_bot from bot import start_bot
from config import configure_logger, settings 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 from redis_client import get_redis_client
@ -17,7 +17,7 @@ def main() -> None:
listen_logging=settings.listen_logging, listen_logging=settings.listen_logging,
) )
session = create_session( database_session = get_session(
url=settings.database_url, engine_echo=True url=settings.database_url, engine_echo=True
) )
@ -33,7 +33,7 @@ def main() -> None:
start_bot( start_bot(
bot_token=settings.bot_token, bot_token=settings.bot_token,
redis_client=redis_client, redis_client=redis_client,
database_session=session, database_session=database_session,
) )
) )