tracktolib 0.62.0__tar.gz → 0.62.1__tar.gz

{tracktolib-0.62.0 → tracktolib-0.62.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracktolib
-Version: 0.62.0
+Version: 0.62.1
 Summary: Utility library for python
 Keywords: utility
 Author-email: julien.brayere@tracktor.fr

{tracktolib-0.62.0 → tracktolib-0.62.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tracktolib"
-version = "0.62.0"
+version = "0.62.1"
 authors = [
     { email = "julien.brayere@tracktor.fr" }
 ]
@@ -91,7 +91,7 @@ pythonPlatform = "Linux"
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.62.0"
+version = "0.62.1"
 tag_format = "$version"
 version_files = [
     "pyproject.toml:version"

{tracktolib-0.62.0 → tracktolib-0.62.1}/tracktolib/pg/query.py

@@ -225,10 +225,14 @@ def get_update_fields(
 
 @dataclass
 class PGUpdateQuery(PGQuery):
-    """Value to start the arguments from:
-    For instance, with a value of 10, the first argument will be $11
+    """
+    Postgresql UPDATE query generator
     """
 
+    """
+    Value to start the arguments from:
+    For instance, with a value of 10, the first argument will be $11
+    """
     start_from: int | None = None
     """Keys to use for the WHERE clause. Theses fields will not be updated"""
     where_keys: list[str] | None = None
@@ -239,6 +243,8 @@ class PGUpdateQuery(PGQuery):
     return_keys: bool = False
     """Values to update using merge (like {}::jsonb || {}::jsonb)"""
     merge_keys: list[str] | None = None
+    """If True, the query is for many items and values will be a list of tuples"""
+    is_many: bool = False
 
     _update_fields: str | None = field(init=False, default=None)
     _values: list | None = field(init=False, default=None)
@@ -264,7 +270,7 @@ class PGUpdateQuery(PGQuery):
     def values(self):
         if not self._values:
             raise ValueError("No values found")
-        if len(self.items) == 1:
+        if len(self.items) == 1 and not self.is_many:
             return self._values
         _where_keys = self.where_keys or []
         _keys_not_where = [k for k in self.keys if k not in _where_keys]
@@ -516,7 +522,13 @@ async def update_many(
     query_callback: QueryCallback[PGUpdateQuery] | None = None,
 ):
     query = PGUpdateQuery(
-        table=table, items=items, start_from=start_from, where_keys=keys, where=where, merge_keys=merge_keys
+        table=table,
+        items=items,
+        start_from=start_from,
+        where_keys=keys,
+        where=where,
+        merge_keys=merge_keys,
+        is_many=True,
     )
     if query_callback is not None:
         query_callback(query)