File size: 6,819 Bytes
17e971c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 | #coding: utf-8
# +-------------------------------------------------------------------
# | 宝塔Linux面板
# +-------------------------------------------------------------------
# | Copyright (c) 2015-2020 宝塔软件(http://www.bt.cn) All rights reserved.
# +-------------------------------------------------------------------
# | Author: lx
# | 消息通道飞书通知模块
# +-------------------------------------------------------------------
import os, sys, public, json, requests
import sys, os
panelPath = '/www/server/panel'
os.chdir(panelPath)
sys.path.insert(0,panelPath + "/class/")
import public, json, requests
from requests.packages import urllib3
# 关闭警告
urllib3.disable_warnings()
import socket
import requests.packages.urllib3.util.connection as urllib3_cn
class feishu_msg:
conf_path = 'data/feishu.json'
__feishu_info = None
__module_name = None
__default_pl = "{}/data/default_msg_channel.pl".format(panelPath)
def __init__(self):
try:
self.__feishu_info = json.loads(public.readFile(self.conf_path))
if not 'feishu_url' in self.__feishu_info or not 'isAtAll' in self.__feishu_info or not 'user' in self.__feishu_info:
self.__feishu_info = None
except :
self.__feishu_info = None
self.__module_name = self.__class__.__name__.replace('_msg','')
def get_version_info(self,get):
"""
获取版本信息
"""
data = {}
data['ps'] = '宝塔飞书消息通道，用于接收面板消息推送'
data['version'] = '1.2'
data['date'] = '2022-08-10'
data['author'] = '宝塔'
data['title'] = '飞书'
data['help'] = 'http://www.bt.cn'
return data
def __get_default_channel(self):
"""
@获取默认消息通道
"""
try:
if public.readFile(self.__default_pl) == self.__module_name:
return True
except:pass
return False
def get_config(self,get):
"""
获取飞书配置
"""
data = {}
if self.__feishu_info :
data = self.__feishu_info
if not 'list' in data: data['list'] = {}
title = '默认'
if 'title' in data: title = data['title']
data['list']['default'] = {'title':title,'data':data['feishu_url']}
data['default'] = self.__get_default_channel()
return data
def set_config(self,get):
"""
设置飞书配置
@url 飞书URL
@atall 默认@全体成员
@user
"""
if not hasattr(get, 'url'):
return public.returnMsg(False, '请填写完整信息')
isAtAll = True
if hasattr(get, "atall"):
if get.atall.lower() == "false":
isAtAll = False
title = '默认'
if hasattr(get, 'title'):
title = get.title
if len(title) > 7:
return public.returnMsg(False, '备注名称不能超过7个字符')
self.__feishu_info = {"feishu_url": get.url.strip(), "isAtAll": isAtAll, "user":1,"title":title}
ret = self.send_msg('宝塔告警测试')
if ret['status']:
if 'default' in get and get['default']:
public.writeFile(self.__default_pl, self.__module_name)
if ret['success'] <= 0:
return public.returnMsg(False, '添加失败,请查看URL是否正确')
public.writeFile(self.conf_path, json.dumps(self.__feishu_info))
return public.returnMsg(True, '设置成功')
else:
return public.returnMsg(False, '添加失败,请查看URL是否正确')
def get_send_msg(self,msg):
"""
@name 处理md格式
"""
try:
import re
title = '宝塔告警通知'
if msg.find("####") >= 0:
try:
title = re.search(r"####(.+)", msg).groups()[0]
except:pass
msg = msg.replace("####",">").replace("\n\n","\n").strip()
s_list = msg.split('\n')
if len(s_list) > 3:
s_title = s_list[0].replace(" ","")
s_list = s_list[1:]
s_list.insert(0,s_title)
msg = '\n'.join(s_list)
reg = '<font.+>(.+)</font>'
tmp = re.search(reg,msg)
if tmp:
tmp = tmp.groups()[0]
msg = re.sub(reg,tmp,msg)
except:pass
return msg,title
def send_msg(self,msg,to_user = 'default'):
"""
飞书发送信息
@msg 消息正文
"""
if not self.__feishu_info :
return public.returnMsg(False,'未正确配置飞书信息。')
msg,title = self.get_send_msg(msg)
if self.__feishu_info["isAtAll"]:
msg += "<at userid='all'>所有人</at>"
data = {
"msg_type": "text",
"content": {
"text": msg
}
}
headers = {'Content-Type': 'application/json'}
res = {}
error,success = 0,0
conf = self.get_config(None)['list']
for to_key in to_user.split(','):
if not to_key in conf: continue
try:
def allowed_gai_family():
family = socket.AF_INET
return family
urllib3_cn.allowed_gai_family = allowed_gai_family
rdata = requests.post(url = conf[to_key]['data'], data = json.dumps(data),verify=False, headers=headers,timeout=10).json()
public.reset_allowed_gai_family()
# x = requests.post(url = conf[to_key]['data'], data=json.dumps(data), headers=headers,verify=False,timeout=10)
# rdata = x.json()
if "StatusCode" in rdata and rdata["StatusCode"] == 0:
success += 1
res[conf[to_key]['title']] = 1
else:
error += 1
res[conf[to_key]['title']] = 0
except:
error += 1
res[conf[to_key]['title']] = 0
try:
public.write_push_log(self.__module_name,title,res)
except:pass
ret = public.returnMsg(True,'发送完成,发送成功{},发送失败{}.'.format(success,error))
ret['success'] = success
ret['error'] = error
return ret
def push_data(self,data):
if not isinstance(data, dict):
return self.send_msg(data)
return self.send_msg(data['msg'])
def uninstall(self):
if os.path.exists(self.conf_path):
os.remove(self.conf_path) |