mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
3afc8ed7ae
This is preparatory work towards adding a Topic model. We plan to use the local variable name as 'topic' for the Topic model objects. Currently, we use *topic as the local variable name for topic names. We rename local variables of the form *topic to *topic_name so that we don't need to think about type collisions in individual code paths where we might want to talk about both Topic objects and strings for the topic name.
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# Webhooks for external integrations.
|
|
from typing import Dict, List
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from typing_extensions import Annotated
|
|
|
|
from zerver.decorator import webhook_view
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.typed_endpoint import ApiParamConfig, JsonBodyPayload, typed_endpoint
|
|
from zerver.lib.validator import WildValue, check_string
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.models import UserProfile
|
|
|
|
|
|
@webhook_view("Alertmanager")
|
|
@typed_endpoint
|
|
def api_alertmanager_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
*,
|
|
payload: JsonBodyPayload[WildValue],
|
|
name_field: Annotated[str, ApiParamConfig("name")] = "instance",
|
|
desc_field: Annotated[str, ApiParamConfig("desc")] = "alertname",
|
|
) -> HttpResponse:
|
|
topics: Dict[str, Dict[str, List[str]]] = {}
|
|
|
|
for alert in payload["alerts"]:
|
|
labels = alert.get("labels", {})
|
|
annotations = alert.get("annotations", {})
|
|
|
|
name = labels.get(name_field, annotations.get(name_field, "(unknown)")).tame(check_string)
|
|
desc = labels.get(
|
|
desc_field, annotations.get(desc_field, f"<missing field: {desc_field}>")
|
|
).tame(check_string)
|
|
|
|
url = alert["generatorURL"].tame(check_string).replace("tab=1", "tab=0")
|
|
|
|
body = f"{desc} ([graph]({url}))"
|
|
if name not in topics:
|
|
topics[name] = {"firing": [], "resolved": []}
|
|
topics[name][alert["status"].tame(check_string)].append(body)
|
|
|
|
for topic_name, statuses in topics.items():
|
|
for status, messages in statuses.items():
|
|
if len(messages) == 0:
|
|
continue
|
|
|
|
if status == "firing":
|
|
icon = ":alert:"
|
|
title = "FIRING"
|
|
else:
|
|
title = "Resolved"
|
|
icon = ":squared_ok:"
|
|
|
|
if len(messages) == 1:
|
|
body = f"{icon} **{title}** {messages[0]}"
|
|
else:
|
|
message_list = "\n".join(f"* {m}" for m in messages)
|
|
body = f"{icon} **{title}**\n{message_list}"
|
|
|
|
check_send_webhook_message(request, user_profile, topic_name, body)
|
|
|
|
return json_success(request)
|