From 01bb32ebb03958aa4704f1240f033aba05cbf02f Mon Sep 17 00:00:00 2001 From: AlanCoding Date: Fri, 6 Sep 2019 16:04:10 -0400 Subject: [PATCH] Deal with limit prompting in factory --- awxkit/awxkit/api/pages/workflow_job_templates.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/awxkit/awxkit/api/pages/workflow_job_templates.py b/awxkit/awxkit/api/pages/workflow_job_templates.py index 4bbee6778a..52c89d8eea 100644 --- a/awxkit/awxkit/api/pages/workflow_job_templates.py +++ b/awxkit/awxkit/api/pages/workflow_job_templates.py @@ -34,7 +34,12 @@ class WorkflowJobTemplate(HasCopy, HasCreate, HasNotifications, HasSurvey, Unifi payload = PseudoNamespace(name=kwargs.get('name') or 'WorkflowJobTemplate - {}'.format(random_title()), description=kwargs.get('description') or random_title(10)) - optional_fields = ("allow_simultaneous", "ask_variables_on_launch", "survey_enabled") + optional_fields = ( + "allow_simultaneous", + "ask_variables_on_launch", "ask_inventory_on_launch", "ask_scm_branch_on_launch", "ask_limit_on_launch", + "limit", "scm_branch", + "survey_enabled" + ) update_payload(payload, optional_fields, kwargs) extra_vars = kwargs.get('extra_vars', not_provided) @@ -48,9 +53,6 @@ class WorkflowJobTemplate(HasCopy, HasCreate, HasNotifications, HasSurvey, Unifi if kwargs.get('inventory'): payload.inventory = kwargs.get('inventory').id - for field_name in ('ask_inventory_on_launch', 'limit', 'scm_branch', 'ask_scm_branch_on_launch'): - if field_name in kwargs: - setattr(payload, field_name, kwargs.get(field_name)) return payload