latch-postgres 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

latch_postgres/postgres.py

@@ -55,7 +55,7 @@ from psycopg.errors import (
     TooManyConnections,
     UndefinedFile,
 )
-from psycopg.rows import AsyncRowFactory, Row, kwargs_row
+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
 from psycopg_pool import AsyncConnectionPool
@@ -184,6 +184,15 @@ class LatchAsyncConnection(AsyncConnection[Row]):
 
             yield curs
 
+    @asynccontextmanager
+    async def _query_no_validate(
+        self, query: sql.SQL, **kwargs: Any
+    ) -> AsyncGenerator[AsyncCursor[Any], None]:
+        async with self.cursor(dict_row) as curs:
+            await curs.execute(query, params=kwargs)
+
+            yield curs
+
     async def queryn(self, model: type[T], query: sql.SQL, **kwargs: Any) -> list[T]:
         async with self._query(model, query, **kwargs) as curs:
             if curs.description is None:
@@ -219,7 +228,7 @@ class LatchAsyncConnection(AsyncConnection[Row]):
         await self.queryn(type(None), query, **kwargs)
 
     async def query_unknown(self, query: sql.SQL, **kwargs: Any) -> Any:
-        async with self._query(JsonObject, query, **kwargs) as curs:
+        async with self._query_no_validate(query, **kwargs) as curs:
             if curs.description is None:
                 return None
 
@@ -529,6 +538,7 @@ def with_conn_retry(
                                             "db.retry.reason",
                                         ),
                                     )
+                                    await conn.rollback()
                                     await asyncio.sleep(delay)
                     except (SerializationFailure, DeadlockDetected):
                         # todo(maximsmol): should be unnecessary if the list below is precise enough

{latch_postgres-0.1.12.dist-info → latch_postgres-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: latch-postgres
-Version: 0.1.12
+Version: 0.1.14
 Summary: Postges wrapper for latch python backend services
 License: CC0 1.0
 Author: Max Smolin

latch_postgres-0.1.14.dist-info/RECORD

@@ -0,0 +1,8 @@
+latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+latch_postgres/postgres.py,sha256=ucmGrKx4vlGhg_1XScYbeahBjfsIUvt9xGDfFwITpCY,24379
+latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
+latch_postgres-0.1.14.dist-info/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
+latch_postgres-0.1.14.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+latch_postgres-0.1.14.dist-info/METADATA,sha256=qbQ6s5fnRZ0H15rX8ITule8UqTKAo-_fTVcK5yXP6Qc,766
+latch_postgres-0.1.14.dist-info/RECORD,,

latch_postgres-0.1.12.dist-info/RECORD

@@ -1,8 +0,0 @@
-latch_postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/postgres.py,sha256=1RVUNNqvw5CFRM6bD47DEThKGdXSkCwxNrjlPAAzZaw,24031
-latch_postgres/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_postgres/retries.py,sha256=r5yH00fd_6EJNYFXAq3dAljJBphScpoeST-_oMvyj8Q,1308
-latch_postgres-0.1.12.dist-info/LICENSE,sha256=wh3JZu6ITG7ceN-1g404ekupj5JXAw4pnQ_Qr4sJfew,7052
-latch_postgres-0.1.12.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-latch_postgres-0.1.12.dist-info/METADATA,sha256=mRRTGlNDI9N5ylZFoI7K__YtoAYVTgx48Le57E83hzs,766
-latch_postgres-0.1.12.dist-info/RECORD,,