Add sync_engine (useful for CLI/Pandas)
This commit is contained in:
parent
d2ae5e4d7b
commit
654dc2c70d
1 changed files with 8 additions and 0 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue