diff --git a/plugins/callbacks/hipchat.py b/plugins/callbacks/hipchat.py new file mode 100644 index 0000000000..ee4aca6aa3 --- /dev/null +++ b/plugins/callbacks/hipchat.py @@ -0,0 +1,204 @@ +# (C) 2014, Matt Martz + +# This file is part of Ansible +# +# Ansible is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Ansible is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Ansible. If not, see . + +import os +import urllib +import urllib2 + +from ansible import utils + +try: + import prettytable + HAS_PRETTYTABLE = True +except ImportError: + HAS_PRETTYTABLE = False + + +class CallbackModule(object): + """This is an example ansible callback plugin that sends status + updates to a HipChat channel during playbook execution. + + This plugin makes use of the following environment variables: + HIPCHAT_TOKEN (required): HipChat API token + HIPCHAT_ROOM (optional): HipChat room to post in. Default: ansible + HIPCHAT_FROM (optional): Name to post as. Default: ansible + + Requires: + prettytable + + """ + + def __init__(self): + if not HAS_PRETTYTABLE: + self.disabled = True + utils.warning('The `prettytable` python module is not installed. ' + 'Disabling the HipChat callback plugin.') + + self.msg_uri = 'https://api.hipchat.com/v1/rooms/message' + self.token = os.getenv('HIPCHAT_TOKEN') + self.room = os.getenv('HIPCHAT_ROOM', 'ansible') + self.from_name = os.getenv('HIPCHAT_FROM', 'ansible') + + if self.token is None: + self.disabled = True + utils.warning('HipChat token could not be loaded. The HipChat ' + 'token can be provided using the `HIPCHAT_TOKEN` ' + 'environment variable.') + + self.printed_playbook = False + self.playbook_name = None + + def send_msg(self, msg, msg_format='text', color='yellow', notify=False): + """Method for sending a message to HipChat""" + + params = {} + params['room_id'] = self.room + params['from'] = self.from_name[:15] # max length is 15 + params['message'] = msg + params['message_format'] = msg_format + params['color'] = color + params['notify'] = int(notify) + + url = ('%s?auth_token=%s' % (self.msg_uri, self.token)) + try: + response = urllib2.urlopen(url, urllib.urlencode(params)) + return response.read() + except: + utils.warning('Could not submit message to hipchat') + + def on_any(self, *args, **kwargs): + pass + + def runner_on_failed(self, host, res, ignore_errors=False): + pass + + def runner_on_ok(self, host, res): + pass + + def runner_on_error(self, host, msg): + pass + + def runner_on_skipped(self, host, item=None): + pass + + def runner_on_unreachable(self, host, res): + pass + + def runner_on_no_hosts(self): + pass + + def runner_on_async_poll(self, host, res, jid, clock): + pass + + def runner_on_async_ok(self, host, res, jid): + pass + + def runner_on_async_failed(self, host, res, jid): + pass + + def playbook_on_start(self): + pass + + def playbook_on_notify(self, host, handler): + pass + + def playbook_on_no_hosts_matched(self): + pass + + def playbook_on_no_hosts_remaining(self): + pass + + def playbook_on_task_start(self, name, is_conditional): + pass + + def playbook_on_vars_prompt(self, varname, private=True, prompt=None, + encrypt=None, confirm=False, salt_size=None, + salt=None, default=None): + pass + + def playbook_on_setup(self): + pass + + def playbook_on_import_for_host(self, host, imported_file): + pass + + def playbook_on_not_import_for_host(self, host, missing_file): + pass + + def playbook_on_play_start(self, pattern): + """Display Playbook and play start messages""" + + # This block sends information about a playbook when it starts + # The playbook object is not immediately available at + # playbook_on_start so we grab it via the play + # + # Displays info about playbook being started by a person on an + # inventory, as well as Tags, Skip Tags and Limits + if not self.printed_playbook: + self.playbook_name, _ = os.path.splitext( + os.path.basename(self.play.playbook.filename)) + host_list = self.play.playbook.inventory.host_list + inventory = os.path.basename(os.path.realpath(host_list)) + self.send_msg("%s: Playbook initiated by %s against %s" % + (self.playbook_name, + self.play.playbook.remote_user, + inventory), notify=True) + self.printed_playbook = True + subset = self.play.playbook.inventory._subset + skip_tags = self.play.playbook.skip_tags + self.send_msg("%s:\nTags: %s\nSkip Tags: %s\nLimit: %s" % + (self.playbook_name, + ', '.join(self.play.playbook.only_tags), + ', '.join(skip_tags) if skip_tags else None, + ', '.join(subset) if subset else subset)) + + # This is where we actually say we are starting a play + self.send_msg("%s: Starting play: %s" % + (self.playbook_name, pattern)) + + def playbook_on_stats(self, stats): + """Display info about playbook statistics""" + hosts = sorted(stats.processed.keys()) + + t = prettytable.PrettyTable(['Host', 'Ok', 'Changed', 'Unreachable', + 'Failures']) + + failures = False + unreachable = False + + for h in hosts: + s = stats.summarize(h) + + if s['failures'] > 0: + failures = True + if s['unreachable'] > 0: + unreachable = True + + t.add_row([h] + [s[k] for k in ['ok', 'changed', 'unreachable', + 'failures']]) + + self.send_msg("%s: Playbook complete" % self.playbook_name, + notify=True) + + if failures or unreachable: + color = 'red' + self.send_msg("%s: Failures detected" % self.playbook_name, + color=color, notify=True) + else: + color = 'green' + + self.send_msg("/code %s:\n%s" % (self.playbook_name, t), color=color)