channel_output_webhook.py

"""
Output channel that sends to webhooks.
"""
import settings
import pprint

from twisted.python import log
import requests
import simplejson

from channel import OutputChannel
from constants import OUTPUT_CHANNEL_WEBHOOK

class WebhookOutputChannel(OutputChannel):
    CHANNEL = OUTPUT_CHANNEL_WEBHOOK


    def do_send_alert(self, input_channel=None, canarydrop=None, **kwargs):
        try:
            payload = input_channel.format_webhook_canaryalert(
                                          canarydrop=canarydrop,
                                          **kwargs)
            self.generic_webhook_send(simplejson.dumps(payload), canarydrop)
        except Exception as e:
            log.err(e)
    def generic_webhook_send(self, payload=None, canarydrop=None):
        try:
            response = requests.post(canarydrop['alert_webhook_url'], payload, headers={'content-type': 'application/json'})
            response.raise_for_status()
            log.msg('Webhook sent to {url}'.format(url=canarydrop['alert_webhook_url']))
            return None
        except requests.exceptions.RequestException as e:
            log.err("Failed sending request to webhook {url} with error {error}".format(url=canarydrop['alert_webhook_url'],error=e))
            return e