tracktolib 0.53.1__tar.gz → 0.54.0__tar.gz

{tracktolib-0.53.1 → tracktolib-0.54.0}/PKG-INFO

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: tracktolib
-Version: 0.53.1
+Version: 0.54.0
 Summary: Utility library for python
-Home-page: https://github.com/tracktor/tracktolib
 License: MIT
 Keywords: utility
 Author: Julien Brayere
@@ -32,6 +31,7 @@ Requires-Dist: pycryptodome (>=3.20.0) ; extra == "s3-minio"
 Requires-Dist: pydantic (>=2) ; extra == "api"
 Requires-Dist: python-json-logger (>=3.2.1) ; extra == "logs"
 Requires-Dist: rich (>=13.6.0) ; extra == "pg"
+Project-URL: Homepage, https://github.com/tracktor/tracktolib
 Project-URL: Repository, https://github.com/tracktor/tracktolib
 Description-Content-Type: text/markdown
 

{tracktolib-0.53.1 → tracktolib-0.54.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tracktolib"
-version = "0.53.1"
+version = "0.54.0"
 description = "Utility library for python"
 authors = ["Julien Brayere <julien.brayere@tracktor.fr>"]
 license = "MIT"
@@ -55,7 +55,7 @@ httpx =  "^0.27.0"
 psycopg = "^3.2.1"
 minio = "^7.2.7"
 pycryptodome = "^3.20.0"
-aiobotocore = "^2.13.1"
+aiobotocore = "2.15.2"
 python-json-logger = "^3.2.0"
 
 
@@ -83,7 +83,7 @@ pythonPlatform = "Linux"
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.53.1"
+version = "0.54.0"
 tag_format = "$version"
 version_files = [
     "pyproject.toml:version"

{tracktolib-0.53.1 → tracktolib-0.54.0}/tracktolib/pg_sync.py

@@ -15,6 +15,21 @@ except ImportError:
 from .pg_utils import get_tmp_table_query
 
 
+__all__ = (
+    "clean_tables",
+    "drop_db",
+    "fetch_all",
+    "fetch_count",
+    "fetch_one",
+    "get_insert_data",
+    "get_tables",
+    "insert_many",
+    "insert_one",
+    "insert_csv",
+    "set_seq_max",
+)
+
+
 def fetch_all(engine: Connection, query: LiteralString, *data) -> list[dict]:
     with engine.cursor(row_factory=dict_row) as cur:
         resp = (cur.execute(query) if not data else cur.execute(query, data)).fetchall()
@@ -59,7 +74,7 @@ def _parse_value(v):
     return v
 
 
-def _get_insert_data(table: LiteralString, data: list[dict]) -> tuple[LiteralString, list[tuple[Any, ...]]]:
+def get_insert_data(table: LiteralString, data: list[dict]) -> 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})"
@@ -67,17 +82,37 @@ def _get_insert_data(table: LiteralString, data: list[dict]) -> tuple[LiteralStr
 
 
 def insert_many(engine: Connection, table: LiteralString, data: list[dict]):
-    query, _data = _get_insert_data(table, data)
+    query, _data = get_insert_data(table, data)
     with engine.cursor() as cur:
         _ = cur.executemany(query, _data)
     engine.commit()
 
 
-def insert_one(engine: Connection, table: LiteralString, data: dict):
-    query, _data = _get_insert_data(table, [data])
-    with engine.cursor() as cur:
+@overload
+def insert_one(engine: Connection, table: LiteralString, data: dict, 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: dict, returning: list[LiteralString] | None = None
+) -> dict | None:
+    query, _data = get_insert_data(table, [data])
+    _is_returning = False
+    if returning:
+        query = f"{query} RETURNING {','.join(returning)}"
+        _is_returning = True
+
+    with engine.cursor(row_factory=dict_row) as cur:
         _ = cur.execute(query, _data[0])
+        if _is_returning:
+            resp = cur.fetchone()
+        else:
+            resp = None
     engine.commit()
+    return resp
 
 
 def drop_db(conn: Connection, db_name: LiteralString):