Coverage for app / core / database.py: 75%

1from typing import AsyncGenerator 

2 

3from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine 

4 

5from app.core.config import settings 

6 

7engine = create_async_engine(settings.db_url) 

8 

9AsyncSessionLocal = async_sessionmaker( 

10 bind=engine, 

11 class_=AsyncSession, 

12 autoflush=False, 

13 autocommit=False, 

14 expire_on_commit=False, 

15) 

16 

17 

18async def get_db() -> AsyncGenerator[AsyncSession, None]: 

19 async with AsyncSessionLocal() as session: 

20 yield session