Add a Domain to WorkflowExecution objects

This will be needed later for finding an activity type for instance.
This commit is contained in:
Jean-Baptiste Barth 2015-10-25 11:30:11 +01:00
commit 53630dc061
4 changed files with 50 additions and 29 deletions

View file

@ -148,7 +148,7 @@ class SWFBackend(BaseBackend):
wf_type = domain.get_type("workflow", workflow_name, workflow_version)
if wf_type.status == "DEPRECATED":
raise SWFTypeDeprecatedFault(wf_type)
wfe = WorkflowExecution(wf_type, workflow_id,
wfe = WorkflowExecution(domain, wf_type, workflow_id,
tag_list=tag_list, **kwargs)
domain.add_workflow_execution(wfe)
wfe.start()

View file

@ -39,7 +39,8 @@ class WorkflowExecution(object):
"CancelWorkflowExecution"
]
def __init__(self, workflow_type, workflow_id, **kwargs):
def __init__(self, domain, workflow_type, workflow_id, **kwargs):
self.domain = domain
self.workflow_id = workflow_id
self.run_id = uuid.uuid4().hex
# WorkflowExecutionInfo