diff --git a/awxkit/awxkit/api/pages/workflow_job_template_nodes.py b/awxkit/awxkit/api/pages/workflow_job_template_nodes.py index dbf6d53505..01e0b41e0a 100644 --- a/awxkit/awxkit/api/pages/workflow_job_template_nodes.py +++ b/awxkit/awxkit/api/pages/workflow_job_template_nodes.py @@ -31,6 +31,7 @@ class WorkflowJobTemplateNode(HasCreate, base.Base): 'skip_tags', 'verbosity', 'extra_data', + 'identifier', 'all_parents_must_converge') update_payload(payload, optional_fields, kwargs) @@ -74,21 +75,21 @@ class WorkflowJobTemplateNode(HasCreate, base.Base): WorkflowJobTemplateNodes( self.connection).post(payload)) - def _add_node(self, endpoint, unified_job_template): + def _add_node(self, endpoint, unified_job_template, **kwargs): node = endpoint.post( - dict(unified_job_template=unified_job_template.id)) + dict(unified_job_template=unified_job_template.id, **kwargs)) node.create_and_update_dependencies( self.ds.workflow_job_template, unified_job_template) return node - def add_always_node(self, unified_job_template): - return self._add_node(self.related.always_nodes, unified_job_template) + def add_always_node(self, unified_job_template, **kwargs): + return self._add_node(self.related.always_nodes, unified_job_template, **kwargs) - def add_failure_node(self, unified_job_template): - return self._add_node(self.related.failure_nodes, unified_job_template) + def add_failure_node(self, unified_job_template, **kwargs): + return self._add_node(self.related.failure_nodes, unified_job_template, **kwargs) - def add_success_node(self, unified_job_template): - return self._add_node(self.related.success_nodes, unified_job_template) + def add_success_node(self, unified_job_template, **kwargs): + return self._add_node(self.related.success_nodes, unified_job_template, **kwargs) def add_credential(self, credential): with suppress(exc.NoContent):