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

latch_postgres/postgres.py

@@ -500,47 +500,48 @@ def with_conn_retry(
                 while True:
                     try:
                         async with pool.connection() as conn:
-                            assert isinstance(conn, LatchAsyncConnection)
+                            with tracer.start_as_current_span("connection acquired"):
+                                assert isinstance(conn, LatchAsyncConnection)
 
-                            backoff = CABackoff(
-                                db_config.tx_retries.delay_quant,
-                                db_config.tx_retries.max_wait_time,
-                            )
-                            while True:
-                                try:
-                                    res = await f(conn, *args, **kwargs)
-                                    # Commit here so we can retry if it fails
-                                    # Otherwise the context manager will commit and fail
-                                    await conn.commit()
-                                    return res
-                                except (
-                                    # Class 40 - Transaction Rollback
-                                    SerializationFailure,
-                                    DeadlockDetected,
-                                ) as e:
-                                    # Retry with the same connection
-                                    delay = backoff.retry()
-
-                                    if delay is None:
-                                        raise e
-
-                                    s.add_event(
-                                        "transaction retry",
-                                        {
-                                            "db.retry.count": retries,
-                                            "db.retry.accum_retry_time": str(
-                                                backoff.acc_wait_time
+                                backoff = CABackoff(
+                                    db_config.tx_retries.delay_quant,
+                                    db_config.tx_retries.max_wait_time,
+                                )
+                                while True:
+                                    try:
+                                        res = await f(conn, *args, **kwargs)
+                                        # Commit here so we can retry if it fails
+                                        # Otherwise the context manager will commit and fail
+                                        await conn.commit()
+                                        return res
+                                    except (
+                                        # Class 40 - Transaction Rollback
+                                        SerializationFailure,
+                                        DeadlockDetected,
+                                    ) as e:
+                                        # Retry with the same connection
+                                        delay = backoff.retry()
+
+                                        if delay is None:
+                                            raise e
+
+                                        s.add_event(
+                                            "transaction retry",
+                                            {
+                                                "db.retry.count": retries,
+                                                "db.retry.accum_retry_time": str(
+                                                    backoff.acc_wait_time
+                                                ),
+                                                "db.retry.delay": str(delay),
+                                            }
+                                            | dict_to_attrs(
+                                                pg_error_to_dict(e, short=True),
+                                                "db.retry.reason",
                                             ),
-                                            "db.retry.delay": str(delay),
-                                        }
-                                        | dict_to_attrs(
-                                            pg_error_to_dict(e, short=True),
-                                            "db.retry.reason",
-                                        ),
-                                    )
+                                        )
 
-                                    await conn.rollback()
-                                    await asyncio.sleep(delay)
+                                        await conn.rollback()
+                                        await asyncio.sleep(delay)
                     except (SerializationFailure, DeadlockDetected):
                         # todo(maximsmol): should be unnecessary if the list below is precise enough
                         raise

{latch_postgres-0.1.17.dist-info → latch_postgres-0.1.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: latch-postgres
-Version: 0.1.17
+Version: 0.1.19
 Summary: Postges wrapper for latch python backend services
 License: CC0 1.0
 Author: Max Smolin
@@ -9,8 +9,10 @@ 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.3,<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)

latch_postgres-0.1.19.dist-info/RECORD

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

{latch_postgres-0.1.17.dist-info → latch_postgres-0.1.19.dist-info}/WHEEL

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

latch_postgres-0.1.17.dist-info/RECORD

@@ -1,8 +0,0 @@
-latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/postgres.py,sha256=1eqzuFv7NCYOru6-mHcJNsyjASwPWL_9qHH86lEXb4g,24611
-latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
-latch_postgres-0.1.17.dist-info/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
-latch_postgres-0.1.17.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-latch_postgres-0.1.17.dist-info/METADATA,sha256=3cpgB1eCGeZf3WXac7-dAdJU0Zpjf1YmELO7pDWatpg,766
-latch_postgres-0.1.17.dist-info/RECORD,,