latch-postgres 0.1.12__tar.gz → 0.1.14__tar.gz

{latch_postgres-0.1.12 → latch_postgres-0.1.14}/PKG-INFO

@@ -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.12 → latch_postgres-0.1.14}/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 → latch_postgres-0.1.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "latch-postgres"
-version = "0.1.12"
+version = "0.1.14"
 description = "Postges wrapper for latch python backend services"
 authors = ["Max Smolin <max@latch.bio>"]
 license = "CC0 1.0"

{latch_postgres-0.1.12 → latch_postgres-0.1.14}/setup.py

@@ -18,7 +18,7 @@ install_requires = \
 
 setup_kwargs = {
     'name': 'latch-postgres',
-    'version': '0.1.12',
+    'version': '0.1.14',
     'description': 'Postges wrapper for latch python backend services',
     'long_description': '# python-postgres\n',
     'author': 'Max Smolin',