Add sync_engine (useful for CLI/Pandas)

This commit is contained in:
phil 2024-04-09 16:16:45 +05:30
parent d2ae5e4d7b
commit 654dc2c70d

View file

@ -3,6 +3,7 @@ from typing import Annotated, Literal, Any
from collections.abc import AsyncGenerator
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy import create_engine
from sqlalchemy.orm import joinedload, QueryableAttribute
from sqlalchemy.sql.selectable import Select
from sqlmodel import SQLModel, select
@ -21,6 +22,13 @@ engine = create_async_engine(
max_overflow=conf.db.max_overflow,
)
sync_engine = create_engine(
conf.db.get_pg_url(),
echo=conf.db.echo,
pool_size=conf.db.pool_size,
max_overflow=conf.db.max_overflow,
)
async def get_db_session() -> AsyncGenerator[AsyncSession]:
async with AsyncSession(engine) as session:
yield session