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

latch_postgres/postgres.py

@@ -1,21 +1,12 @@
 import asyncio
 import functools
 import random
+from collections.abc import AsyncGenerator, Awaitable, Callable, Coroutine, Iterable
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import timedelta
 from textwrap import dedent
-from typing import (
-    Any,
-    AsyncGenerator,
-    Awaitable,
-    Callable,
-    Concatenate,
-    Iterable,
-    ParamSpec,
-    TypeVar,
-    cast,
-)
+from typing import Any, Concatenate, ParamSpec, TypeVar, cast
 
 import psycopg.sql as sql
 from latch_config.config import PostgresConnectionConfig
@@ -37,9 +28,6 @@ from psycopg.errors import (
     DeadlockDetected,
     DiskFull,
     DuplicateFile,
-)
-from psycopg.errors import Error as PGError
-from psycopg.errors import (
     IdleSessionTimeout,
     InsufficientResources,
     IoError,
@@ -55,6 +43,7 @@ from psycopg.errors import (
     TooManyConnections,
     UndefinedFile,
 )
+from psycopg.errors import Error as PGError
 from psycopg.rows import AsyncRowFactory, Row, dict_row, kwargs_row
 from psycopg.types.composite import CompositeInfo, register_composite
 from psycopg.types.enum import EnumInfo, register_enum
@@ -63,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__)
@@ -158,15 +149,9 @@ class LatchAsyncConnection(AsyncConnection[Row]):
     trace_attributes: Attributes
 
     def cursor(
-        self,
-        row_factory: AsyncRowFactory[Any],
-        *,
-        binary: bool = True,
+        self, row_factory: AsyncRowFactory[Any], *, binary: bool = True
     ) -> AsyncCursor[Any]:
-        res = super().cursor(
-            row_factory=row_factory,
-            binary=binary,
-        )
+        res = super().cursor(row_factory=row_factory, binary=binary)
         assert isinstance(res, TracedAsyncCursor)
         res.trace_attributes = self.trace_attributes
         return res
@@ -291,18 +276,12 @@ class TracedAsyncConnectionPool(AsyncConnectionPool):
                 )
 
                 async def run_setup(cmd: sql.SQL):
-                    with tracer.start_as_current_span(
-                        "setup command",
-                    ):
+                    with tracer.start_as_current_span("setup command"):
                         await conn.query_opt(dict, cmd)
 
                 async def run_composite_setup():
-                    with tracer.start_as_current_span(
-                        "composite type setup",
-                    ):
-                        with tracer.start_as_current_span(
-                            "register composite types",
-                        ):
+                    with tracer.start_as_current_span("composite type setup"):
+                        with tracer.start_as_current_span("register composite types"):
                             for db_type, f in self.composite_type_map.items():
                                 type_info = await CompositeInfo.fetch(conn, db_type)
                                 if type_info is None:
@@ -312,12 +291,8 @@ class TracedAsyncConnectionPool(AsyncConnectionPool):
                                 register_composite(type_info, conn, f)
 
                 async def run_enum_setup():
-                    with tracer.start_as_current_span(
-                        "enum setup",
-                    ):
-                        with tracer.start_as_current_span(
-                            "fetch type info",
-                        ):
+                    with tracer.start_as_current_span("enum setup"):
+                        with tracer.start_as_current_span("fetch type info"):
                             # query from
                             # https://github.com/psycopg/psycopg/blob/fd1659118e96a48f22b4e67ff17c2cdab8bd0e84/psycopg/psycopg/_typeinfo.py#L148
                             raw_responses = await conn.queryn(
@@ -406,9 +381,7 @@ class TracedAsyncConnectionPool(AsyncConnectionPool):
 
     async def getconn(self, timeout: float | None = None) -> AsyncConnection[object]:
         with tracer.start_as_current_span(
-            "postgres.connect",
-            kind=SpanKind.CLIENT,
-            attributes=self._trace_attributes,
+            "postgres.connect", kind=SpanKind.CLIENT, attributes=self._trace_attributes
         ):
             return await super().getconn(timeout)
 
@@ -443,9 +416,7 @@ def mixin_dict(a: dict[str, object], b: dict[str, object]):
 def pg_error_to_dict(x: PGError, *, short: bool = False):
     diagnostic_obj = {
         "severity": x.diag.severity,
-        "message": {
-            "detail": x.diag.message_detail,
-        },
+        "message": {"detail": x.diag.message_detail},
     }
 
     if not short:
@@ -486,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:
@@ -616,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.21.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: latch-postgres
+Version: 0.1.21
+Summary: Postges wrapper for latch python backend services
+Author-email: maximsmol <max@latch.bio>
+License: CC0-1.0
+License-File: LICENSE
+Requires-Python: >=3.11
+Requires-Dist: latch-config<1.0.0,>=0.1.6
+Requires-Dist: latch-data-validation<1.0.0,>=0.1.10
+Requires-Dist: latch-o11y<2.0.0,>=1.0.0
+Requires-Dist: opentelemetry-api<2.0.0,>=1.15.0
+Requires-Dist: opentelemetry-sdk<2.0.0,>=1.15.0
+Requires-Dist: psycopg[binary,pool]<4.0.0,>=3.1.8
+Requires-Dist: typing-extensions<5.0.0,>=4.4.0
+Description-Content-Type: text/markdown
+
+# python-postgres

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.19.dist-info → latch_postgres-0.1.21.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.0.0
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

latch_postgres-0.1.19.dist-info/METADATA

@@ -1,24 +0,0 @@
-Metadata-Version: 2.3
-Name: latch-postgres
-Version: 0.1.19
-Summary: Postges wrapper for latch python backend services
-License: CC0 1.0
-Author: Max Smolin
-Author-email: max@latch.bio
-Requires-Python: >=3.11,<4.0
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: latch-config (>=0.1.6,<0.2.0)
-Requires-Dist: latch-data-validation (>=0.1.10,<0.2.0)
-Requires-Dist: latch-o11y (>=0.1.4,<0.2.0)
-Requires-Dist: opentelemetry-api (>=1.15.0,<2.0.0)
-Requires-Dist: opentelemetry-sdk (>=1.15.0,<2.0.0)
-Requires-Dist: psycopg[binary,pool] (>=3.1.8,<4.0.0)
-Requires-Dist: typing-extensions (>=4.4.0,<5.0.0)
-Description-Content-Type: text/markdown
-
-# python-postgres
-

latch_postgres-0.1.19.dist-info/RECORD

@@ -1,8 +0,0 @@
-latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/postgres.py,sha256=tGUhVLq5J4tyNrKt4xiRf-RlkqFNrUZ9Jl7WEHLqIPQ,24845
-latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
-latch_postgres-0.1.19.dist-info/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
-latch_postgres-0.1.19.dist-info/METADATA,sha256=fFn_TvOGV1kApPUiG9ow5gMgWQrMl3jVI2ZbNpQD_M0,869
-latch_postgres-0.1.19.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-latch_postgres-0.1.19.dist-info/RECORD,,