1 files changed, 135 insertions, 0 deletions
diff --git a/azure/aria/aria-extension-cloudify/src/aria/aria/orchestrator/context/workflow.py b/azure/aria/aria-extension-cloudify/src/aria/aria/orchestrator/context/workflow.py
new file mode 100644
index 0000000..738d2fd
--- /dev/null
+++ b/azure/aria/aria-extension-cloudify/src/aria/aria/orchestrator/context/workflow.py
@@ -0,0 +1,135 @@
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with
+# the License.  You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Workflow context.
+"""
+
+import threading
+from contextlib import contextmanager
+
+from .exceptions import ContextException
+from .common import BaseContext
+
+
+class WorkflowContext(BaseContext):
+    """
+    Context used during workflow creation and execution.
+    """
+    def __init__(self,
+                 workflow_name,
+                 parameters=None,
+                 task_max_attempts=1,
+                 task_retry_interval=0,
+                 task_ignore_failure=False,
+                 *args, **kwargs):
+        super(WorkflowContext, self).__init__(*args, **kwargs)
+        self._workflow_name = workflow_name
+        self._parameters = parameters or {}
+        self._task_max_attempts = task_max_attempts
+        self._task_retry_interval = task_retry_interval
+        self._task_ignore_failure = task_ignore_failure
+        self._execution_graph = None
+        self._register_logger()
+
+    def __repr__(self):
+        return (
+            '{name}(deployment_id={self._service_id}, '
+            'workflow_name={self._workflow_name}, execution_id={self._execution_id})'.format(
+                name=self.__class__.__name__, self=self))
+
+    @property
+    def workflow_name(self):
+        return self._workflow_name
+
+    @property
+    def execution(self):
+        """
+        Execution model.
+        """
+        return self.model.execution.get(self._execution_id)
+
+    @execution.setter
+    def execution(self, value):
+        """
+        Stores the execution in the storage model API ("MAPI").
+        """
+        self.model.execution.put(value)
+
+    @property
+    def node_templates(self):
+        """
+        Iterates over nodes templates.
+        """
+        key = 'service_{0}'.format(self.model.node_template.model_cls.name_column_name())
+
+        return self.model.node_template.iter(
+            filters={
+                key: getattr(self.service, self.service.name_column_name())
+            }
+        )
+
+    @property
+    def nodes(self):
+        """
+        Iterates over nodes.
+        """
+        key = 'service_{0}'.format(self.model.node.model_cls.name_column_name())
+        return self.model.node.iter(
+            filters={
+                key: getattr(self.service, self.service.name_column_name())
+            }
+        )
+
+    @property
+    @contextmanager
+    def persist_changes(self):
+        yield
+        self._model.execution.update(self.execution)
+
+
+class _CurrentContext(threading.local):
+    """
+    Provides a thread-level context, with sugar for the task MAPI.
+    """
+
+    def __init__(self):
+        super(_CurrentContext, self).__init__()
+        self._workflow_context = None
+
+    def _set(self, value):
+        self._workflow_context = value
+
+    def get(self):
+        """
+        Retrieves the current workflow context.
+        """
+        if self._workflow_context is not None:
+            return self._workflow_context
+        raise ContextException("No context was set")
+
+    @contextmanager
+    def push(self, workflow_context):
+        """
+        Switches the current context to the provided context.
+        """
+        prev_workflow_context = self._workflow_context
+        self._set(workflow_context)
+        try:
+            yield self
+        finally:
+            self._set(prev_workflow_context)
+
+current = _CurrentContext()