latch-postgres 0.1.21__py3-none-any.whl → 0.2.0__py3-none-any.whl

latch_postgres/postgres.py

@@ -616,9 +616,12 @@ async def reset_conn(
 # fixme(maximsmol): use autocommit transactions
 def get_pool(
     config: PostgresConnectionConfig,
+    *,
     application_name: str,
     read_only: bool = True,
     isolation_level: IsolationLevel = IsolationLevel.SERIALIZABLE,
+    configure: Callable[[AsyncConnection[object]], Awaitable[None]] | None = None,
+    reset: Callable[[AsyncConnection[object]], Awaitable[None]] | None = None,
 ) -> TracedAsyncConnectionPool:
     conn_str = make_conninfo(
         host=config.host,
@@ -628,17 +631,24 @@ def get_pool(
         password=config.password,
         application_name=application_name,
     )
+
+    async def configure_impl(x: AsyncConnection[object]) -> None:
+        await reset_conn(x, read_only=read_only, isolation_level=isolation_level)
+        if configure is not None:
+            await configure(x)
+
+    async def reset_impl(x: AsyncConnection[object]) -> None:
+        await reset_conn(x, read_only=read_only, isolation_level=isolation_level)
+        if reset is not None:
+            await reset(x)
+
     return TracedAsyncConnectionPool(
         conn_str,
         min_size=1,
         max_size=config.pool_size,
         timeout=timedelta(seconds=5) / timedelta(seconds=1),
         open=False,
-        configure=functools.partial(
-            reset_conn, read_only=read_only, isolation_level=isolation_level
-        ),
-        reset=functools.partial(
-            reset_conn, read_only=read_only, isolation_level=isolation_level
-        ),
+        configure=configure_impl,
+        reset=reset_impl,
         connection_class=LatchAsyncConnection,
     )

{latch_postgres-0.1.21.dist-info → latch_postgres-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latch-postgres
-Version: 0.1.21
+Version: 0.2.0
 Summary: Postges wrapper for latch python backend services
 Author-email: maximsmol <max@latch.bio>
 License: CC0-1.0

latch_postgres-0.2.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+latch_postgres/postgres.py,sha256=FKYt0C66xDSOEc63x7z1HuOAylgPcZWGPDNlD2Htbz8,24922
+latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
+latch_postgres-0.2.0.dist-info/METADATA,sha256=GFxp7RS1vAyE60rlXgxY9KV0ETD_titcMqK57a-jzWM,606
+latch_postgres-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+latch_postgres-0.2.0.dist-info/licenses/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
+latch_postgres-0.2.0.dist-info/RECORD,,

latch_postgres-0.1.21.dist-info/RECORD

@@ -1,8 +0,0 @@
-latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/postgres.py,sha256=-TpNtSGe-EKy-hKOhajkriqbYkKvYV3LegDqx3towJ8,24522
-latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
-latch_postgres-0.1.21.dist-info/METADATA,sha256=12J7_YVLtpW3MbfvEYAxbP2VAK36ZSjHD7a5vTli-jY,607
-latch_postgres-0.1.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-latch_postgres-0.1.21.dist-info/licenses/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
-latch_postgres-0.1.21.dist-info/RECORD,,