2017-09-30 04:18:16 +02:00
|
|
|
# Webhooks pfor external integrations.
|
2018-05-10 19:34:01 +02:00
|
|
|
from typing import Any, Dict
|
2016-05-25 15:02:02 +02:00
|
|
|
|
2016-06-06 00:43:22 +02:00
|
|
|
from django.http import HttpRequest, HttpResponse
|
2016-05-25 15:02:02 +02:00
|
|
|
|
2017-10-31 04:25:48 +01:00
|
|
|
from zerver.decorator import api_key_only_webhook_view
|
|
|
|
from zerver.lib.request import REQ, has_request_variables
|
2019-02-02 23:53:55 +01:00
|
|
|
from zerver.lib.response import json_success
|
2020-06-11 00:54:34 +02:00
|
|
|
from zerver.lib.webhooks.common import UnexpectedWebhookEventType, check_send_webhook_message
|
2017-05-02 01:00:50 +02:00
|
|
|
from zerver.models import UserProfile
|
2016-03-14 22:12:46 +01:00
|
|
|
|
2018-11-09 21:02:59 +01:00
|
|
|
PINGDOM_TOPIC_TEMPLATE = '{name} status.'
|
2016-03-14 22:12:46 +01:00
|
|
|
|
2019-04-17 21:59:11 +02:00
|
|
|
MESSAGE_TEMPLATE = """
|
|
|
|
Service {service_url} changed its {type} status from {previous_state} to {current_state}:
|
|
|
|
""".strip()
|
|
|
|
|
|
|
|
DESC_TEMPLATE = """
|
|
|
|
|
|
|
|
``` quote
|
|
|
|
{description}
|
|
|
|
```
|
|
|
|
""".rstrip()
|
2016-03-14 22:12:46 +01:00
|
|
|
|
|
|
|
SUPPORTED_CHECK_TYPES = (
|
|
|
|
'HTTP',
|
|
|
|
'HTTP_CUSTOM'
|
|
|
|
'HTTPS',
|
|
|
|
'SMTP',
|
|
|
|
'POP3',
|
|
|
|
'IMAP',
|
|
|
|
'PING',
|
|
|
|
'DNS',
|
|
|
|
'UDP',
|
|
|
|
'PORT_TCP',
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2016-05-12 22:49:36 +02:00
|
|
|
@api_key_only_webhook_view('Pingdom')
|
2016-03-14 22:12:46 +01:00
|
|
|
@has_request_variables
|
2017-12-18 15:09:25 +01:00
|
|
|
def api_pingdom_webhook(request: HttpRequest, user_profile: UserProfile,
|
2018-03-16 22:53:50 +01:00
|
|
|
payload: Dict[str, Any]=REQ(argument_type='body')) -> HttpResponse:
|
2016-03-14 22:12:46 +01:00
|
|
|
check_type = get_check_type(payload)
|
|
|
|
|
|
|
|
if check_type in SUPPORTED_CHECK_TYPES:
|
|
|
|
subject = get_subject_for_http_request(payload)
|
|
|
|
body = get_body_for_http_request(payload)
|
|
|
|
else:
|
2018-05-22 16:46:45 +02:00
|
|
|
raise UnexpectedWebhookEventType('Pingdom', check_type)
|
2016-03-14 22:12:46 +01:00
|
|
|
|
2018-03-16 22:53:50 +01:00
|
|
|
check_send_webhook_message(request, user_profile, subject, body)
|
2016-03-14 22:12:46 +01:00
|
|
|
return json_success()
|
|
|
|
|
|
|
|
|
2018-05-10 19:34:01 +02:00
|
|
|
def get_subject_for_http_request(payload: Dict[str, Any]) -> str:
|
2018-11-09 21:02:59 +01:00
|
|
|
return PINGDOM_TOPIC_TEMPLATE.format(name=payload['check_name'])
|
2016-03-14 22:12:46 +01:00
|
|
|
|
|
|
|
|
2018-05-10 19:34:01 +02:00
|
|
|
def get_body_for_http_request(payload: Dict[str, Any]) -> str:
|
2016-03-14 22:12:46 +01:00
|
|
|
current_state = payload['current_state']
|
|
|
|
previous_state = payload['previous_state']
|
|
|
|
|
|
|
|
data = {
|
|
|
|
'service_url': payload['check_params']['hostname'],
|
|
|
|
'previous_state': previous_state,
|
|
|
|
'current_state': current_state,
|
python: Use trailing commas consistently.
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
2020-04-10 05:23:40 +02:00
|
|
|
'type': get_check_type(payload),
|
2016-03-14 22:12:46 +01:00
|
|
|
}
|
2019-04-17 21:59:11 +02:00
|
|
|
body = MESSAGE_TEMPLATE.format(**data)
|
2016-03-14 22:12:46 +01:00
|
|
|
if current_state == 'DOWN' and previous_state == 'UP':
|
2019-04-17 21:59:11 +02:00
|
|
|
description = DESC_TEMPLATE.format(description=payload['long_description'])
|
|
|
|
body += description
|
|
|
|
else:
|
2020-06-09 00:25:09 +02:00
|
|
|
body = f'{body[:-1]}.'
|
2019-04-17 21:59:11 +02:00
|
|
|
|
2016-03-14 22:12:46 +01:00
|
|
|
return body
|
|
|
|
|
|
|
|
|
2018-05-10 19:34:01 +02:00
|
|
|
def get_check_type(payload: Dict[str, Any]) -> str:
|
2016-03-14 22:12:46 +01:00
|
|
|
return payload['check_type']
|