# -*- coding: utf-8 -*- # Copyright 2010 - 2017 RhodeCode GmbH and the AppEnlight project authors # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import logging from appenlight.models.alert_channel import AlertChannel from appenlight.models.integrations.hipchat import HipchatIntegration from webhelpers2.text import truncate log = logging.getLogger(__name__) class HipchatAlertChannel(AlertChannel): __mapper_args__ = { 'polymorphic_identity': 'hipchat' } def notify_reports(self, **kwargs): """ Notify user of individual reports kwargs: application: application that the event applies for, user: user that should be notified request: request object since_when: reports are newer than this time value, reports: list of reports to render """ template_vars = self.report_alert_notification_vars(kwargs) app_url = kwargs['request'].registry.settings['_mail_url'] destination_url = kwargs['request'].route_url('/', _app_url=app_url) f_args = ('report', template_vars['resource'].resource_id, template_vars['url_start_date'].strftime('%Y-%m-%dT%H:%M'), template_vars['url_end_date'].strftime('%Y-%m-%dT%H:%M')) destination_url += 'ui/{}?resource={}&start_date={}&end_date={}'.format( *f_args) if template_vars['confirmed_total'] > 1: template_vars["title"] = "%s - %s reports" % ( template_vars['resource_name'], template_vars['confirmed_total'], ) else: error_title = truncate(template_vars['reports'][0][1].error or 'slow report', 90) template_vars["title"] = "%s - '%s' report" % ( template_vars['resource_name'], error_title) template_vars["title"] += ' ' + destination_url log_msg = 'NOTIFY : %s via %s :: %s reports' % ( kwargs['user'].user_name, self.channel_visible_value, template_vars['confirmed_total']) log.warning(log_msg) client = HipchatIntegration.create_client( self.integration.config['api_token']) for room in self.integration.config['rooms'].split(','): client.send({ "message_format": 'text', "message": template_vars["title"], "from": "AppEnlight", "room_id": room.strip(), "color": "yellow" }) def notify_report_alert(self, **kwargs): """ Build and send report alert notification Kwargs: application: application that the event applies for, event: event that is notified, user: user that should be notified request: request object """ template_vars = self.report_alert_notification_vars(kwargs) if kwargs['event'].unified_alert_action() == 'OPEN': title = 'ALERT %s: %s - %s %s' % ( template_vars['alert_action'], template_vars['resource_name'], kwargs['event'].values['reports'], template_vars['report_type'], ) else: title = 'ALERT %s: %s type: %s' % ( template_vars['alert_action'], template_vars['resource_name'], template_vars['alert_type'].replace('_', ' '), ) title += '\n ' + template_vars['destination_url'] api_token = self.integration.config['api_token'] client = HipchatIntegration.create_client(api_token) for room in self.integration.config['rooms'].split(','): client.send({ "message_format": 'text', "message": title, "from": "AppEnlight", "room_id": room.strip(), "color": "red", "notify": '1' }) def notify_uptime_alert(self, **kwargs): """ Build and send uptime alert notification Kwargs: application: application that the event applies for, event: event that is notified, user: user that should be notified request: request object """ template_vars = self.uptime_alert_notification_vars(kwargs) message = 'ALERT %s: %s has uptime issues\n' % ( template_vars['alert_action'], template_vars['resource_name'], ) message += template_vars['reason'] message += '\n{}'.format(template_vars['destination_url']) api_token = self.integration.config['api_token'] client = HipchatIntegration.create_client(api_token) for room in self.integration.config['rooms'].split(','): client.send({ "message_format": 'text', "message": message, "from": "AppEnlight", "room_id": room.strip(), "color": "red", "notify": '1' }) def notify_chart_alert(self, **kwargs): """ Build and send chart alert notification Kwargs: application: application that the event applies for, event: event that is notified, user: user that should be notified request: request object """ template_vars = self.chart_alert_notification_vars(kwargs) message = 'ALERT {}: value in "{}" chart: ' \ 'met alert "{}" criteria\n'.format( template_vars['alert_action'], template_vars['chart_name'], template_vars['action_name'], ) for item in template_vars['readable_values']: message += '{}: {}\n'.format(item['label'], item['value']) message += template_vars['destination_url'] api_token = self.integration.config['api_token'] client = HipchatIntegration.create_client(api_token) for room in self.integration.config['rooms'].split(','): client.send({ "message_format": 'text', "message": message, "from": "AppEnlight", "room_id": room.strip(), "color": "red", "notify": '1' }) def send_digest(self, **kwargs): """ Build and send daily digest notification kwargs: application: application that the event applies for, user: user that should be notified request: request object since_when: reports are newer than this time value, reports: list of reports to render """ template_vars = self.report_alert_notification_vars(kwargs) f_args = (template_vars['resource_name'], template_vars['confirmed_total'],) message = "Daily report digest: %s - %s reports" % f_args message += '\n{}'.format(template_vars['destination_url']) api_token = self.integration.config['api_token'] client = HipchatIntegration.create_client(api_token) for room in self.integration.config['rooms'].split(','): client.send({ "message_format": 'text', "message": message, "from": "AppEnlight", "room_id": room.strip(), "color": "green", "notify": '1' }) log_msg = 'DIGEST : %s via %s :: %s reports' % ( kwargs['user'].user_name, self.channel_visible_value, template_vars['confirmed_total']) log.warning(log_msg)