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

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')
@@ -103,22 +103,32 @@ def insert_one(
 ) -> dict: ...
 
 
+@overload
 def insert_one(
-    engine: Connection, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString] | None = None
-) -> dict | None:
+    engine: Cursor, table: LiteralString, data: Mapping[str, Any], returning: list[LiteralString]
+) -> DictRow | TupleRow | None: ...
+
+
+def insert_one(
+    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
 
 

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

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

{tracktolib-0.57.0.dist-info → tracktolib-0.58.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=snfYtj5ICWa2C_QWRFKZecHPgVx1yhofhGst0e6CGxs,6351
+tracktolib/pg_sync.py,sha256=GvrJfEkML2p7L4DY7EAsL2qft-3l87jngpgYwC3iz18,6703
 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.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.58.0.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.58.0.dist-info/METADATA,sha256=GHO46HMt3K1DMn4jMoOjIUaUlnPCjpUBTciyu9qljs0,3704
+tracktolib-0.58.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+tracktolib-0.58.0.dist-info/RECORD,,