From 654dc2c70d71d13b6b7c42bba2e39c85566f6d47 Mon Sep 17 00:00:00 2001 From: phil Date: Tue, 9 Apr 2024 16:16:45 +0530 Subject: [PATCH] Add sync_engine (useful for CLI/Pandas) --- src/gisaf/database.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/gisaf/database.py b/src/gisaf/database.py index 1976b7e..3547b2c 100644 --- a/src/gisaf/database.py +++ b/src/gisaf/database.py @@ -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