tracktolib 0.59.0__py3-none-any.whl → 0.60.0__py3-none-any.whl

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/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/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")
 

{tracktolib-0.59.0.dist-info → tracktolib-0.60.0.dist-info}/METADATA

@@ -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.dist-info → tracktolib-0.60.0.dist-info}/RECORD

@@ -2,17 +2,17 @@ tracktolib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tracktolib/api.py,sha256=ZLMgjH3Y8r3MpXc8m3IuZbzTj3fgrZKZORtSVgbuP-M,10221
 tracktolib/http_utils.py,sha256=c10JGmHaBw3VSDMYhz2dvVw2lo4PUAq1xMub74I7xDc,2625
 tracktolib/logs.py,sha256=xgfRDnviZPr9qinSOgab8neIWPv9Rj2xzDHnIBvWQcU,2201
-tracktolib/pg/__init__.py,sha256=j67e3B3gBbCHLD20QBybptmNdbbVMzNhZE6XjIPuKVo,349
-tracktolib/pg/query.py,sha256=MBYOTL9aLnYRBhaLy8-P6uGNXOmyluejNyUdt3Gfx5M,14504
+tracktolib/pg/__init__.py,sha256=Ul_hgwvTXZvQBt7sHKi4ZI-0DDpnXmoFtmVkGRy-1J0,366
+tracktolib/pg/query.py,sha256=PQ-vMTwsdvGpM82oE6moHVfD6eta61yPgB7IYRU8CVU,14973
 tracktolib/pg/utils.py,sha256=cL24KEt4SWJQ7LJPzaO3c8Xg0ZLmjhn22DtTWg86nwc,6324
 tracktolib/pg_sync.py,sha256=MKDaV7dYsRy59Y0EE5RGZL0DlZ-RUdBeaN9eSBwiQJg,6718
-tracktolib/pg_utils.py,sha256=VXPpy1jGq6aCgTlfFJDIrq6JDujR83JN5ZRiCi8Lx4E,2582
+tracktolib/pg_utils.py,sha256=ArYNdf9qsdYdzGEWmev8tZpyx8_1jaGGdkfYkauM7UM,2582
 tracktolib/s3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tracktolib/s3/minio.py,sha256=wMEjkSes9Fp39fD17IctALpD6zB2xwDRQEmO7Vzan3g,1387
 tracktolib/s3/s3.py,sha256=0HbSAPoaup5-W4LK54zRCjrQ5mr8OWR-N9WjW99Q4aw,5937
 tracktolib/tests.py,sha256=gKE--epQjgMZGXc5ydbl4zjOdmwztJS42UMV0p4hXEA,399
 tracktolib/utils.py,sha256=ysTBF9V35fVXQVBPk0kfE_84SGRxzrayqmg9RbtoJq4,5761
-tracktolib-0.59.0.dist-info/METADATA,sha256=DLi6sZ3qfyvPFWz06zWPOe8-eszB5Dv1CSIcIENi8B8,3777
-tracktolib-0.59.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-tracktolib-0.59.0.dist-info/licenses/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
-tracktolib-0.59.0.dist-info/RECORD,,
+tracktolib-0.60.0.dist-info/METADATA,sha256=1qNZSiff8zDxWsmA-lgHQ45EnBufiABtkMy6EK4KNbM,3777
+tracktolib-0.60.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+tracktolib-0.60.0.dist-info/licenses/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.60.0.dist-info/RECORD,,