External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-05 02:31:03 -03:30
Code Issues Packages Projects Releases Wiki Activity

support slack color notification #1490

Browse Source 
Signed-off-by: paihu <paihu_j@yahoo.co.jp>
This commit is contained in:
paihu
2018-03-12 11:33:37 +09:00
parent dcab97f94f
commit 9b5e088d70
3 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
awx/main/notifications/slack_backend.py
View File

@@ -16,13 +16,17 @@ WEBSOCKET_TIMEOUT = 30
class SlackBackend(AWXBaseEmailBackend):
init_parameters = {"token": {"label": "Token", "type": "password"},
"hex_color": {"label": "Notification Color", "type": "string"},
"channels": {"label": "Destination Channels", "type": "list"}}
recipient_parameter = "channels"
sender_parameter = None
def __init__(self, token, fail_silently=False, **kwargs):
def __init__(self, token, hex_color, fail_silently=False, **kwargs):
super(SlackBackend, self).__init__(fail_silently=fail_silently)
self.token = token
self.color = None
if hex_color.startswith("#") and (len(hex_color) == 4 or len(hex_color) == 7):
self.color = hex_color
self.connection = None
def open(self):
@@ -51,6 +55,37 @@ class SlackBackend(AWXBaseEmailBackend):
self.connection = None
def send_messages(self, messages):
if self.color:
return self._send_attachments(messages)
else:
return self._send_rtm_messages(messages)
def _send_attachments(self, messages):
connection = SlackClient(self.token)
sent_messages = 0
for m in messages:
try:
for r in m.recipients():
if r.startswith('#'):
r = r[1:]
ret = connection.api_call("chat.postMessage",
channel=r,
attachments=[{
"color": self.color,
"text": m.subject
}])
logger.debug(ret)
if ret['ok']:
sent_messages += 1
else:
raise RuntimeError("Slack Notification unable to send {}: {}".format(r, m.subject))
except Exception as e:
logger.error(smart_text(_("Exception sending messages: {}").format(e)))
if not self.fail_silently:
raise
return sent_messages
def _send_rtm_messages(self, messages):
if self.connection is None:
self.open()
sent_messages = 0