Files
zulip/zerver/webhooks/opsgenie/view.py
Anders Kaseorg 69730a78cc 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-06-11 16:04:12 -07:00

85 lines
2.8 KiB
Python

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 check_send_webhook_message
from zerver.models import UserProfile
@api_key_only_webhook_view('OpsGenie')
@has_request_variables
def api_opsgenie_webhook(request: HttpRequest, user_profile: UserProfile,
payload: Dict[str, Any]=REQ(argument_type='body')) -> HttpResponse:
# construct the body of the message
info = {
"additional_info": '',
"alert_type": payload['action'],
"alert_id": payload['alert']['alertId'],
"integration_name": payload['integrationName'],
"tags": ', '.join(['`' + tag + '`' for tag in payload['alert'].get('tags', [])]),
}
topic = info['integration_name']
bullet_template = "* **{key}**: {value}\n"
if 'note' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Note',
value=payload['alert']['note'],
)
if 'recipient' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Recipient',
value=payload['alert']['recipient'],
)
if 'addedTags' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Tags added',
value=payload['alert']['addedTags'],
)
if 'team' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Team added',
value=payload['alert']['team'],
)
if 'owner' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Assigned owner',
value=payload['alert']['owner'],
)
if 'escalationName' in payload:
info['additional_info'] += bullet_template.format(
key='Escalation',
value=payload['escalationName'],
)
if 'removedTags' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Tags removed',
value=payload['alert']['removedTags'],
)
if 'message' in payload['alert']:
info['additional_info'] += bullet_template.format(
key='Message',
value=payload['alert']['message'],
)
if info['tags']:
info['additional_info'] += bullet_template.format(
key='Tags',
value=info['tags'],
)
body_template = """
[OpsGenie Alert for {integration_name}](https://app.opsgenie.com/alert/V2#/show/{alert_id}):
* **Type**: {alert_type}
{additional_info}
""".strip()
body = body_template.format(**info)
check_send_webhook_message(request, user_profile, topic, body)
return json_success()