Allow setting Lambda role

Unfortunately, boto2 doesn't support passing lambdaRole; we must
override two of its methods (register type and start workflow).

Signed-off-by: Yves Bastide <yves@botify.com>

Author: ybastide

simpleflow/command.py

@@ -151,6 +151,9 @@ def transform_input(wf_input):
               type=comma_separated_list,
               required=False,
               help='Tags for the workflow execution.')
+@click.option('--lambda-role',
+              required=False,
+              help='Lambda role.')
 @click.option('--decision-tasks-timeout',
               required=False,
               help='Timeout for the decision tasks.')
@@ -175,6 +178,7 @@ def start_workflow(workflow,
                    task_list,
                    execution_timeout,
                    tags,
+                   lambda_role,
                    decision_tasks_timeout,
                    input,
                    input_file,
@@ -200,6 +204,7 @@ def start_workflow(workflow,
         execution_timeout=execution_timeout,
         input=wf_input,
         tag_list=tags,
+        lambda_role=lambda_role or workflow_class.lambda_role,
         decision_tasks_timeout=decision_tasks_timeout,
     )
     print('{workflow_id} {run_id}'.format(
@@ -456,6 +461,9 @@ def create_unique_task_list(workflow_id=''):
               type=comma_separated_list,
               required=False,
               help='Tags identifying the workflow execution.')
+@click.option('--lambda-role',
+              required=False,
+              help='Lambda role.')
 @click.option('--decision-tasks-timeout',
               required=False,
               help='Decision tasks timeout.')
@@ -490,6 +498,7 @@ def standalone(context,
                workflow_id,
                execution_timeout,
                tags,
+               lambda_role,
                decision_tasks_timeout,
                input,
                input_file,
@@ -579,6 +588,7 @@ def standalone(context,
         task_list,
         execution_timeout,
         tags,
+        lambda_role,
         decision_tasks_timeout,
         json_dumps(wf_input),
         None,

simpleflow/workflow.py

@@ -31,6 +31,7 @@ class Workflow(Submittable):
     version = None
     task_list = None
     task_priority = None
+    lambda_role = None
 
     def __init__(self, executor):
         self._executor = executor

swf/models/decision/workflow.py

@@ -57,8 +57,9 @@ def terminate(self, reason=None, details=None):
     def continue_as_new(self, child_policy=None,
                         execution_timeout=None, task_timeout=None,
                         input=None, tag_list=None, task_list=None,
+                        lambda_role=None,
                         workflow_type_version=None):
-        """Coninue as new workflow execution decision builder
+        """Continue as new workflow execution decision builder
         :param  child_policy: specifies the policy to use for the
                               child workflow executions of the new execution
         :type   child_policy: CHILD_POLICIES.{TERMINATE | REQUEST_CANCEL | ABANDON}
@@ -75,6 +76,9 @@ def continue_as_new(self, child_policy=None,
         :param  task_list: task list name
         :type   task_list: str
 
+        :param lambda_role: Lambda role
+        :type lambda_role: str
+
         :param  task_timeout: maximum duration of decision tasks for the new workflow execution
         :type   task_timeout: str
 
@@ -91,6 +95,7 @@ def continue_as_new(self, child_policy=None,
             'input': input,
             'tagList': tag_list,
             'taskList': task_list,
+            'lambdaRole': lambda_role,
             'workflowTypeVersion': workflow_type_version,
         })
 
@@ -101,7 +106,7 @@ class ChildWorkflowExecutionDecision(Decision):
     @decision_action
     def start(self, workflow_type, workflow_id, child_policy=CHILD_POLICIES.TERMINATE,
               execution_timeout='300', task_timeout='300', control=None,
-              input=None, tag_list=None, task_list=None):
+              input=None, tag_list=None, task_list=None, lambda_role=None):
         """Child workflow execution decision builder
 
         :param  workflow_type: workflow type to start
@@ -117,6 +122,12 @@ def start(self, workflow_type, workflow_id, child_policy=CHILD_POLICIES.TERMINAT
         :param  execution_timeout: specifies the total duration for this workflow execution
         :type   execution_timeout: str
 
+        :param  task_timeout: maximum duration of decision tasks for the child workflow execution
+        :type   task_timeout: str
+
+        :param control: data attached to the event that can be used by the decider in subsequent workflow tasks
+        :type control: Optional[str]
+
         :param  input: The input provided to the child workflow execution
         :type   input: str
 
@@ -126,8 +137,8 @@ def start(self, workflow_type, workflow_id, child_policy=CHILD_POLICIES.TERMINAT
         :param  task_list: task list name
         :type   task_list: str
 
-        :param  task_timeout: maximum duration of decision tasks for the child workflow execution
-        :type   task_timeout: str
+        :param lambda_role: workflow lambda role
+        :type lambda_role: Optional[str]
 
         """
         if input is not None:
@@ -143,6 +154,7 @@ def start(self, workflow_type, workflow_id, child_policy=CHILD_POLICIES.TERMINAT
             'taskList': {
                 'name': task_list,
             },
+            'lambdaRole': lambda_role,
             'workflowId': workflow_id,
             'workflowType': {
                 'name': workflow_type.name,