datashare-python 0.10.5__tar.gz → 0.10.7__tar.gz

{datashare_python-0.10.5 → datashare_python-0.10.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datashare-python
-Version: 0.10.5
+Version: 0.10.7
 Summary: Manage Python tasks and local resources in Datashare
 Project-URL: Homepage, https://icij.github.io/datashare-python/
 Project-URL: Documentation, https://icij.github.io/datashare-python/

{datashare_python-0.10.5 → datashare_python-0.10.7}/datashare_python/interceptors.py

@@ -14,7 +14,6 @@ from typing import (
     Annotated,
     Any,
     NoReturn,
-    ParamSpec,
     Self,
     TypeVar,
     get_args,
@@ -452,11 +451,7 @@ class _HeartbeatInboundInterceptor(ActivityInboundInterceptor):
                 await asyncio.wait([heartbeat_task])
 
 
-P = ParamSpec("P")
-T = TypeVar("T")
-
-
-def _fail_safe[P, T](
+def _fail_safe[**P, T](
     fn: Callable[P, T],
     action: str,
     excs: tuple[type[Exception]] | None = None,
@@ -466,9 +461,9 @@ def _fail_safe[P, T](
     if iscoroutinefunction(fn):
 
         @wraps(fn)
-        async def wrapper(*args, **kwargs) -> None:
+        async def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
             try:
-                await fn(*args, **kwargs)
+                return await fn(*args, **kwargs)
             except excs as exc:
                 msg = f"failed to {action} due to {exc}"
                 logger.exception(msg)
@@ -476,9 +471,9 @@ def _fail_safe[P, T](
     else:
 
         @wraps(fn)
-        def wrapper(*args, **kwargs) -> None:
+        def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
             try:
-                fn(*args, **kwargs)
+                return fn(*args, **kwargs)
             except excs as exc:
                 msg = f"failed to {action} due to {exc}"
                 logger.exception(msg)

{datashare_python-0.10.5 → datashare_python-0.10.7}/datashare_python/objects.py

@@ -1,3 +1,5 @@
+import hashlib
+import json
 import logging
 import os
 from abc import ABC
@@ -63,6 +65,12 @@ Predicate = Callable[[T], bool] | Callable[[T], Awaitable[bool]]
 class BaseModel(_BaseModel):
     model_config = merge_configs(icij_config(), no_enum_values_config())
 
+    def __hash__(self) -> int:
+        digest = hashlib.md5(
+            json.dumps(self.model_dump(mode="json"), sort_keys=True).encode()
+        ).digest()
+        return int.from_bytes(digest[:8])
+
 
 class DatashareModel(BaseModel):
     model_config = merge_configs(BaseModel.model_config, lowercamel_case_config())

{datashare_python-0.10.5 → datashare_python-0.10.7}/datashare_python/utils.py

@@ -32,6 +32,7 @@ import temporalio
 from aiofile import async_open
 from lru import LRU
 from pydantic import ValidationError
+from pydantic.alias_generators import to_snake
 from temporalio import activity, workflow
 from temporalio.api.common.v1 import Payload
 from temporalio.client import Client
@@ -62,6 +63,8 @@ from .objects import (
 )
 from .types_ import RawAsyncProgressHandler
 
+_CACHE_DIR = "__cache__"
+
 logger = logging.getLogger(__name__)
 
 DependencyLabel = str | None
@@ -533,7 +536,11 @@ activity_contextual_id = contextual_id
 
 
 def _contextual_path(
-    *, wf_context: bool = True, act_context: bool = True, run_context: bool = False
+    *,
+    wf_context: bool = True,
+    act_context: bool = True,
+    run_context: bool = False,
+    caching_key: str | None = None,
 ) -> Path:
     act_info = activity.info()
     path = []
@@ -541,14 +548,18 @@ def _contextual_path(
         raise ValueError("at least one of wf_context and act_context must be True")
     if wf_context:
         path = [act_info.workflow_type]
-        path.append(act_info.workflow_id)
-        if run_context:
-            path.append(act_info.workflow_run_id)
+        if not caching_key:
+            path.append(act_info.workflow_id)
+            if run_context:
+                path.append(act_info.workflow_run_id)
     if act_context:
         path.append(act_info.activity_type)
-        path.append(act_info.activity_id)
-        if run_context:
-            path.append(act_info.activity_run_id)
+        if not caching_key:
+            path.append(act_info.activity_id)
+            if run_context:
+                path.append(act_info.activity_run_id)
+    if caching_key:
+        path += [_CACHE_DIR, caching_key]
     return Path(*path)
 
 
@@ -559,9 +570,13 @@ def activity_workdir(
     wf_context: bool = True,
     act_context: bool = True,
     run_context: bool = False,
+    caching_key: str | None = None,
 ) -> Path:
     ctx_path = _contextual_path(
-        wf_context=wf_context, act_context=act_context, run_context=run_context
+        wf_context=wf_context,
+        act_context=act_context,
+        run_context=run_context,
+        caching_key=caching_key,
     )
     return workdir.joinpath(project, ctx_path)
 
@@ -808,4 +823,10 @@ def async_enter_cm(
 
 
 def config_cache_key(config: BaseModel) -> str:
-    return str(hash(config))
+    # BaseModel support deterministic hash we can call hash on it
+    config_label = (
+        to_snake(config.__class__.__name__.replace("Config", ""))
+        .lower()
+        .replace("_", "-")
+    )
+    return f"{config_label}-{str(hash(config))}"

{datashare_python-0.10.5 → datashare_python-0.10.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datashare-python"
-version = "0.10.5"
+version = "0.10.7"
 description = "Manage Python tasks and local resources in Datashare"
 authors = [
   { name = "Clément Doumouro", email = "cdoumouro@icij.org" },