1
0
mirror of https://github.com/ansible/awx.git synced 2024-10-30 05:25:29 +03:00

Working version of callback module to log to database.

This commit is contained in:
Chris Church 2013-03-31 19:24:53 -04:00
parent c149f03119
commit 3c7cac00f9
2 changed files with 79 additions and 22 deletions

View File

@ -78,10 +78,16 @@ class RunLaunchJobTest(BaseCeleryTest):
launch_job_status = self.launch_job.start()
self.assertEqual(launch_job_status.status, 'pending')
launch_job_status = LaunchJobStatus.objects.get(pk=launch_job_status.pk)
self.assertEqual(launch_job_status.status, 'successful')
self.assertTrue(launch_job_status.result_stdout)
#print 'stdout:', launch_job_status.result_stdout
#print 'stderr:', launch_job_status.result_stderr
self.assertEqual(launch_job_status.status, 'successful')
self.assertTrue(launch_job_status.result_stdout)
launch_job_status_events = launch_job_status.launch_job_status_events.all()
#for ev in launch_job_status_events:
# print ev.event, ev.event_data
self.assertEqual(launch_job_status_events.filter(event='playbook_on_start').count(), 1)
self.assertEqual(launch_job_status_events.filter(event='playbook_on_play_start').count(), 1)
self.assertEqual(launch_job_status_events.filter(event='playbook_on_task_start').count(), 1)
self.assertEqual(launch_job_status_events.filter(event='playbook_on_stats').count(), 1)
# FIXME: Test with a task that fails.
# FIXME: Get callback working.

View File

@ -16,44 +16,87 @@
# along with Ansible Commander. If not, see <http://www.gnu.org/licenses/>.
import os
import sys
class CallbackModule(object):
'''
Stub callback module for logging ansible-playbook events.
Callback module for logging ansible-playbook events to the database.
'''
def _log_event(self, event, *args, **kwargs):
# FIXME: Push these events back to the server.
pass#print '====', event, args, kwargs
def __init__(self):
# he DJANGO_SETTINGS_MODULE environment variable *should* already
# be set if this callback is called when executing a playbook via a
# celery task, otherwise just bail out.
settings_module_name = os.environ.get('DJANGO_SETTINGS_MODULE', None)
if not settings_module_name:
return
# FIXME: Not particularly fond of this sys.path hack, but it is needed
# when a celery task calls ansible-playbook and needs to execute this
# script directly.
try:
settings_parent_module = __import__(settings_module_name)
except ImportError:
top_dir = os.path.join(os.path.dirname(__file__), '..', '..', '..')
sys.path.insert(0, os.path.abspath(top_dir))
settings_parent_module = __import__(settings_module_name)
settings_module = getattr(settings_parent_module, settings_module_name.split('.')[-1])
# Use the ACOM_TEST_DATABASE_NAME environment variable to specify the test
# database name when called from unit tests.
if os.environ.get('ACOM_TEST_DATABASE_NAME', None):
settings_module.DATABASES['default']['NAME'] = os.environ['ACOM_TEST_DATABASE_NAME']
# Try to get the launch job status ID from the environment, otherwise
# just bail out now.
try:
launch_job_status_pk = int(os.environ.get('ACOM_LAUNCH_JOB_STATUS_ID', ''))
except ValueError:
return
from lib.main.models import LaunchJobStatus
try:
self.launch_job_status = LaunchJobStatus.objects.get(pk=launch_job_status_pk)
except LaunchJobStatus.DoesNotExist:
pass
def _log_event(self, event, **event_data):
#print '====', event, args, kwargs
if hasattr(self, 'launch_job_status'):
kwargs = {
'event': event,
'event_data': event_data,
}
self.launch_job_status.launch_job_status_events.create(**kwargs)
def on_any(self, *args, **kwargs):
pass
def runner_on_failed(self, host, res, ignore_errors=False):
self._log_event('runner_on_failed', host, res, ignore_errors)
self._log_event('runner_on_failed', host=host, res=res,
ignore_errors=ignore_errors)
def runner_on_ok(self, host, res):
self._log_event('runner_on_ok', host, res)
self._log_event('runner_on_ok', host=host, res=res)
def runner_on_error(self, host, msg):
self._log_event('runner_on_error', host, msg)
self._log_event('runner_on_error', host=host, msg=msg)
def runner_on_skipped(self, host, item=None):
self._log_event('runner_on_skipped', host, item)
self._log_event('runner_on_skipped', host=host, item=item)
def runner_on_unreachable(self, host, res):
self._log_event('runner_on_unreachable', host, res)
self._log_event('runner_on_unreachable', host=host, res=res)
def runner_on_no_hosts(self):
self._log_event('runner_on_no_hosts')
def runner_on_async_poll(self, host, res, jid, clock):
self._log_event('runner_on_async_poll', host, res, jid, clock)
self._log_event('runner_on_async_poll', host=host, res=res, jid=jid,
clock=clock)
def runner_on_async_ok(self, host, res, jid):
self._log_event('runner_on_async_ok', host, res, jid)
self._log_event('runner_on_async_ok', host=host, res=res, jid=jid)
def runner_on_async_failed(self, host, res, jid):
self._log_event('runner_on_async_failed', host, res, jid)
self._log_event('runner_on_async_failed', host=host, res=res, jid=jid)
def playbook_on_start(self):
self._log_event('playbook_on_start')
@ -62,25 +105,33 @@ class CallbackModule(object):
self._log_event('playbook_on_notify')
def playbook_on_task_start(self, name, is_conditional):
self._log_event('playbook_on_task_start', name, is_conditional)
self._log_event('playbook_on_task_start', name=name,
is_conditional=is_conditional)
def playbook_on_vars_prompt(self, varname, private=True, prompt=None, encrypt=None, confirm=False, salt_size=None, salt=None, default=None):
self._log_event('playbook_on_vars_prompt', varname, private, prompt, encrypt, confirm, salt_size, salt, default)
def playbook_on_vars_prompt(self, varname, private=True, prompt=None,
encrypt=None, confirm=False, salt_size=None,
salt=None, default=None):
self._log_event('playbook_on_vars_prompt', varname=varname,
private=private, prompt=prompt, encrypt=encrypt,
confirm=confirm, salt_size=salt_size, salt=salf,
default=default)
def playbook_on_setup(self):
self._log_event('playbook_on_setup')
def playbook_on_import_for_host(self, host, imported_file):
self._log_event('playbook_on_import_for_host', host, imported_file)
self._log_event('playbook_on_import_for_host', host=host,
imported_file=imported_file)
def playbook_on_not_import_for_host(self, host, missing_file):
self._log_event('playbook_on_not_import_for_host', host, missing_file)
self._log_event('playbook_on_not_import_for_host', host=host,
missing_file=missing_file)
def playbook_on_play_start(self, pattern):
self._log_event('playbook_on_play_start', pattern)
self._log_event('playbook_on_play_start', pattern=pattern)
def playbook_on_stats(self, stats):
d = {}
for attr in ('changed', 'dark', 'failures', 'ok', 'processed', 'skipped'):
d[attr] = getattr(stats, attr)
self._log_event('playbook_on_stats', d)
self._log_event('playbook_on_stats', **d)