From 82dae3091ffc530da481f9a5624bf727a2255ebf Mon Sep 17 00:00:00 2001 From: chris Date: Fri, 1 Dec 2023 13:19:22 -0500 Subject: [PATCH] Update plugins/callback/job_status.py --- plugins/callback/job_status.py | 31 ++++++++++++++++++++++--------- 1 file changed, 22 insertions(+), 9 deletions(-) diff --git a/plugins/callback/job_status.py b/plugins/callback/job_status.py index 2ecee7e..a4aec33 100644 --- a/plugins/callback/job_status.py +++ b/plugins/callback/job_status.py @@ -15,6 +15,27 @@ class CallbackModule(CallbackBase): def __init__(self, *args, **kwargs): super(CallbackModule, self).__init__() + self.playbook_status_url = None + self.playbook_status_token = None + self.playbook_status_headers = None + self.playbook_status_action = None + + + def set_options(self, task_keys=None, var_options=None, direct=None): + super(CallbackModule, self).set_options(task_keys=task_keys, var_options=var_options, direct=direct) + + # Setup and use extra_vars from custom credential: + vm = play.get_variable_manager() + extra_vars = vm.extra_vars + + self.job_status_url = extra_vars['job_status_url'] + self.job_status_token = extra_vars['job_status_token'] + self.callbacks_enabled = extra_vars['callbacks_enabled'] + + self.playbook_status_url = self.job_status_url + self.playbook_status_token = self.job_status_token + self.playbook_status_headers = { 'Authorization': "Token %s" % self.playbook_status_token, 'Content-type': 'application/json' } + self.playbook_status_action = self.get_option('playbook_status_action') def v2_playbook_on_start(self, playbook): @@ -25,18 +46,10 @@ class CallbackModule(CallbackBase): def v2_playbook_on_play_start(self,play): self._display.display("Custom commands for go here", color=C.COLOR_OK) - # Setup and use extra_vars from custom credential: - vm = play.get_variable_manager() - extra_vars = vm.extra_vars - - self.job_status_url = extra_vars['job_status_url'] - self.job_status_token = extra_vars['job_status_token'] - self.callbacks_enabled = extra_vars['callbacks_enabled'] - def v2_playbook_on_task_start(self, task, is_conditional): self._display.display("Custom commands for go here", color=C.COLOR_OK) - self._display.display("EXTRA_VAR: job_status_url > %s" % (self.job_status_url)) + self._display.display("EXTRA_VAR: job_status_url > %s %s" % (self.job_status_url, self.playbook_status_url)) self._display.display("EXTRA_VAR: job_status_token > %s" % (self.job_status_token)) self._display.display("EXTRA_VAR: callbacks_enabled > %s" % (self.callbacks_enabled))