| 1234567891011121314151617 |
- from collections.abc import AsyncGenerator
- from sqlalchemy.ext.asyncio import AsyncSession
- from app.core.database import AsyncSessionLocal
- async def get_db() -> AsyncGenerator[AsyncSession, None]:
- async with AsyncSessionLocal() as session:
- try:
- yield session
- except Exception:
- await session.rollback()
- raise
- finally:
- await session.close()
|
