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

tracktolib/pg_sync.py

@@ -14,7 +14,6 @@ except ImportError:
 
 from .pg_utils import get_tmp_table_query
 
-
 __all__ = (
     "clean_tables",
     "drop_db",
@@ -81,11 +80,15 @@ def get_insert_data(table: LiteralString, data: list[dict]) -> tuple[LiteralStri
     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: list[dict]):
     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
@@ -138,11 +141,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.56.0.dist-info}/METADATA

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

{tracktolib-0.55.0.dist-info → tracktolib-0.56.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=82fh2XUlH55veWkdIRfirOSjGUWwK759U51ZsBXc2_U,6214
 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.56.0.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.56.0.dist-info/METADATA,sha256=kigOjHx1QJ-xf6L0tUnhJUiZiYMB6VDGso-yzuspREk,3704
+tracktolib-0.56.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+tracktolib-0.56.0.dist-info/RECORD,,

{tracktolib-0.55.0.dist-info → tracktolib-0.56.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