tracktolib 0.55.0__tar.gz → 0.56.0__tar.gz

{tracktolib-0.55.0 → tracktolib-0.56.0}/PKG-INFO

@@ -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 → tracktolib-0.56.0}/pyproject.toml

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

{tracktolib-0.55.0 → tracktolib-0.56.0}/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