269 lines
9.1 KiB
Python
269 lines
9.1 KiB
Python
|
#coding: utf-8
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | YakPanel
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Author: 沐落 <cjx@yakpanel.com>
|
|||
|
|
# | Author: lx
|
|||
|
|
# | 消息通道邮箱模块
|
|||
|
|
# | 常用功能
|
|||
|
|
# 字体加粗 **bold** ，[这是一个链接](http://www.yakpanel.com)，代码段：`code`
|
|||
|
|
# 支持3种字体颜色 <font color="info">绿色</font> <font color="comment">灰色</font> <font color="warning">橙红色</font>
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
import os, sys, public, json, requests,re
|
|||
|
|
import sys, os,time
|
|||
|
|
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 weixin_msg:
|
|||
|
|
|
|||
|
|
__module_name = None
|
|||
|
|
__default_pl = "{}/data/default_msg_channel.pl".format(panelPath)
|
|||
|
|
conf_path = 'data/weixin.json'
|
|||
|
|
__weixin_info = None
|
|||
|
|
def __init__(self):
|
|||
|
|
try:
|
|||
|
|
self.__weixin_info = json.loads(public.readFile(self.conf_path))
|
|||
|
|
if not 'weixin_url' in self.__weixin_info:
|
|||
|
|
self.__weixin_info = None
|
|||
|
|
except :
|
|||
|
|
self.__weixin_info = None
|
|||
|
|
self.__module_name = self.__class__.__name__.replace('_msg','')
|
|||
|
|
|
|||
|
|
def get_version_info(self,get):
|
|||
|
|
"""
|
|||
|
|
获取版本信息
|
|||
|
|
"""
|
|||
|
|
data = {}
|
|||
|
|
data['ps'] = 'Wecom used to receive panel message push'
|
|||
|
|
data['version'] = '1.2'
|
|||
|
|
data['date'] = '2022-08-10'
|
|||
|
|
data['author'] = 'YakPanel'
|
|||
|
|
data['title'] = 'Wecom'
|
|||
|
|
data['help'] = 'http://www.yakpanel.com'
|
|||
|
|
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.__weixin_info :
|
|||
|
|
|
|||
|
|
#全局配置开关,1开启，0关闭
|
|||
|
|
if not 'state' in self.__weixin_info:
|
|||
|
|
self.__weixin_info['state'] = 1
|
|||
|
|
|
|||
|
|
data = self.__weixin_info
|
|||
|
|
|
|||
|
|
if not 'list' in data: data['list'] = {}
|
|||
|
|
|
|||
|
|
title = 'Default'
|
|||
|
|
if 'title' in data: title = data['title']
|
|||
|
|
|
|||
|
|
data['list']['default'] = {'title':title,'data':data['weixin_url'],'state':self.__weixin_info['state']}
|
|||
|
|
data['default'] = self.__get_default_channel()
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def set_config(self,get):
|
|||
|
|
"""
|
|||
|
|
设置微信配置
|
|||
|
|
@url 微信URL
|
|||
|
|
@atall 默认@全体成员
|
|||
|
|
@key 唯一标识，default=兼容之前配置
|
|||
|
|
@title string 备注
|
|||
|
|
@user
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
if not hasattr(get, 'url'):
|
|||
|
|
return public.returnMsg(False, public.lang("Please fill in the complete information"))
|
|||
|
|
|
|||
|
|
title = 'Default'
|
|||
|
|
if hasattr(get, 'title'):
|
|||
|
|
title = get.title
|
|||
|
|
if len(title) > 7:
|
|||
|
|
return public.returnMsg(False, public.lang("Note name cannot exceed 7 characters"))
|
|||
|
|
|
|||
|
|
key,status,state ='default', 1, 1
|
|||
|
|
if 'key' in get: key = get.key
|
|||
|
|
if 'status' in get: status = int(get.status)
|
|||
|
|
if 'state' in get: state = int(get.state)
|
|||
|
|
|
|||
|
|
if not self.__weixin_info: self.__weixin_info = {}
|
|||
|
|
if not 'list' in self.__weixin_info: self.__weixin_info['list'] = {}
|
|||
|
|
|
|||
|
|
#全局配置开关,1开启，0关闭
|
|||
|
|
self.__weixin_info['state'] = state
|
|||
|
|
|
|||
|
|
#增加多个机器人
|
|||
|
|
self.__weixin_info['list'][key] = {
|
|||
|
|
"data": get.url.strip(),
|
|||
|
|
"title":title,
|
|||
|
|
"status":status,
|
|||
|
|
"addtime":int(time.time())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#兼容旧配置只有一条url的情况
|
|||
|
|
if key == 'default':
|
|||
|
|
self.__weixin_info['weixin_url'] = get.url.strip()
|
|||
|
|
self.__weixin_info['title'] = title
|
|||
|
|
|
|||
|
|
#统一格式化输出，包含主机名，ip，推送时间
|
|||
|
|
try:
|
|||
|
|
info = public.get_push_info('Message channel configuration reminder',['>configuration status: <font color=#20a53a>Success</font>\n\n'])
|
|||
|
|
ret = self.send_msg(info['msg'])
|
|||
|
|
except:
|
|||
|
|
ret = self.send_msg('YakPanel alarm test')
|
|||
|
|
|
|||
|
|
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, public.lang("Failed to add, please check whether the URL is correct"))
|
|||
|
|
|
|||
|
|
public.writeFile(self.conf_path, json.dumps(self.__weixin_info))
|
|||
|
|
return public.returnMsg(True, public.lang("successfully set"))
|
|||
|
|
else:
|
|||
|
|
return public.returnMsg(False, public.lang("Failed to add, please check whether the URL is correct"))
|
|||
|
|
|
|||
|
|
def get_send_msg(self,msg):
|
|||
|
|
"""
|
|||
|
|
@name 处理md格式
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
title = 'YakPanel warning notification'
|
|||
|
|
if msg.find("####") >= 0:
|
|||
|
|
msg = msg.replace("\n\n","""
|
|||
|
|
""").strip()
|
|||
|
|
try:
|
|||
|
|
title = re.search(r"####(.+)", msg).groups()[0]
|
|||
|
|
except:pass
|
|||
|
|
|
|||
|
|
except:pass
|
|||
|
|
return msg,title
|
|||
|
|
|
|||
|
|
def send_msg(self,msg,to_user = 'default'):
|
|||
|
|
"""
|
|||
|
|
@name 微信发送信息
|
|||
|
|
@msg string 消息正文(正文内容，必须包含
|
|||
|
|
1、服务器名称
|
|||
|
|
2、IP地址
|
|||
|
|
3、发送时间
|
|||
|
|
)
|
|||
|
|
@to_user string 指定发送人
|
|||
|
|
"""
|
|||
|
|
if not self.__weixin_info :
|
|||
|
|
return public.returnMsg(False, public.lang("Information is not configured correctly."))
|
|||
|
|
|
|||
|
|
if 'state' in self.__weixin_info and self.__weixin_info['state'] == 0:
|
|||
|
|
return public.returnMsg(False, public.lang("Notifications have been turned off, please turn them on and try again."))
|
|||
|
|
|
|||
|
|
if msg.find('####') == -1:
|
|||
|
|
try:
|
|||
|
|
msg = public.get_push_info('Notification Configuration Reminder',['>Send Content：{}\n\n'.format(msg)])['msg']
|
|||
|
|
except:pass
|
|||
|
|
|
|||
|
|
msg,title = self.get_send_msg(msg)
|
|||
|
|
data = {
|
|||
|
|
"msgtype": "markdown",
|
|||
|
|
"markdown": {
|
|||
|
|
"content": msg
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
headers = {'Content-Type': 'application/json'}
|
|||
|
|
|
|||
|
|
error,success = 0,0
|
|||
|
|
conf = self.get_config(None)['list']
|
|||
|
|
|
|||
|
|
res = {}
|
|||
|
|
for to_key in to_user.split(','):
|
|||
|
|
if not to_key in conf: continue
|
|||
|
|
try:
|
|||
|
|
#x = requests.post(url = conf[to_key]['data'], data=json.dumps(data), headers=headers,verify=False,timeout=10)
|
|||
|
|
|
|||
|
|
allowed_gai_family_lib=urllib3_cn.allowed_gai_family
|
|||
|
|
def allowed_gai_family():
|
|||
|
|
family = socket.AF_INET
|
|||
|
|
return family
|
|||
|
|
urllib3_cn.allowed_gai_family = allowed_gai_family
|
|||
|
|
x = requests.post(url = conf[to_key]['data'], data = json.dumps(data),verify=False, headers=headers,timeout=10)
|
|||
|
|
urllib3_cn.allowed_gai_family=allowed_gai_family_lib
|
|||
|
|
|
|||
|
|
if x.json()["errcode"] == 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,'Send completed, send successfully {}, send failed {}.'.format(success,error))
|
|||
|
|
ret['success'] = success
|
|||
|
|
ret['error'] = error
|
|||
|
|
|
|||
|
|
return ret
|
|||
|
|
|
|||
|
|
|
|||
|
|
def push_data(self,data):
|
|||
|
|
"""
|
|||
|
|
@name 统一发送接口
|
|||
|
|
@data 消息内容
|
|||
|
|
{"module":"mail","title":"提醒","msg":"提醒","to_email":"xx@qq.com","sm_type":"","sm_args":{}}
|
|||
|
|
"""
|
|||
|
|
if not 'to_user' in data:
|
|||
|
|
data['to_user'] = 'default'
|
|||
|
|
|
|||
|
|
return self.send_msg(data['msg'],data['to_user'])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _write_log(self,module,msg,res):
|
|||
|
|
"""
|
|||
|
|
@name 写日志
|
|||
|
|
"""
|
|||
|
|
user = '[ 默认 ] '
|
|||
|
|
# for key in res:
|
|||
|
|
# status = '<span style="color:#20a53a;">成功</span>'
|
|||
|
|
# if res[key] == 0: status = '<span style="color:red;">成功</span>'
|
|||
|
|
# user += '[ {}:{} ] '.format(key,status)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
msg_obj = public.init_msg(module)
|
|||
|
|
if msg_obj: module = msg_obj.get_version_info(None)['title']
|
|||
|
|
except:pass
|
|||
|
|
|
|||
|
|
log = '[{}] sent to {}, sending content: [{}]'.format(module,user,public.xsssec(msg))
|
|||
|
|
public.WriteLog('message push',log)
|
|||
|
|
|
|||
|
|
def uninstall(self):
|
|||
|
|
if os.path.exists(self.conf_path):
|
|||
|
|
os.remove(self.conf_path)
|
|||
|
|
|
|||
|
|
|