Add orchestration & activity trigger for durable function

Update durable triggers

Author: Hanzhang Zeng (Roger)

azure/functions/__init__.py

@@ -7,6 +7,7 @@
 from ._http_wsgi import WsgiMiddleware # NoQA
 from ._queue import QueueMessage  # NoQA
 from ._servicebus import ServiceBusMessage  # NoQA
+from ._durable_functions import OrchestrationContext  # NoQA
 from .meta import get_binding_registry  # NoQA
 
 # Import binding implementations to register them
@@ -18,6 +19,7 @@
 from . import queue  # NoQA
 from . import servicebus  # NoQA
 from . import timer  # NoQA
+from . import durable_functions  # NoQA
 
 
 __all__ = (
@@ -35,9 +37,12 @@
     'EventHubEvent',
     'HttpRequest',
     'HttpResponse',
-    'WsgiMiddleware',
     'InputStream',
+    'OrchestrationContext',
     'QueueMessage',
     'ServiceBusMessage',
     'TimerRequest',
+
+    # Middlewares
+    'WsgiMiddleware',
 )

azure/functions/_abc.py

@@ -300,3 +300,10 @@ def enqueued_time(self) -> typing.Optional[datetime.datetime]:
     @abc.abstractmethod
     def offset(self) -> typing.Optional[str]:
         pass
+
+
+class OrchestrationContext(abc.ABC):
+    @property
+    @abc.abstractmethod
+    def body(self) -> str:
+        pass

azure/functions/_durable_functions.py

@@ -0,0 +1,23 @@
+from . import _abc
+
+
+class OrchestrationContext(_abc.OrchestrationContext):
+    """A durable function orchestration context.
+
+    :param str body:
+        The body of orchestration context.
+    """
+
+    def __init__(self,
+                 body: str) -> None:
+        self.__body = body
+
+    @property
+    def body(self):
+        return self.__body
+
+    def __repr__(self):
+        return (
+            f'<azure.OrchestrationContext '
+            f'body={self.body}>'
+        )

azure/functions/durable_functions.py

@@ -0,0 +1,45 @@
+from typing import Any
+from azure.functions import _durable_functions
+
+from . import meta
+
+
+# Durable Function Orchestration Trigger
+class OrchestrationTriggerConverter(meta.InConverter,
+                                    binding='orchestrationTrigger',
+                                    trigger=True):
+    @classmethod
+    def check_input_type_annotation(cls, pytype):
+        return (issubclass(pytype, str)
+                or issubclass(pytype, bytes)
+                or issubclass(pytype, _durable_functions.OrchestrationContext))
+
+    @classmethod
+    def decode(cls,
+               data: meta.Datum, *,
+               trigger_metadata) -> _durable_functions.OrchestrationContext:
+        return _durable_functions.OrchestrationContext(data.value)
+
+    @classmethod
+    def has_implicit_output(cls) -> bool:
+        return True
+
+
+# Durable Function Activity Trigger
+class ActivityTriggerConverter(meta.InConverter,
+                               binding='activityTrigger',
+                               trigger=True):
+    @classmethod
+    def check_input_type_annotation(cls, pytype):
+        # Activity Trigger's arguments should accept any types
+        return True
+
+    @classmethod
+    def decode(cls,
+               data: meta.Datum, *,
+               trigger_metadata) -> Any:
+        return data.value
+
+    @classmethod
+    def has_implicit_output(cls) -> bool:
+        return True

azure/functions/meta.py

@@ -266,6 +266,10 @@ def check_input_type_annotation(cls, pytype: type) -> bool:
     def decode(cls, data: Datum, *, trigger_metadata) -> typing.Any:
         raise NotImplementedError
 
+    @abc.abstractclassmethod
+    def has_implicit_output(cls) -> bool:
+        return False
+
 
 class OutConverter(_BaseConverter, binding=None):