15 lines
404 B
Python
15 lines
404 B
Python
from contextlib import asynccontextmanager
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from app.db import async_session
|
|
|
|
|
|
async def get_session() -> AsyncSession:
|
|
async with async_session() as session, session.begin():
|
|
yield session
|
|
|
|
|
|
@asynccontextmanager
|
|
async def get_session_context() -> AsyncSession:
|
|
async with async_session() as session, session.begin():
|
|
yield session
|