2016-05-25 20:42:28 +02:00
|
|
|
# Webhooks for external integrations.
|
|
|
|
import re
|
2017-03-03 19:01:52 +01:00
|
|
|
from typing import Any, Dict, List
|
2017-11-16 00:43:10 +01:00
|
|
|
|
2016-05-25 20:42:28 +02:00
|
|
|
from django.http import HttpRequest, HttpResponse
|
2017-07-21 01:55:10 +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
|
2019-02-02 23:53:55 +01:00
|
|
|
from zerver.models import UserProfile
|
2016-05-25 20:42:28 +02:00
|
|
|
|
2018-11-09 21:02:59 +01:00
|
|
|
TOPIC_TEMPLATE = "{service_url}"
|
2016-05-25 20:42:28 +02:00
|
|
|
|
2018-03-16 22:53:50 +01:00
|
|
|
def send_message_for_event(request: HttpRequest, user_profile: UserProfile,
|
|
|
|
event: Dict[str, Any]) -> None:
|
2017-08-24 17:31:04 +02:00
|
|
|
event_type = get_event_type(event)
|
2018-11-09 21:02:59 +01:00
|
|
|
subject = TOPIC_TEMPLATE.format(service_url=event['check']['url'])
|
2017-08-24 17:31:04 +02:00
|
|
|
body = EVENT_TYPE_BODY_MAPPER[event_type](event)
|
2018-03-16 22:53:50 +01:00
|
|
|
check_send_webhook_message(request, user_profile, subject, body)
|
2016-05-25 20:42:28 +02:00
|
|
|
|
2017-11-04 07:47:46 +01:00
|
|
|
def get_body_for_up_event(event: Dict[str, Any]) -> str:
|
2016-05-25 20:42:28 +02:00
|
|
|
body = "Service is `up`"
|
|
|
|
event_downtime = event['downtime']
|
|
|
|
if event_downtime['started_at']:
|
2020-06-09 00:25:09 +02:00
|
|
|
body = f"{body} again"
|
2016-05-25 20:42:28 +02:00
|
|
|
string_date = get_time_string_based_on_duration(event_downtime['duration'])
|
|
|
|
if string_date:
|
2020-06-09 00:25:09 +02:00
|
|
|
body = f"{body} after {string_date}"
|
|
|
|
return f"{body}."
|
2016-05-25 20:42:28 +02:00
|
|
|
|
2017-11-04 07:47:46 +01:00
|
|
|
def get_time_string_based_on_duration(duration: int) -> str:
|
2016-05-25 20:42:28 +02:00
|
|
|
days, reminder = divmod(duration, 86400)
|
|
|
|
hours, reminder = divmod(reminder, 3600)
|
|
|
|
minutes, seconds = divmod(reminder, 60)
|
|
|
|
|
|
|
|
string_date = ''
|
|
|
|
string_date += add_time_part_to_string_date_if_needed(days, 'day')
|
|
|
|
string_date += add_time_part_to_string_date_if_needed(hours, 'hour')
|
|
|
|
string_date += add_time_part_to_string_date_if_needed(minutes, 'minute')
|
|
|
|
string_date += add_time_part_to_string_date_if_needed(seconds, 'second')
|
|
|
|
return string_date.rstrip()
|
|
|
|
|
2017-11-04 07:47:46 +01:00
|
|
|
def add_time_part_to_string_date_if_needed(value: int, text_name: str) -> str:
|
2016-05-25 20:42:28 +02:00
|
|
|
if value == 1:
|
2020-06-09 00:25:09 +02:00
|
|
|
return f"1 {text_name} "
|
2016-05-25 20:42:28 +02:00
|
|
|
if value > 1:
|
2020-06-09 00:25:09 +02:00
|
|
|
return f"{value} {text_name}s "
|
2016-05-25 20:42:28 +02:00
|
|
|
return ''
|
|
|
|
|
2017-11-04 07:47:46 +01:00
|
|
|
def get_body_for_down_event(event: Dict[str, Any]) -> str:
|
2017-02-26 09:10:14 +01:00
|
|
|
return "Service is `down`. It returned a {} error at {}.".format(
|
|
|
|
event['downtime']['error'],
|
|
|
|
event['downtime']['started_at'].replace('T', ' ').replace('Z', ' UTC'))
|
2016-05-25 20:42:28 +02:00
|
|
|
|
|
|
|
@api_key_only_webhook_view('Updown')
|
|
|
|
@has_request_variables
|
2018-03-16 22:53:50 +01:00
|
|
|
def api_updown_webhook(
|
|
|
|
request: HttpRequest, user_profile: UserProfile,
|
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
|
|
|
payload: List[Dict[str, Any]]=REQ(argument_type='body'),
|
2018-03-16 22:53:50 +01:00
|
|
|
) -> HttpResponse:
|
2016-05-25 20:42:28 +02:00
|
|
|
for event in payload:
|
2018-03-16 22:53:50 +01:00
|
|
|
send_message_for_event(request, user_profile, event)
|
2016-05-25 20:42:28 +02:00
|
|
|
return json_success()
|
|
|
|
|
|
|
|
EVENT_TYPE_BODY_MAPPER = {
|
|
|
|
'up': get_body_for_up_event,
|
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
|
|
|
'down': get_body_for_down_event,
|
2016-05-25 20:42:28 +02:00
|
|
|
}
|
|
|
|
|
2017-11-04 07:47:46 +01:00
|
|
|
def get_event_type(event: Dict[str, Any]) -> str:
|
2016-05-25 20:42:28 +02:00
|
|
|
event_type_match = re.match('check.(.*)', event['event'])
|
|
|
|
if event_type_match:
|
|
|
|
event_type = event_type_match.group(1)
|
|
|
|
if event_type in EVENT_TYPE_BODY_MAPPER:
|
|
|
|
return event_type
|
2018-05-22 16:46:45 +02:00
|
|
|
raise UnexpectedWebhookEventType('Updown', event['event'])
|