Update plugins/callback/job_status.py

This commit is contained in:
Chris Hammer 2023-12-01 13:19:22 -05:00
parent 24593f012b
commit 82dae3091f

View File

@ -15,6 +15,27 @@ class CallbackModule(CallbackBase):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(CallbackModule, self).__init__() 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): def v2_playbook_on_start(self, playbook):
@ -25,18 +46,10 @@ class CallbackModule(CallbackBase):
def v2_playbook_on_play_start(self,play): def v2_playbook_on_play_start(self,play):
self._display.display("Custom commands for <v2_playbook_on_play_start> go here", color=C.COLOR_OK) self._display.display("Custom commands for <v2_playbook_on_play_start> 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): def v2_playbook_on_task_start(self, task, is_conditional):
self._display.display("Custom commands for <v2_playbook_on_task_start> go here", color=C.COLOR_OK) self._display.display("Custom commands for <v2_playbook_on_task_start> 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: job_status_token > %s" % (self.job_status_token))
self._display.display("EXTRA_VAR: callbacks_enabled > %s" % (self.callbacks_enabled)) self._display.display("EXTRA_VAR: callbacks_enabled > %s" % (self.callbacks_enabled))