tracktolib 0.61.1__tar.gz → 0.61.2__tar.gz

{tracktolib-0.61.1 → tracktolib-0.61.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracktolib
-Version: 0.61.1
+Version: 0.61.2
 Summary: Utility library for python
 License: MIT
 License-File: LICENSE

{tracktolib-0.61.1 → tracktolib-0.61.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tracktolib"
-version = "0.61.1"
+version = "0.61.2"
 description = "Utility library for python"
 authors = ["Julien Brayere <julien.brayere@tracktor.fr>"]
 license = "MIT"
@@ -83,7 +83,7 @@ pythonPlatform = "Linux"
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.61.1"
+version = "0.61.2"
 tag_format = "$version"
 version_files = [
     "pyproject.toml:version"

{tracktolib-0.61.1 → tracktolib-0.61.2}/tracktolib/pg/query.py

@@ -105,8 +105,8 @@ class PGQuery(Generic[K, V]):
     def columns(self):
         return [f'"{x}"' for x in self.keys] if self.quote_columns else self.keys
 
-    def iter_values(self) -> Iterator[tuple]:
-        _keys = self.keys
+    def iter_values(self, keys: list[K] | None = None) -> Iterator[tuple]:
+        _keys = keys if keys is not None else self.keys
         for _item in self.items:
             yield tuple(_item[k] for k in _keys)
 
@@ -264,7 +264,12 @@ class PGUpdateQuery(PGQuery):
     def values(self):
         if not self._values:
             raise ValueError("No values found")
-        return self._values if len(self.items) == 1 else super().values
+        if len(self.items) == 1:
+            return self._values
+        _where_keys = self.where_keys or []
+        _keys_not_where = [k for k in self.keys if k not in _where_keys]
+        values = list(self.iter_values(keys=_keys_not_where + _where_keys))
+        return values
 
     def _get_where_query(self) -> str:
         if self.where: