generated from geekiot/python-template
Refactor [database]: update function for connect to db name
This commit is contained in:
parent
619b2be6ea
commit
43a2002889
3 changed files with 6 additions and 6 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
)
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue