tracktolib 0.57.0__tar.gz → 0.58.1__tar.gz

{tracktolib-0.57.0 → tracktolib-0.58.1}/PKG-INFO

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

{tracktolib-0.57.0 → tracktolib-0.58.1}/pyproject.toml

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

{tracktolib-0.57.0 → tracktolib-0.58.1}/tracktolib/pg_sync.py

@@ -7,7 +7,7 @@ try:
     from psycopg import Connection, Cursor
     from psycopg.abc import Query
     from psycopg.errors import InvalidCatalogName
-    from psycopg.rows import dict_row
+    from psycopg.rows import dict_row, DictRow, TupleRow
     from psycopg.types.json import Json
 except ImportError:
     raise ImportError('Please install psycopg or tracktolib with "pg-sync" to use this module')
@@ -94,7 +94,9 @@ def insert_many(engine: Connection | Cursor, table: LiteralString, data: Sequenc
 
 
 @overload
-def insert_one(engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: None = None) -> None: ...
+def insert_one(
+    engine: Connection | Cursor, table: LiteralString, data: Mapping[str, Any], returning: None = None
+) -> None: ...
 
 
 @overload
@@ -103,22 +105,32 @@ def insert_one(
 ) -> dict: ...
 
 
+@overload
+def insert_one(
+    engine: Cursor, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString]
+) -> DictRow | TupleRow | None: ...
+
+
 def insert_one(
-    engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString] | None = None
-) -> dict | None:
+    engine: Connection | Cursor,
+    table: LiteralString,
+    data: Mapping[str, Any],
+    returning: list[LiteralString] | None = None,
+) -> dict | DictRow | TupleRow | 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()
+    if isinstance(engine, Connection):
+        with engine.cursor(row_factory=dict_row) as cur:
+            _ = cur.execute(query, _data[0])
+            resp = cur.fetchone() if _is_returning else None
+        engine.commit()
+    else:
+        _ = engine.execute(query, _data[0])
+        resp = engine.fetchone() if _is_returning else None
     return resp