upd project structure
This commit is contained in:
19
src/qbot/db/__init__.py
Normal file
19
src/qbot/db/__init__.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from sqlmodel.ext.asyncio.session import AsyncSession
|
||||
from sqlalchemy.ext.asyncio import create_async_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from ..config import config
|
||||
|
||||
# import logging
|
||||
# logger = logging.getLogger('sqlalchemy.engine')
|
||||
# logger.setLevel(logging.DEBUG)
|
||||
|
||||
async_engine = create_async_engine(config.DATABASE_URI)
|
||||
async_session = sessionmaker[AsyncSession](
|
||||
async_engine, class_=AsyncSession, expire_on_commit=False
|
||||
)
|
||||
|
||||
|
||||
async def get_db() -> AsyncSession: # type: ignore
|
||||
async with async_session() as session:
|
||||
yield session
|
||||
Reference in New Issue
Block a user