tracktolib 0.53.1__tar.gz → 0.55.0__tar.gz

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

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: tracktolib
-Version: 0.53.1
+Version: 0.55.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.55.0}/pyproject.toml

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

{tracktolib-0.53.1 → tracktolib-0.55.0}/tracktolib/api.py

@@ -69,6 +69,7 @@ class MethodMeta(TypedDict):
     name: str | None
     summary: str | None
     description: str | None
+    deprecated: bool | None
 
 
 @dataclass
@@ -89,6 +90,7 @@ class Endpoint:
         name: str | None = None,
         summary: str | None = None,
         description: str | None = None,
+        deprecated: bool | None = None,
     ):
         return _get_method_wrapper(
             cls=self,
@@ -101,6 +103,7 @@ class Endpoint:
             name=name,
             summary=summary,
             description=description,
+            deprecated=deprecated,
         )
 
     def post(
@@ -114,6 +117,7 @@ class Endpoint:
         name: str | None = None,
         summary: str | None = None,
         description: str | None = None,
+        deprecated: bool | None = None,
     ):
         return _get_method_wrapper(
             cls=self,
@@ -126,6 +130,7 @@ class Endpoint:
             name=name,
             summary=summary,
             description=description,
+            deprecated=deprecated,
         )
 
     def put(
@@ -138,6 +143,7 @@ class Endpoint:
         name: str | None = None,
         summary: str | None = None,
         description: str | None = None,
+        deprecated: bool | None = None,
     ):
         return _get_method_wrapper(
             cls=self,
@@ -150,6 +156,7 @@ class Endpoint:
             name=name,
             summary=summary,
             description=description,
+            deprecated=deprecated,
         )
 
     def delete(
@@ -162,6 +169,7 @@ class Endpoint:
         name: str | None = None,
         summary: str | None = None,
         description: str | None = None,
+        deprecated: bool | None = None,
     ):
         return _get_method_wrapper(
             cls=self,
@@ -174,6 +182,7 @@ class Endpoint:
             name=name,
             summary=summary,
             description=description,
+            deprecated=deprecated,
         )
 
     def patch(
@@ -186,6 +195,7 @@ class Endpoint:
         name: str | None = None,
         summary: str | None = None,
         description: str | None = None,
+        deprecated: bool | None = None,
     ):
         return _get_method_wrapper(
             cls=self,
@@ -198,6 +208,7 @@ class Endpoint:
             name=name,
             summary=summary,
             description=description,
+            deprecated=deprecated,
         )
 
 
@@ -213,6 +224,7 @@ def _get_method_wrapper(
     name: str | None = None,
     summary: str | None = None,
     description: str | None = None,
+    deprecated: bool | None = None,
 ):
     def _set_method_wrapper(func: EnpointFn):
         if model is not None:
@@ -232,6 +244,7 @@ def _get_method_wrapper(
             "name": name,
             "summary": summary,
             "description": description,
+            "deprecated": deprecated,
         }
         cls._methods[method] = _meta
 
@@ -265,6 +278,7 @@ def add_endpoint(
         _name = _meta.get("name")
         _summary = _meta.get("summary")
         _description = _meta.get("description")
+        _deprecated = _meta.get("deprecated")
         if not _response_model:
             try:
                 _response_model = _get_return_type(_fn)
@@ -292,6 +306,7 @@ def add_endpoint(
             status_code=_status_code,
             dependencies=[*(_dependencies or []), *(dependencies or [])],
             openapi_extra=_meta.get("openapi_extra"),
+            deprecated=_deprecated,
         )
 
 

{tracktolib-0.53.1 → tracktolib-0.55.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):