tracktolib 0.55.0__py3-none-any.whl → 0.57.0__py3-none-any.whl

tracktolib/pg_sync.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Iterable, Any, overload, Literal, cast, Optional
+from typing import Iterable, Any, overload, Literal, cast, Optional, Mapping, Sequence
 
 from typing_extensions import LiteralString
 
@@ -14,7 +14,6 @@ except ImportError:
 
 from .pg_utils import get_tmp_table_query
 
-
 __all__ = (
     "clean_tables",
     "drop_db",
@@ -68,36 +67,44 @@ def fetch_one(engine: Connection, query: Query, *args, required: bool = False) -
     return _data
 
 
-def _parse_value(v):
+def _parse_value(v: Any) -> Any:
     if isinstance(v, dict):
         return Json(v)
     return v
 
 
-def get_insert_data(table: LiteralString, data: list[dict]) -> tuple[LiteralString, list[tuple[Any, ...]]]:
+def get_insert_data(
+    table: LiteralString, data: Sequence[Mapping[str, Any]]
+) -> tuple[LiteralString, list[tuple[Any, ...]]]:
     keys = data[0].keys()
     _values = ",".join("%s" for _ in range(0, len(keys)))
-    query = f"INSERT INTO {table} as t ({','.join(keys)}) VALUES ({_values})"
+    query = cast(LiteralString, f"INSERT INTO {table} as t ({','.join(keys)}) VALUES ({_values})")
     return query, [tuple(_parse_value(_x) for _x in x.values()) for x in data]
 
 
-def insert_many(engine: Connection, table: LiteralString, data: list[dict]):
+def insert_many(engine: Connection | Cursor, table: LiteralString, data: Sequence[Mapping[str, Any]]):
     query, _data = get_insert_data(table, data)
-    with engine.cursor() as cur:
-        _ = cur.executemany(query, _data)
-    engine.commit()
+    if isinstance(engine, Connection):
+        with engine.cursor() as cur:
+            _ = cur.executemany(query, _data)
+        engine.commit()
+    else:
+        # If we are using a cursor, we can use executemany directly
+        _ = engine.executemany(query, _data)
 
 
 @overload
-def insert_one(engine: Connection, table: LiteralString, data: dict, returning: None = None) -> None: ...
+def insert_one(engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: None = None) -> None: ...
 
 
 @overload
-def insert_one(engine: Connection, table: LiteralString, data: dict, returning: list[LiteralString]) -> dict: ...
+def insert_one(
+    engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString]
+) -> dict: ...
 
 
 def insert_one(
-    engine: Connection, table: LiteralString, data: dict, returning: list[LiteralString] | None = None
+    engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString] | None = None
 ) -> dict | None:
     query, _data = get_insert_data(table, [data])
     _is_returning = False
@@ -138,11 +145,11 @@ def clean_tables(engine: Connection, tables: Iterable[LiteralString], reset_seq:
 
 def get_tables(engine: Connection, schemas: list[str], ignored_tables: Iterable[str] | None = None):
     table_query = """
-    SELECT CONCAT_WS('.', schemaname, tablename) AS table
-    FROM pg_catalog.pg_tables
-    WHERE schemaname = ANY(%s)
-    ORDER BY schemaname, tablename
-    """
+                  SELECT CONCAT_WS('.', schemaname, tablename) AS table
+                  FROM pg_catalog.pg_tables
+                  WHERE schemaname = ANY (%s)
+                  ORDER BY schemaname, tablename \
+                  """
     resp = fetch_all(engine, table_query, schemas)
 
     # Foreign keys

{tracktolib-0.55.0.dist-info → tracktolib-0.57.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: tracktolib
-Version: 0.55.0
+Version: 0.57.0
 Summary: Utility library for python
 License: MIT
 Keywords: utility

{tracktolib-0.55.0.dist-info → tracktolib-0.57.0.dist-info}/RECORD

@@ -5,14 +5,14 @@ tracktolib/logs.py,sha256=xgfRDnviZPr9qinSOgab8neIWPv9Rj2xzDHnIBvWQcU,2201
 tracktolib/pg/__init__.py,sha256=j67e3B3gBbCHLD20QBybptmNdbbVMzNhZE6XjIPuKVo,349
 tracktolib/pg/query.py,sha256=MBYOTL9aLnYRBhaLy8-P6uGNXOmyluejNyUdt3Gfx5M,14504
 tracktolib/pg/utils.py,sha256=cL24KEt4SWJQ7LJPzaO3c8Xg0ZLmjhn22DtTWg86nwc,6324
-tracktolib/pg_sync.py,sha256=QSvt5C92UTcc6EhtjrdUe6eDs7m9mIyBYclzQo9wKGQ,5959
+tracktolib/pg_sync.py,sha256=snfYtj5ICWa2C_QWRFKZecHPgVx1yhofhGst0e6CGxs,6351
 tracktolib/pg_utils.py,sha256=VXPpy1jGq6aCgTlfFJDIrq6JDujR83JN5ZRiCi8Lx4E,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.55.0.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
-tracktolib-0.55.0.dist-info/METADATA,sha256=MRCRmul0rCAPnWJe3HZJH28llOPQowxPEVXc6q4JGHI,3704
-tracktolib-0.55.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-tracktolib-0.55.0.dist-info/RECORD,,
+tracktolib-0.57.0.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.57.0.dist-info/METADATA,sha256=88PzWoI3eX1blMUAWEHUzuYFPWndH7O0sCPr9_ScSNE,3704
+tracktolib-0.57.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+tracktolib-0.57.0.dist-info/RECORD,,

{tracktolib-0.55.0.dist-info → tracktolib-0.57.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any