datashare-python 0.10.4__tar.gz → 0.10.6__tar.gz

{datashare_python-0.10.4 → datashare_python-0.10.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datashare-python
-Version: 0.10.4
+Version: 0.10.6
 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.4 → datashare_python-0.10.6}/datashare_python/interceptors.py

@@ -8,7 +8,7 @@ from contextlib import contextmanager
 from contextvars import ContextVar
 from copy import deepcopy
 from functools import wraps
-from inspect import signature
+from inspect import iscoroutinefunction, signature
 from types import UnionType
 from typing import (
     Annotated,
@@ -28,6 +28,7 @@ from temporalio.activity import _Definition
 from temporalio.api.common.v1 import Payload
 from temporalio.client import WorkflowHandle
 from temporalio.converter import DataConverter
+from temporalio.service import RPCError
 from temporalio.worker import (
     ActivityInboundInterceptor,
     ActivityOutboundInterceptor,
@@ -332,7 +333,8 @@ def _get_progress_handler(
         weight=weight,
         min_progress_interval_s=min_progress_interval_s,
     )
-    return handler.progress
+    handler = _fail_safe(handler.progress, "emit progress", excs=(RPCError,))
+    return handler
 
 
 def _is_progress(t: type) -> bool:
@@ -438,7 +440,7 @@ class _HeartbeatInboundInterceptor(ActivityInboundInterceptor):
             period = heartbeat_timeout.total_seconds() / self._n_missed_before_timeout
             # We don't want a failing hearbeat to fail the worker task so we just
             # ignore any exception
-            heartbeat_fn = _fail_safe(self._outbound.heartbeat)
+            heartbeat_fn = _fail_safe(self._outbound.heartbeat, "heartbeat")
             heartbeat_task = asyncio.create_task(_heartbeat_every(heartbeat_fn, period))
         try:
             activity.heartbeat()
@@ -449,19 +451,32 @@ class _HeartbeatInboundInterceptor(ActivityInboundInterceptor):
                 await asyncio.wait([heartbeat_task])
 
 
-def _fail_safe(
-    fn: Callable[[Any], None], excs: tuple[type[Exception]] | None = None
-) -> Callable[[Any], None]:
+def _fail_safe[**P, T](
+    fn: Callable[P, T],
+    action: str,
+    excs: tuple[type[Exception]] | None = None,
+) -> Callable[P, T]:
     if excs is None:
         excs = (Exception,)
+    if iscoroutinefunction(fn):
 
-    @wraps(fn)
-    def wrapper(*args, **kwargs) -> None:
-        try:
-            fn(*args, **kwargs)
-        except excs as exc:
-            msg = f"failed to heartbeat due to {exc}"
-            logger.exception(msg)
+        @wraps(fn)
+        async def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
+            try:
+                return await fn(*args, **kwargs)
+            except excs as exc:
+                msg = f"failed to {action} due to {exc}"
+                logger.exception(msg)
+
+    else:
+
+        @wraps(fn)
+        def wrapper(*args: P.args, **kwargs: P.kwargs) -> None:
+            try:
+                return fn(*args, **kwargs)
+            except excs as exc:
+                msg = f"failed to {action} due to {exc}"
+                logger.exception(msg)
 
     return wrapper
 

{datashare_python-0.10.4 → datashare_python-0.10.6}/pyproject.toml

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