summaryrefslogtreecommitdiffstats
path: root/ansible_collections/community/general/plugins/callback/hipchat.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-13 12:04:41 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-13 12:04:41 +0000
commit975f66f2eebe9dadba04f275774d4ab83f74cf25 (patch)
tree89bd26a93aaae6a25749145b7e4bca4a1e75b2be /ansible_collections/community/general/plugins/callback/hipchat.py
parentInitial commit. (diff)
downloadansible-975f66f2eebe9dadba04f275774d4ab83f74cf25.tar.xz
ansible-975f66f2eebe9dadba04f275774d4ab83f74cf25.zip
Adding upstream version 7.7.0+dfsg.upstream/7.7.0+dfsg
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'ansible_collections/community/general/plugins/callback/hipchat.py')
-rw-r--r--ansible_collections/community/general/plugins/callback/hipchat.py229
1 files changed, 229 insertions, 0 deletions
diff --git a/ansible_collections/community/general/plugins/callback/hipchat.py b/ansible_collections/community/general/plugins/callback/hipchat.py
new file mode 100644
index 000000000..3e10b69e7
--- /dev/null
+++ b/ansible_collections/community/general/plugins/callback/hipchat.py
@@ -0,0 +1,229 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) 2014, Matt Martz <matt@sivel.net>
+# Copyright (c) 2017 Ansible Project
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+DOCUMENTATION = '''
+ author: Unknown (!UNKNOWN)
+ name: hipchat
+ type: notification
+ requirements:
+ - whitelist in configuration.
+ - prettytable (python lib)
+ short_description: post task events to hipchat
+ description:
+ - This callback plugin sends status updates to a HipChat channel during playbook execution.
+ - Before 2.4 only environment variables were available for configuring this plugin.
+ options:
+ token:
+ description: HipChat API token for v1 or v2 API.
+ required: true
+ env:
+ - name: HIPCHAT_TOKEN
+ ini:
+ - section: callback_hipchat
+ key: token
+ api_version:
+ description: HipChat API version, v1 or v2.
+ required: false
+ default: v1
+ env:
+ - name: HIPCHAT_API_VERSION
+ ini:
+ - section: callback_hipchat
+ key: api_version
+ room:
+ description: HipChat room to post in.
+ default: ansible
+ env:
+ - name: HIPCHAT_ROOM
+ ini:
+ - section: callback_hipchat
+ key: room
+ from:
+ description: Name to post as
+ default: ansible
+ env:
+ - name: HIPCHAT_FROM
+ ini:
+ - section: callback_hipchat
+ key: from
+ notify:
+ description: Add notify flag to important messages
+ type: bool
+ default: true
+ env:
+ - name: HIPCHAT_NOTIFY
+ ini:
+ - section: callback_hipchat
+ key: notify
+
+'''
+
+import os
+import json
+
+try:
+ import prettytable
+ HAS_PRETTYTABLE = True
+except ImportError:
+ HAS_PRETTYTABLE = False
+
+from ansible.plugins.callback import CallbackBase
+from ansible.module_utils.six.moves.urllib.parse import urlencode
+from ansible.module_utils.urls import open_url
+
+
+class CallbackModule(CallbackBase):
+ """This is an example ansible callback plugin that sends status
+ updates to a HipChat channel during playbook execution.
+ """
+
+ CALLBACK_VERSION = 2.0
+ CALLBACK_TYPE = 'notification'
+ CALLBACK_NAME = 'community.general.hipchat'
+ CALLBACK_NEEDS_WHITELIST = True
+
+ API_V1_URL = 'https://api.hipchat.com/v1/rooms/message'
+ API_V2_URL = 'https://api.hipchat.com/v2/'
+
+ def __init__(self):
+
+ super(CallbackModule, self).__init__()
+
+ if not HAS_PRETTYTABLE:
+ self.disabled = True
+ self._display.warning('The `prettytable` python module is not installed. '
+ 'Disabling the HipChat callback plugin.')
+ self.printed_playbook = False
+ self.playbook_name = None
+ self.play = 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)
+
+ self.token = self.get_option('token')
+ self.api_version = self.get_option('api_version')
+ self.from_name = self.get_option('from')
+ self.allow_notify = self.get_option('notify')
+ self.room = self.get_option('room')
+
+ if self.token is None:
+ self.disabled = True
+ self._display.warning('HipChat token could not be loaded. The HipChat '
+ 'token can be provided using the `HIPCHAT_TOKEN` '
+ 'environment variable.')
+
+ # Pick the request handler.
+ if self.api_version == 'v2':
+ self.send_msg = self.send_msg_v2
+ else:
+ self.send_msg = self.send_msg_v1
+
+ def send_msg_v2(self, msg, msg_format='text', color='yellow', notify=False):
+ """Method for sending a message to HipChat"""
+
+ headers = {'Authorization': 'Bearer %s' % self.token, 'Content-Type': 'application/json'}
+
+ body = {}
+ body['room_id'] = self.room
+ body['from'] = self.from_name[:15] # max length is 15
+ body['message'] = msg
+ body['message_format'] = msg_format
+ body['color'] = color
+ body['notify'] = self.allow_notify and notify
+
+ data = json.dumps(body)
+ url = self.API_V2_URL + "room/{room_id}/notification".format(room_id=self.room)
+ try:
+ response = open_url(url, data=data, headers=headers, method='POST')
+ return response.read()
+ except Exception as ex:
+ self._display.warning('Could not submit message to hipchat: {0}'.format(ex))
+
+ def send_msg_v1(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(self.allow_notify and notify)
+
+ url = ('%s?auth_token=%s' % (self.API_V1_URL, self.token))
+ try:
+ response = open_url(url, data=urlencode(params))
+ return response.read()
+ except Exception as ex:
+ self._display.warning('Could not submit message to hipchat: {0}'.format(ex))
+
+ def v2_playbook_on_play_start(self, play):
+ """Display Playbook and play start messages"""
+
+ self.play = play
+ name = play.name
+ # 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, dummy = 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, name))
+
+ 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)