shaba/awx
1
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2024-10-31 15:21:13 +03:00
Code Releases Activity

Add custom HTTP method

Browse Source
This commit is contained in:
beeankha 2019-06-21 15:06:59 -04:00
parent 5071e1c75f
commit fbb3fd2799
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
awx/main/notifications/webhook_backend.py
View File

@ -3,7 +3,6 @@
import logging import logging
import requests import requests
import base64
from django.utils.encoding import smart_text from django.utils.encoding import smart_text
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
@ -16,6 +15,7 @@ logger = logging.getLogger('awx.main.notifications.webhook_backend')
class WebhookBackend(AWXBaseEmailBackend): class WebhookBackend(AWXBaseEmailBackend):
init_parameters = {"url": {"label": "Target URL", "type": "string"}, init_parameters = {"url": {"label": "Target URL", "type": "string"},
"method": {"label": "HTTP Method", "type": "string", "default": "POST"},
"disable_ssl_verification": {"label": "Verify SSL", "type": "bool", "default": False}, "disable_ssl_verification": {"label": "Verify SSL", "type": "bool", "default": False},
"username": {"label": "Username", "type": "string", "default": ""}, "username": {"label": "Username", "type": "string", "default": ""},
"password": {"label": "Password", "type": "password", "default": ""}, "password": {"label": "Password", "type": "password", "default": ""},
@ -23,7 +23,8 @@ class WebhookBackend(AWXBaseEmailBackend):
recipient_parameter = "url" recipient_parameter = "url"
sender_parameter = None sender_parameter = None
def __init__(self, headers, disable_ssl_verification=False, fail_silently=False, username=None, password=None, **kwargs): def __init__(self, headers, method, disable_ssl_verification=False, fail_silently=False, username=None, password=None, **kwargs):
self.method = method
self.disable_ssl_verification = disable_ssl_verification self.disable_ssl_verification = disable_ssl_verification
self.headers = headers self.headers = headers
self.username = username self.username = username
@ -37,15 +38,19 @@ class WebhookBackend(AWXBaseEmailBackend):
sent_messages = 0 sent_messages = 0
if 'User-Agent' not in self.headers: if 'User-Agent' not in self.headers:
self.headers['User-Agent'] = "Tower {}".format(get_awx_version()) self.headers['User-Agent'] = "Tower {}".format(get_awx_version())
self.headers['Authorization'] = base64.b64encode("{}:{}".format(self.username, self.password).encode()) if self.method.lower() not in ('put', 'post'):
raise ValueError("Method must be either 'POST' or 'PUT'.")
chosen_method = getattr(requests, self.method.lower(), None)
for m in messages: for m in messages:
r = requests.post("{}".format(m.recipients()[0]), if chosen_method is not None:
json=m.body, r = chosen_method("{}".format(m.recipients()[0]),
headers=self.headers, auth=(self.username, self.password),
verify=(not self.disable_ssl_verification)) json=m.body,
if r.status_code >= 400: headers=self.headers,
logger.error(smart_text(_("Error sending notification webhook: {}").format(r.text))) verify=(not self.disable_ssl_verification))
if not self.fail_silently: if r.status_code >= 400:
raise Exception(smart_text(_("Error sending notification webhook: {}").format(r.text))) logger.error(smart_text(_("Error sending notification webhook: {}").format(r.text)))
sent_messages += 1 if not self.fail_silently:
raise Exception(smart_text(_("Error sending notification webhook: {}").format(r.text)))
sent_messages += 1
return sent_messages return sent_messages