durabletask 0.0.0.dev21__tar.gz → 0.0.0.dev22__tar.gz

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev21
+Version: 0.0.0.dev22
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/client.py

@@ -11,8 +11,7 @@ from typing import Any, Optional, Sequence, TypeVar, Union
 import grpc
 from google.protobuf import wrappers_pb2
 
-from durabletask.entities import EntityInstanceId
-from durabletask.entities.entity_metadata import EntityMetadata
+from durabletask.entities import EntityInstanceId, EntityMetadata
 import durabletask.internal.helpers as helpers
 import durabletask.internal.orchestrator_service_pb2 as pb
 import durabletask.internal.orchestrator_service_pb2_grpc as stubs
@@ -230,7 +229,10 @@ class TaskHubGrpcClient:
         self._logger.info(f"Purging instance '{instance_id}'.")
         self._stub.PurgeInstances(req)
 
-    def signal_entity(self, entity_instance_id: EntityInstanceId, operation_name: str, input: Optional[Any] = None):
+    def signal_entity(self,
+                      entity_instance_id: EntityInstanceId[TInput, TOutput],
+                      operation_name: str,
+                      input: Optional[TInput] = None):
         req = pb.SignalEntityRequest(
             instanceId=str(entity_instance_id),
             name=operation_name,
@@ -244,7 +246,7 @@ class TaskHubGrpcClient:
         self._stub.SignalEntity(req, None)  # TODO: Cancellation timeout?
 
     def get_entity(self,
-                   entity_instance_id: EntityInstanceId,
+                   entity_instance_id: EntityInstanceId[Any, Any],
                    include_state: bool = True
                    ) -> Optional[EntityMetadata]:
         req = pb.GetEntityRequest(instanceId=str(entity_instance_id), includeState=include_state)

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/entities/durable_entity.py

@@ -4,6 +4,7 @@ from durabletask.entities.entity_context import EntityContext
 from durabletask.entities.entity_instance_id import EntityInstanceId
 
 TState = TypeVar("TState")
+TInput = TypeVar("TInput")
 
 
 class DurableEntity:
@@ -49,7 +50,10 @@ class DurableEntity:
         """
         self.entity_context.set_state(state)
 
-    def signal_entity(self, entity_instance_id: EntityInstanceId, operation: str, input: Optional[Any] = None) -> None:
+    def signal_entity(self,
+                      entity_instance_id: EntityInstanceId[TInput, Any],
+                      operation: str,
+                      input: Optional[TInput] = None) -> None:
         """Signal another entity to perform an operation.
 
         Parameters

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/entities/entity_context.py

@@ -7,6 +7,7 @@ from durabletask.internal.entity_state_shim import StateShim
 import durabletask.internal.orchestrator_service_pb2 as pb
 
 TState = TypeVar("TState")
+TInput = TypeVar("TInput")
 
 
 class EntityContext:
@@ -81,7 +82,7 @@ class EntityContext:
         """
         self._state.set_state(new_state)
 
-    def signal_entity(self, entity_instance_id: EntityInstanceId, operation: str, input: Optional[Any] = None) -> None:
+    def signal_entity(self, entity_instance_id: EntityInstanceId[TInput, Any], operation: str, input: Optional[TInput] = None) -> None:
         """Signal another entity to perform an operation.
 
         Parameters

durabletask-0.0.0.dev22/durabletask/entities/entity_instance_id.py

@@ -0,0 +1,89 @@
+from typing import Any, Callable, TypeVar, Union, overload, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from durabletask import task
+    from durabletask.entities.durable_entity import DurableEntity
+    from durabletask.entities.entity_context import EntityContext
+
+
+TInput = TypeVar('TInput')
+TOutput = TypeVar('TOutput')
+
+
+class EntityInstanceId[TInput, TOutput]:
+    @overload
+    def __new__(
+        cls,
+        entity: Callable[[EntityContext, TInput], TOutput],
+        key: str
+    ) -> "EntityInstanceId[TInput, TOutput]": ...
+
+    @overload
+    def __new__(
+        cls,
+        entity: type[DurableEntity],
+        key: str
+    ) -> "EntityInstanceId[Any, Any]": ...
+
+    @overload
+    def __new__(
+        cls,
+        entity: str,
+        key: str
+    ) -> "EntityInstanceId[Any, Any]": ...
+
+    def __new__(
+        cls,
+        entity: Union[task.Entity[TInput, TOutput], str],
+        key: str
+    ) -> "EntityInstanceId[Any, Any]":
+        return super().__new__(cls)
+
+    def __init__(
+        self,
+        entity: Union[task.Entity[TInput, TOutput], str],
+        key: str
+    ):
+        if not isinstance(entity, str):
+            from durabletask import task
+            entity = task.get_entity_name(entity)
+        self.entity: str = entity
+        self.key: str = key
+
+    def __str__(self) -> str:
+        return f"@{self.entity}@{self.key}"
+
+    def __eq__(self, other):
+        if not isinstance(other, EntityInstanceId):
+            return False
+        return self.entity == other.entity and self.key == other.key
+
+    def __lt__(self, other):
+        if not isinstance(other, EntityInstanceId):
+            return self < other
+        return str(self) < str(other)
+
+    @staticmethod
+    def parse(entity_id: str) -> "EntityInstanceId[Any, Any]":
+        """Parse a string representation of an entity ID into an EntityInstanceId object.
+
+        Parameters
+        ----------
+        entity_id : str
+            The string representation of the entity ID, in the format '@entity@key'.
+
+        Returns
+        -------
+        EntityInstanceId
+            The parsed EntityInstanceId object.
+
+        Raises
+        ------
+        ValueError
+            If the input string is not in the correct format.
+        """
+        try:
+            _, entity, key = entity_id.split("@", 2)
+            return EntityInstanceId(entity=entity, key=key)
+        except ValueError as ex:
+            raise ValueError(f"Invalid entity ID format: {entity_id}", ex)

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/entities/entity_metadata.py

@@ -24,7 +24,7 @@ class EntityMetadata:
     """
 
     def __init__(self,
-                 id: EntityInstanceId,
+                 id: EntityInstanceId[Any, Any],
                  last_modified: datetime,
                  backlog_queue_size: int,
                  locked_by: str,

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/internal/helpers.py

@@ -116,11 +116,18 @@ def new_sub_orchestration_failed_event(event_id: int, ex: Exception) -> pb.Histo
     )
 
 
-def new_failure_details(ex: Exception) -> pb.TaskFailureDetails:
+def new_failure_details(ex: Exception, _visited: Optional[set[int]] = None) -> pb.TaskFailureDetails:
+    if _visited is None:
+        _visited = set()
+    _visited.add(id(ex))
+    inner: Optional[BaseException] = ex.__cause__ or ex.__context__
+    if len(_visited) > 10 or (inner and id(inner) in _visited) or not isinstance(inner, Exception):
+        inner = None
     return pb.TaskFailureDetails(
         errorType=type(ex).__name__,
         errorMessage=str(ex),
-        stackTrace=wrappers_pb2.StringValue(value=''.join(traceback.format_tb(ex.__traceback__)))
+        stackTrace=wrappers_pb2.StringValue(value=''.join(traceback.format_tb(ex.__traceback__))),
+        innerFailure=new_failure_details(inner, _visited) if inner else None
     )
 
 

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/task.py

@@ -140,14 +140,14 @@ class OrchestrationContext(ABC):
 
     @abstractmethod
     def call_entity(self,
-                    entity: EntityInstanceId,
+                    entity: EntityInstanceId[TInput, TOutput],
                     operation: str,
-                    input: Optional[TInput] = None) -> CompletableTask:
+                    input: Optional[TInput] = None) -> CompletableTask[TOutput]:
         """Schedule entity function for execution.
 
         Parameters
         ----------
-        entity: EntityInstanceId
+        entity: EntityInstanceId[TInput, TOutput]
             The ID of the entity instance to call.
         operation: str
             The name of the operation to invoke on the entity.
@@ -164,7 +164,7 @@ class OrchestrationContext(ABC):
     @abstractmethod
     def signal_entity(
             self,
-            entity_id: EntityInstanceId,
+            entity_id: EntityInstanceId[TInput, TOutput],
             operation_name: str,
             input: Optional[TInput] = None
     ) -> None:
@@ -172,7 +172,7 @@ class OrchestrationContext(ABC):
 
         Parameters
         ----------
-        entity_id: EntityInstanceId
+        entity_id: EntityInstanceId[TInput, TOutput]
             The ID of the entity instance to signal.
         operation_name: str
             The name of the operation to invoke on the entity.
@@ -182,7 +182,7 @@ class OrchestrationContext(ABC):
         pass
 
     @abstractmethod
-    def lock_entities(self, entities: list[EntityInstanceId]) -> CompletableTask[EntityLock]:
+    def lock_entities(self, entities: list[EntityInstanceId[Any, Any]]) -> CompletableTask[EntityLock]:
         """Creates a Task object that locks the specified entity instances.
 
         The locks will be acquired the next time the orchestrator yields.
@@ -191,7 +191,7 @@ class OrchestrationContext(ABC):
 
         Parameters
         ----------
-        entities: list[EntityInstanceId]
+        entities: list[EntityInstanceId[Any, Any]]
             The list of entity instance IDs to lock.
 
         Returns
@@ -302,8 +302,10 @@ class FailureDetails:
 class TaskFailedError(Exception):
     """Exception type for all orchestration task failures."""
 
-    def __init__(self, message: str, details: pb.TaskFailureDetails):
+    def __init__(self, message: str, details: Union[pb.TaskFailureDetails, Exception]):
         super().__init__(message)
+        if isinstance(details, Exception):
+            details = pbh.new_failure_details(details)
         self._details = FailureDetails(
             details.errorMessage,
             details.errorType,
@@ -424,7 +426,7 @@ class CompletableTask(Task[T]):
         if self._parent is not None:
             self._parent.on_child_completed(self)
 
-    def fail(self, message: str, details: pb.TaskFailureDetails):
+    def fail(self, message: str, details: Union[Exception, pb.TaskFailureDetails]):
         if self._is_complete:
             raise ValueError('The task has already completed.')
         self._exception = TaskFailedError(message, details)
@@ -536,8 +538,8 @@ class ActivityContext:
         return self._task_id
 
 
-# Orchestrators are generators that yield tasks and receive/return any type
-Orchestrator = Callable[[OrchestrationContext, TInput], Union[Generator[Task, Any, Any], TOutput]]
+# Orchestrators are generators that yield tasks, recieve any type, and return TOutput
+Orchestrator = Callable[[OrchestrationContext, TInput], Union[Generator[Task[Any], Any, TOutput], TOutput]]
 
 # Activities are simple functions that can be scheduled by orchestrators
 Activity = Callable[[ActivityContext, TInput], TOutput]
@@ -613,6 +615,14 @@ class RetryPolicy:
         return self._retry_timeout
 
 
+def get_entity_name(fn: Entity) -> str:
+    if hasattr(fn, "__durable_entity_name__"):
+        return getattr(fn, "__durable_entity_name__")
+    if isinstance(fn, type) and issubclass(fn, DurableEntity):
+        return fn.__name__
+    return get_name(fn)
+
+
 def get_name(fn: Callable) -> str:
     """Returns the name of the provided function"""
     name = fn.__name__

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask/worker.py

@@ -150,7 +150,7 @@ class _Registry:
         self.entities = {}
         self.entity_instances = {}
 
-    def add_orchestrator(self, fn: task.Orchestrator) -> str:
+    def add_orchestrator(self, fn: task.Orchestrator[TInput, TOutput]) -> str:
         if fn is None:
             raise ValueError("An orchestrator function argument is required.")
 
@@ -158,7 +158,7 @@ class _Registry:
         self.add_named_orchestrator(name, fn)
         return name
 
-    def add_named_orchestrator(self, name: str, fn: task.Orchestrator) -> None:
+    def add_named_orchestrator(self, name: str, fn: task.Orchestrator[TInput, TOutput]) -> None:
         if not name:
             raise ValueError("A non-empty orchestrator name is required.")
         if name in self.orchestrators:
@@ -166,7 +166,7 @@ class _Registry:
 
         self.orchestrators[name] = fn
 
-    def get_orchestrator(self, name: str) -> Optional[task.Orchestrator]:
+    def get_orchestrator(self, name: str) -> Optional[task.Orchestrator[Any, Any]]:
         return self.orchestrators.get(name)
 
     def add_activity(self, fn: task.Activity) -> str:
@@ -188,16 +188,13 @@ class _Registry:
     def get_activity(self, name: str) -> Optional[task.Activity]:
         return self.activities.get(name)
 
-    def add_entity(self, fn: task.Entity) -> str:
+    def add_entity(self, fn: task.Entity, name: Optional[str] = None) -> str:
         if fn is None:
             raise ValueError("An entity function argument is required.")
 
-        if isinstance(fn, type) and issubclass(fn, DurableEntity):
-            name = fn.__name__
-            self.add_named_entity(name, fn)
-        else:
-            name = task.get_name(fn)
-            self.add_named_entity(name, fn)
+        if name is None:
+            name = task.get_entity_name(fn)
+        self.add_named_entity(name, fn)
         return name
 
     def add_named_entity(self, name: str, fn: task.Entity) -> None:
@@ -207,6 +204,7 @@ class _Registry:
             raise ValueError(f"A '{name}' entity already exists.")
 
         self.entities[name] = fn
+        setattr(fn, "__durable_entity_name__", name)
 
     def get_entity(self, name: str) -> Optional[task.Entity]:
         return self.entities.get(name)
@@ -362,7 +360,7 @@ class TaskHubGrpcWorker:
     def __exit__(self, type, value, traceback):
         self.stop()
 
-    def add_orchestrator(self, fn: task.Orchestrator) -> str:
+    def add_orchestrator(self, fn: task.Orchestrator[TInput, TOutput]) -> str:
         """Registers an orchestrator function with the worker."""
         if self._is_running:
             raise RuntimeError(
@@ -378,13 +376,13 @@ class TaskHubGrpcWorker:
             )
         return self._registry.add_activity(fn)
 
-    def add_entity(self, fn: task.Entity) -> str:
+    def add_entity(self, fn: task.Entity, name: Optional[str] = None) -> str:
         """Registers an entity function with the worker."""
         if self._is_running:
             raise RuntimeError(
                 "Entities cannot be added while the worker is running."
             )
-        return self._registry.add_entity(fn)
+        return self._registry.add_entity(fn, name)
 
     def use_versioning(self, version: VersioningOptions) -> None:
         """Initializes versioning options for sub-orchestrators and activities."""
@@ -1044,21 +1042,21 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
 
     def call_entity(
             self,
-            entity: EntityInstanceId,
+            entity: EntityInstanceId[TInput, TOutput],
             operation: str,
             input: Optional[TInput] = None,
-    ) -> task.CompletableTask:
+    ) -> task.CompletableTask[TOutput]:
         id = self.next_sequence_number()
 
         self.call_entity_function_helper(
             id, entity, operation, input=input
         )
 
-        return self._pending_tasks.get(id, task.CompletableTask())
+        return self._pending_tasks.get(id, task.CompletableTask[TOutput]())
 
     def signal_entity(
             self,
-            entity_id: EntityInstanceId,
+            entity_id: EntityInstanceId[TInput, TOutput],
             operation_name: str,
             input: Optional[TInput] = None
     ) -> None:
@@ -1068,7 +1066,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             id, entity_id, operation_name, input
         )
 
-    def lock_entities(self, entities: list[EntityInstanceId]) -> task.CompletableTask[EntityLock]:
+    def lock_entities(self, entities: list[EntityInstanceId[Any, Any]]) -> task.CompletableTask[EntityLock]:
         id = self.next_sequence_number()
 
         self.lock_entities_function_helper(
@@ -1158,11 +1156,11 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
     def call_entity_function_helper(
             self,
             id: Optional[int],
-            entity_id: EntityInstanceId,
+            entity_id: EntityInstanceId[TInput, TOutput],
             operation: str,
             *,
             input: Optional[TInput] = None,
-    ):
+    ) -> None:
         if id is None:
             id = self.next_sequence_number()
 
@@ -1180,7 +1178,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
     def signal_entity_function_helper(
             self,
             id: Optional[int],
-            entity_id: EntityInstanceId,
+            entity_id: EntityInstanceId[TInput, TOutput],
             operation: str,
             input: Optional[TInput]
     ) -> None:
@@ -1197,7 +1195,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
         action = ph.new_signal_entity_action(id, entity_id, operation, encoded_input, self.new_uuid())
         self._pending_actions[id] = action
 
-    def lock_entities_function_helper(self, id: int, entities: list[EntityInstanceId]) -> None:
+    def lock_entities_function_helper(self, id: int, entities: list[EntityInstanceId[Any, Any]]) -> None:
         if id is None:
             id = self.next_sequence_number()
 
@@ -1792,7 +1790,7 @@ class _OrchestrationExecutor:
             # The orchestrator generator function completed
             ctx.set_complete(generatorStopped.value, pb.ORCHESTRATION_STATUS_COMPLETED)
 
-    def _parse_entity_event_sent_input(self, event: pb.HistoryEvent) -> Tuple[EntityInstanceId, str]:
+    def _parse_entity_event_sent_input(self, event: pb.HistoryEvent) -> Tuple[EntityInstanceId[Any, Any], str]:
         try:
             entity_id = EntityInstanceId.parse(event.eventSent.instanceId)
         except ValueError:
@@ -1806,7 +1804,7 @@ class _OrchestrationExecutor:
     def _handle_entity_event_raised(self,
                                     ctx: _RuntimeOrchestrationContext,
                                     event: pb.HistoryEvent,
-                                    entity_id: Optional[EntityInstanceId],
+                                    entity_id: Optional[EntityInstanceId[Any, Any]],
                                     task_id: Optional[int],
                                     is_lock_event: bool):
         # This eventRaised represents the result of an entity operation after being translated to the old
@@ -1919,7 +1917,7 @@ class _EntityExecutor:
     def execute(
             self,
             orchestration_id: str,
-            entity_id: EntityInstanceId,
+            entity_id: EntityInstanceId[TInput, TOutput],
             operation: str,
             state: StateShim,
             encoded_input: Optional[str],

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/durabletask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev21
+Version: 0.0.0.dev22
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev21 → durabletask-0.0.0.dev22}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durabletask"
-version = "0.0.0.dev21"
+version = "0.0.0.dev22"
 description = "A Durable Task Client SDK for Python"
 keywords = [
     "durable",

durabletask-0.0.0.dev21/durabletask/entities/entity_instance_id.py

@@ -1,42 +0,0 @@
-class EntityInstanceId:
-    def __init__(self, entity: str, key: str):
-        self.entity = entity
-        self.key = key
-
-    def __str__(self) -> str:
-        return f"@{self.entity}@{self.key}"
-
-    def __eq__(self, other):
-        if not isinstance(other, EntityInstanceId):
-            return False
-        return self.entity == other.entity and self.key == other.key
-
-    def __lt__(self, other):
-        if not isinstance(other, EntityInstanceId):
-            return self < other
-        return str(self) < str(other)
-
-    @staticmethod
-    def parse(entity_id: str) -> "EntityInstanceId":
-        """Parse a string representation of an entity ID into an EntityInstanceId object.
-
-        Parameters
-        ----------
-        entity_id : str
-            The string representation of the entity ID, in the format '@entity@key'.
-
-        Returns
-        -------
-        EntityInstanceId
-            The parsed EntityInstanceId object.
-
-        Raises
-        ------
-        ValueError
-            If the input string is not in the correct format.
-        """
-        try:
-            _, entity, key = entity_id.split("@", 2)
-            return EntityInstanceId(entity=entity, key=key)
-        except ValueError as ex:
-            raise ValueError(f"Invalid entity ID format: {entity_id}", ex)