add asyncpg

uc-cdis · May 24, 2024 · b28594f · b28594f
1 parent 253cfe4
commit b28594f
Show file tree

Hide file tree

Showing 3 changed files with 71 additions and 2 deletions.
diff --git a/bin/migrate_with_asyncio.py b/bin/migrate_with_asyncio.py
@@ -66,7 +66,7 @@ def __init__(self, conf_data=None):
         self.n_workers = self.thread_pool_size + self.concurrency
 
         self.engine = create_async_engine(
-            f"postgresql+psycopg2://{usr}:{psw}@{pghost}:{pgport}/{db}", echo=True
+            f"postgresql+asyncpg://{usr}:{psw}@{pghost}:{pgport}/{db}", echo=True
         )
         self.async_session = sessionmaker(
             self.engine, expire_on_commit=False, class_=AsyncSession

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,6 +26,7 @@ psycopg2 = "^2.7"
 sqlalchemy = "^1.4.0"
 sqlalchemy-utils = "^0.37.3"
 PyYAML = "^5.4"
+asyncpg = "^0.29.0"
 
 
 [tool.poetry.dev-dependencies]