zulip/zerver/webhooks/pingdom/view.py

81 lines
2.2 KiB
Python
Raw Normal View History

# Webhooks pfor external integrations.
from typing import Any, Dict
from django.http import HttpRequest, HttpResponse
from zerver.decorator import api_key_only_webhook_view
from zerver.lib.request import REQ, has_request_variables
from zerver.lib.response import json_success
from zerver.lib.webhooks.common import UnexpectedWebhookEventType, check_send_webhook_message
from zerver.models import UserProfile
2016-03-14 22:12:46 +01:00
PINGDOM_TOPIC_TEMPLATE = '{name} status.'
2016-03-14 22:12:46 +01: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',
)
@api_key_only_webhook_view('Pingdom')
2016-03-14 22:12:46 +01:00
@has_request_variables
def api_pingdom_webhook(request: HttpRequest, user_profile: UserProfile,
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:
raise UnexpectedWebhookEventType('Pingdom', check_type)
2016-03-14 22:12:46 +01:00
check_send_webhook_message(request, user_profile, subject, body)
2016-03-14 22:12:46 +01:00
return json_success()
def get_subject_for_http_request(payload: Dict[str, Any]) -> str:
return PINGDOM_TOPIC_TEMPLATE.format(name=payload['check_name'])
2016-03-14 22:12:46 +01: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,
'type': get_check_type(payload),
2016-03-14 22:12:46 +01:00
}
body = MESSAGE_TEMPLATE.format(**data)
2016-03-14 22:12:46 +01:00
if current_state == 'DOWN' and previous_state == 'UP':
description = DESC_TEMPLATE.format(description=payload['long_description'])
body += description
else:
body = f'{body[:-1]}.'
2016-03-14 22:12:46 +01:00
return body
def get_check_type(payload: Dict[str, Any]) -> str:
2016-03-14 22:12:46 +01:00
return payload['check_type']