tracktolib 0.59.0__tar.gz → 0.60.0__tar.gz

{tracktolib-0.59.0 → tracktolib-0.60.0}/PKG-INFO

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

{tracktolib-0.59.0 → tracktolib-0.60.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tracktolib"
-version = "0.59.0"
+version = "0.60.0"
 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.59.0"
+version = "0.60.0"
 tag_format = "$version"
 version_files = [
     "pyproject.toml:version"

{tracktolib-0.59.0 → tracktolib-0.60.0}/tracktolib/pg/__init__.py

@@ -12,5 +12,6 @@ from .query import (
     update_one,
     insert_pg,
     OnConflict,
+    update_many,
 )
 from .utils import iterate_pg, upsert_csv, safe_pg, safe_pg_context, PGError, PGException

{tracktolib-0.59.0 → tracktolib-0.60.0}/tracktolib/pg/query.py

@@ -17,7 +17,7 @@ V = TypeVar("V")
 
 def _get_insert_query(table: str, columns: Iterable[K], values: str) -> str:
     _columns = ", ".join(columns)
-    return f"INSERT INTO {table} AS t ({_columns}) VALUES ( {values} )"
+    return f"INSERT INTO {table} AS t ({_columns}) VALUES ({values})"
 
 
 def _get_returning_query(query: str, returning: Iterable[K]) -> str:
@@ -217,7 +217,7 @@ def get_update_fields(
         fields.append(
             f"{_col} = ${_counter}"
             if k not in _merge_keys
-            else f"{_col} = COALESCE(t.{_col}, jsonb_build_object()) || " f"${_counter}"
+            else f"{_col} = COALESCE(t.{_col}, JSONB_BUILD_OBJECT()) || " f"${_counter}"
         )
         counter += 1
     return ",\n".join(fields), values + where_values
@@ -264,7 +264,7 @@ class PGUpdateQuery(PGQuery):
     def values(self):
         if not self._values:
             raise ValueError("No values found")
-        return self._values
+        return self._values if len(self.items) == 1 else super().values
 
     def _get_where_query(self) -> str:
         if self.where:
@@ -469,3 +469,18 @@ async def update_returning(
     )
     fn = conn.fetchval if len(returning_values or []) == 1 else conn.fetchrow
     return await fn(query.query, *args, *query.values)
+
+
+async def update_many(
+    conn: _Connection,
+    table: str,
+    items: list[dict],
+    keys: list[str] | None = None,
+    start_from: int | None = None,
+    where: str | None = None,
+    merge_keys: list[str] | None = None,
+):
+    query = PGUpdateQuery(
+        table=table, items=items, start_from=start_from, where_keys=keys, where=where, merge_keys=merge_keys
+    )
+    await conn.executemany(query.query, query.values)

{tracktolib-0.59.0 → tracktolib-0.60.0}/tracktolib/pg_utils.py

@@ -70,7 +70,7 @@ def get_conflict_query(
     fields = ", ".join(f"{x} = COALESCE(EXCLUDED.{x}, t.{x})" for x in columns if x not in _ignore_columns)
     if merge_columns:
         fields = fields + ", " if fields else fields
-        fields += ", ".join(f"{x} = COALESCE(t.{x}, jsonb_build_object()) || EXCLUDED.{x}" for x in merge_columns)
+        fields += ", ".join(f"{x} = COALESCE(t.{x}, JSONB_BUILD_OBJECT()) || EXCLUDED.{x}" for x in merge_columns)
     if not fields:
         raise ValueError("No fields set")