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

latch_postgres/postgres.py

@@ -1,7 +1,7 @@
 import asyncio
 import functools
 import random
-from collections.abc import AsyncGenerator, Awaitable, Callable, Iterable
+from collections.abc import AsyncGenerator, Awaitable, Callable, Coroutine, Iterable
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import timedelta
@@ -52,6 +52,8 @@ from typing_extensions import Self
 
 from latch_postgres.retries import CABackoff
 
+YT = TypeVar("YT")
+ST = TypeVar("ST")
 T = TypeVar("T")
 
 tracer = get_tracer(__name__)
@@ -455,10 +457,10 @@ def pg_error_to_dict(x: PGError, *, short: bool = False):
 
 
 def with_conn_retry(
-    f: Callable[Concatenate[LatchAsyncConnection[Any], P], Awaitable[T]],
+    f: Callable[Concatenate[LatchAsyncConnection[Any], P], Coroutine[YT, ST, T]],
     pool: AsyncConnectionPool,
     db_config: PostgresConnectionConfig,
-) -> Callable[P, Awaitable[T]]:
+) -> Callable[P, Coroutine[YT, ST, T]]:
     @functools.wraps(f)
     async def inner(*args: P.args, **kwargs: P.kwargs):
         with tracer.start_as_current_span("database session") as s:
@@ -585,8 +587,8 @@ def with_conn_retry(
 def get_with_conn_retry(
     pool: AsyncConnectionPool, db_config: PostgresConnectionConfig
 ) -> Callable[
-    [Callable[Concatenate[LatchAsyncConnection[Any], P], Awaitable[T]]],
-    Callable[P, Awaitable[T]],
+    [Callable[Concatenate[LatchAsyncConnection[Any], P], Coroutine[YT, ST, T]]],
+    Callable[P, Coroutine[YT, ST, T]],
 ]:
     return functools.partial(with_conn_retry, pool=pool, db_config=db_config)
 

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

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

latch_postgres-0.1.21.dist-info/RECORD

@@ -0,0 +1,8 @@
+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,,

latch_postgres-0.1.20.dist-info/RECORD

@@ -1,8 +0,0 @@
-latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/postgres.py,sha256=EtrOpQN7VO1_dFFY_Gpivrv9p47mS-YGWh_NCYoy2L4,24441
-latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
-latch_postgres-0.1.20.dist-info/METADATA,sha256=B3RxN3paL3WpYQV6XLpThnVmgKZUD5a4Jd0perGWag4,607
-latch_postgres-0.1.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-latch_postgres-0.1.20.dist-info/licenses/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
-latch_postgres-0.1.20.dist-info/RECORD,,