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

Add custom HTTP method

This commit is contained in:
beeankha 2019-06-21 15:06:59 -04:00
parent 5071e1c75f
commit fbb3fd2799

View File

@ -3,7 +3,6 @@
import logging
import requests
import base64
from django.utils.encoding import smart_text
from django.utils.translation import ugettext_lazy as _
@ -16,6 +15,7 @@ logger = logging.getLogger('awx.main.notifications.webhook_backend')
class WebhookBackend(AWXBaseEmailBackend):
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},
"username": {"label": "Username", "type": "string", "default": ""},
"password": {"label": "Password", "type": "password", "default": ""},
@ -23,7 +23,8 @@ class WebhookBackend(AWXBaseEmailBackend):
recipient_parameter = "url"
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.headers = headers
self.username = username
@ -37,15 +38,19 @@ class WebhookBackend(AWXBaseEmailBackend):
sent_messages = 0
if 'User-Agent' not in self.headers:
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:
r = requests.post("{}".format(m.recipients()[0]),
json=m.body,
headers=self.headers,
verify=(not self.disable_ssl_verification))
if r.status_code >= 400:
logger.error(smart_text(_("Error sending notification webhook: {}").format(r.text)))
if not self.fail_silently:
raise Exception(smart_text(_("Error sending notification webhook: {}").format(r.text)))
sent_messages += 1
if chosen_method is not None:
r = chosen_method("{}".format(m.recipients()[0]),
auth=(self.username, self.password),
json=m.body,
headers=self.headers,
verify=(not self.disable_ssl_verification))
if r.status_code >= 400:
logger.error(smart_text(_("Error sending notification webhook: {}").format(r.text)))
if not self.fail_silently:
raise Exception(smart_text(_("Error sending notification webhook: {}").format(r.text)))
sent_messages += 1
return sent_messages