spug/spug_api/libs/spug.py

# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
# Copyright: (c) <spug.dev@gmail.com>
# Released under the AGPL-3.0 License.
from apps.alarm.models import Group, Contact
from apps.setting.utils import AppSetting
from apps.notify.models import Notify
from libs.mail import Mail
from libs.utils import human_datetime
import requests
import json

spug_server = 'https://api.spug.cc'
notify_source = 'monitor'


def send_login_wx_code(wx_token, code):
    url = f'{spug_server}/apis/login/wx/'
    spug_key = AppSetting.get_default('spug_key')
    res = requests.post(url, json={'token': spug_key, 'user': wx_token, 'code': code}, timeout=30)
    if res.status_code != 200:
        raise Exception(f'status code: {res.status_code}')
    res = res.json()
    if res.get('error'):
        raise Exception(res['error'])


class Notification:
    def __init__(self, grp, event, target, title, message, duration):
        self.event = event
        self.title = title
        self.target = target
        self.message = message
        self.duration = duration
        self.spug_key, self.u_ids = self._parse_args(grp)

    def _parse_args(self, grp):
        spug_key = AppSetting.get_default('spug_key')
        return spug_key, sum([json.loads(x.contacts) for x in Group.objects.filter(id__in=grp)], [])

    def _handle_request(self, mode, url, data):
        try:
            res = requests.post(url, json=data, timeout=30)
        except Exception as e:
            Notify.make_notify(notify_source, '1', '告警通知发送失败', f'接口调用异常：{e}')
            return
        if res.status_code != 200:
            Notify.make_notify(notify_source, '1', '告警通知发送失败', f'返回状态码：{res.status_code}, 请求URL：{res.url}')
        if mode in ['dd', 'wx']:
            res = res.json()
            if res.get('errcode') != 0:
                Notify.make_notify(notify_source, '1', '告警通知发送失败', f'返回数据：{res}')
        if mode == 'spug':
            res = res.json()
            if res.get('error'):
                Notify.make_notify(notify_source, '1', '告警通知发送失败', f'错误信息：{res}')

    def _by_wx(self):
        if not self.spug_key:
            Notify.make_notify(notify_source, '1', '发送报警信息失败', '未配置报警服务调用凭据，请在系统管理/系统设置/报警服务设置中配置。')
            return
        users = set(x.wx_token for x in Contact.objects.filter(id__in=self.u_ids, wx_token__isnull=False))
        if users:
            data = {
                'token': self.spug_key,
                'event': self.event,
                'subject': self.title,
                'desc': self.message,
                'remark': f'故障持续{self.duration}' if self.event == '2' else None,
                'users': list(users)
            }
            self._handle_request('spug', f'{spug_server}/apis/notify/wx/', data)
        else:
            Notify.make_notify(notify_source, '1', '发送报警信息失败', '未找到可用的通知对象，请确保设置了相关报警联系人的微信Token。')

    def _by_email(self):
        users = set(x.email for x in Contact.objects.filter(id__in=self.u_ids, email__isnull=False))
        if users:
            mail_service = json.loads(AppSetting.get_default('mail_service', '{}'))
            body = [
                f'告警名称：{self.title}',
                f'告警对象：{self.target}',
                f'{"告警" if self.event == "1" else "恢复"}时间：{human_datetime()}',
                f'告警描述：{self.message}'
            ]
            if self.event == '2':
                body.append('故障持续：' + self.duration)
            if mail_service.get('server'):
                event_map = {'1': '监控告警通知', '2': '告警恢复通知'}
                subject = f'{event_map[self.event]}-{self.title}'
                mail = Mail(**mail_service)
                mail.send_text_mail(users, subject, '\r\n'.join(body) + '\r\n\r\n自动发送，请勿回复。')
            elif self.spug_key:
                data = {
                    'token': self.spug_key,
                    'event': self.event,
                    'subject': self.title,
                    'body': '\r\n'.join(body),
                    'users': list(users)
                }
                self._handle_request('spug', f'{spug_server}/apis/notify/mail/', data)
            else:
                Notify.make_notify(notify_source, '1', '发送报警信息失败', '未配置报警服务调用凭据，请在系统管理/系统设置/报警服务设置中配置。')
        else:
            Notify.make_notify(notify_source, '1', '发送报警信息失败', '未找到可用的通知对象，请确保设置了相关报警联系人的邮件地址。')

    def _by_dd(self):
        users = set(x.ding for x in Contact.objects.filter(id__in=self.u_ids, ding__isnull=False))
        if users:
            texts = [
                '## %s ## ' % ('监控告警通知' if self.event == '1' else '告警恢复通知'),
                f'**告警名称：** <font color="#{"f90202" if self.event == "1" else "008000"}">{self.title}</font> ',
                f'**告警对象：** {self.target} ',
                f'**{"告警" if self.event == "1" else "恢复"}时间：** {human_datetime()} ',
                f'**告警描述：** {self.message} ',
            ]
            if self.event == '2':
                texts.append(f'**持续时间：** {self.duration} ')
            data = {
                'msgtype': 'markdown',
                'markdown': {
                    'title': '监控告警通知',
                    'text': '\n\n'.join(texts) + '\n\n> ###### 来自 Spug运维平台'
                }
            }
            for url in users:
                self._handle_request('dd', url, data)
        else:
            Notify.make_notify(notify_source, '1', '发送报警信息失败', '未找到可用的通知对象，请确保设置了相关报警联系人的钉钉。')

    def _by_qy_wx(self):
        users = set(x.qy_wx for x in Contact.objects.filter(id__in=self.u_ids, qy_wx__isnull=False))
        if users:
            color, title = ('warning', '监控告警通知') if self.event == '1' else ('info', '告警恢复通知')
            texts = [
                f'## {title}',
                f'**告警名称：** <font color="{color}">{self.title}</font> ',
                f'**告警对象：** {self.target}',
                f'**{"告警" if self.event == "1" else "恢复"}时间：** {human_datetime()} ',
                f'**告警描述：** {self.message} ',
            ]
            if self.event == '2':
                texts.append(f'**持续时间：** {self.duration} ')
            data = {
                'msgtype': 'markdown',
                'markdown': {
                    'content': '\n'.join(texts) + '\n> 来自 Spug运维平台'
                }
            }
            for url in users:
                self._handle_request('wx', url, data)
        else:
            Notify.make_notify(notify_source, '1', '发送报警信息失败', '未找到可用的通知对象，请确保设置了相关报警联系人的企业微信。')

    def dispatch(self, modes):
        for mode in modes:
            if mode == '1':
                self._by_wx()
            elif mode == '3':
                self._by_dd()
            elif mode == '4':
                self._by_email()
            elif mode == '5':
                self._by_qy_wx()