mirror of
https://github.com/zulip/zulip.git
synced 2025-11-10 08:56:10 +00:00
dcbb2a78ca
JsonableError has two major benefits over json_error: * It can be raised from anywhere in the codebase, rather than being a return value, which is much more convenient for refactoring, as one doesn't potentially need to change error handling style when extracting a bit of view code to a function. * It is guaranteed to contain the `code` property, which is helpful for API consistency. Various stragglers are not updated because JsonableError requires subclassing in order to specify custom data or HTTP status codes.
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
from typing import Any, Dict
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.utils.translation import gettext as _
|
|
|
|
from zerver.decorator import REQ, has_request_variables, webhook_view
|
|
from zerver.lib.actions import send_rate_limited_pm_notification_to_bot_owner
|
|
from zerver.lib.exceptions import JsonableError
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.send_email import FromAddress
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.models import UserProfile
|
|
|
|
MISCONFIGURED_PAYLOAD_ERROR_MESSAGE = """
|
|
Hi there! Your bot {bot_name} just received a Zabbix payload that is missing
|
|
some data that Zulip requires. This usually indicates a configuration issue
|
|
in your Zabbix webhook settings. Please make sure that you set the
|
|
script correctly and provide all the required parameters
|
|
when configuring the Zabbix webhook. Contact {support_email} if you
|
|
need further help!
|
|
"""
|
|
|
|
ZABBIX_TOPIC_TEMPLATE = "{hostname}"
|
|
ZABBIX_MESSAGE_TEMPLATE = """
|
|
{status} ({severity}) alert on [{hostname}]({link}):
|
|
* {trigger}
|
|
* {item}
|
|
""".strip()
|
|
|
|
|
|
@webhook_view("Zabbix")
|
|
@has_request_variables
|
|
def api_zabbix_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
payload: Dict[str, Any] = REQ(argument_type="body"),
|
|
) -> HttpResponse:
|
|
|
|
try:
|
|
body = get_body_for_http_request(payload)
|
|
subject = get_subject_for_http_request(payload)
|
|
except KeyError:
|
|
message = MISCONFIGURED_PAYLOAD_ERROR_MESSAGE.format(
|
|
bot_name=user_profile.full_name,
|
|
support_email=FromAddress.SUPPORT,
|
|
).strip()
|
|
send_rate_limited_pm_notification_to_bot_owner(user_profile, user_profile.realm, message)
|
|
|
|
raise JsonableError(_("Invalid payload"))
|
|
|
|
check_send_webhook_message(request, user_profile, subject, body)
|
|
return json_success()
|
|
|
|
|
|
def get_subject_for_http_request(payload: Dict[str, Any]) -> str:
|
|
return ZABBIX_TOPIC_TEMPLATE.format(hostname=payload["hostname"])
|
|
|
|
|
|
def get_body_for_http_request(payload: Dict[str, Any]) -> str:
|
|
hostname = payload["hostname"]
|
|
severity = payload["severity"]
|
|
status = payload["status"]
|
|
item = payload["item"]
|
|
trigger = payload["trigger"]
|
|
link = payload["link"]
|
|
|
|
data = {
|
|
"hostname": hostname,
|
|
"severity": severity,
|
|
"status": status,
|
|
"item": item,
|
|
"trigger": trigger,
|
|
"link": link,
|
|
}
|
|
return ZABBIX_MESSAGE_TEMPLATE.format(**data)
|