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()