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
|
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
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue