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
def create_session(
def get_session(
url: str, engine_echo: bool = True
) -> async_sessionmaker[AsyncSession]:
global engine

View file

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