224 lines
7.3 KiB
Python
224 lines
7.3 KiB
Python
|
#coding: utf-8
|
||
|
|
# +-------------------------------------------------------------------
|
||
|
|
# | YakPanel
|
||
|
|
# +-------------------------------------------------------------------
|
||
|
|
# | Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
||
|
|
# +-------------------------------------------------------------------
|
||
|
|
# | Author: 沐落 <cjx@yakpanel.com>
|
||
|
|
# | Author: lx
|
||
|
|
# | 消息通道邮箱模块
|
||
|
|
# +-------------------------------------------------------------------
|
||
|
|
|
||
|
|
import os, sys, public, base64, json, re,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 dingding_msg:
|
||
|
|
|
||
|
|
conf_path = 'data/dingding.json'
|
||
|
|
__dingding_info = None
|
||
|
|
__module_name = None
|
||
|
|
__default_pl = "{}/data/default_msg_channel.pl".format(panelPath)
|
||
|
|
def __init__(self):
|
||
|
|
try:
|
||
|
|
self.__dingding_info = json.loads(public.readFile(self.conf_path))
|
||
|
|
if not 'dingding_url' in self.__dingding_info or not 'isAtAll' in self.__dingding_info or not 'user' in self.__dingding_info:
|
||
|
|
self.__dingding_info = None
|
||
|
|
except :
|
||
|
|
self.__dingding_info = None
|
||
|
|
self.__module_name = self.__class__.__name__.replace('_msg','')
|
||
|
|
|
||
|
|
def get_version_info(self,get):
|
||
|
|
"""
|
||
|
|
获取版本信息
|
||
|
|
"""
|
||
|
|
data = {}
|
||
|
|
data['ps'] = 'Dingding is used to receive panel message push'
|
||
|
|
data['version'] = '1.2'
|
||
|
|
data['date'] = '2022-08-10'
|
||
|
|
data['author'] = 'YakPanel'
|
||
|
|
data['title'] = 'Dingding'
|
||
|
|
data['help'] = 'http://www.yakpanel.com'
|
||
|
|
return data
|
||
|
|
|
||
|
|
def get_config(self,get):
|
||
|
|
"""
|
||
|
|
获取钉钉配置
|
||
|
|
"""
|
||
|
|
data = {}
|
||
|
|
if self.__dingding_info :
|
||
|
|
data = self.__dingding_info
|
||
|
|
|
||
|
|
if not 'list' in data: data['list'] = {}
|
||
|
|
|
||
|
|
title = 'Default'
|
||
|
|
if 'title' in data: title = data['title']
|
||
|
|
|
||
|
|
data['list']['default'] = {'title':title,'data':data['dingding_url']}
|
||
|
|
data['default'] = self.__get_default_channel()
|
||
|
|
|
||
|
|
return data
|
||
|
|
|
||
|
|
def set_config(self,get):
|
||
|
|
"""
|
||
|
|
设置钉钉配置
|
||
|
|
@url 钉钉URL
|
||
|
|
@atall 默认@全体成员
|
||
|
|
@user
|
||
|
|
"""
|
||
|
|
|
||
|
|
if not hasattr(get, 'url') or not hasattr(get, 'atall'):
|
||
|
|
return public.returnMsg(False, public.lang("Please fill in the complete information"))
|
||
|
|
|
||
|
|
user = []
|
||
|
|
status = 1
|
||
|
|
atall = False
|
||
|
|
if 'status' in get: status = int(get.status)
|
||
|
|
if 'user' in get: user = get.user.split('\n')
|
||
|
|
|
||
|
|
if 'atall' in get and get.atall == 'True':
|
||
|
|
atall = True
|
||
|
|
|
||
|
|
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"))
|
||
|
|
|
||
|
|
self.__dingding_info = {"dingding_url": get.url.strip(),"isAtAll": atall, "user":user,"title":title}
|
||
|
|
|
||
|
|
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.__dingding_info))
|
||
|
|
return public.returnMsg(True, public.lang("Notification set successfully"))
|
||
|
|
else:
|
||
|
|
return ret
|
||
|
|
|
||
|
|
|
||
|
|
def get_send_msg(self,msg):
|
||
|
|
"""
|
||
|
|
@name 处理md格式
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
title = 'YakPanel alarm notification'
|
||
|
|
if msg.find("####") >= 0:
|
||
|
|
try:
|
||
|
|
title = re.search(r"####(.+)", msg).groups()[0]
|
||
|
|
except:pass
|
||
|
|
else:
|
||
|
|
info = public.get_push_info('Alarm Mode Configuration Reminder',['>Send Content: ' + msg])
|
||
|
|
msg = info['msg']
|
||
|
|
except:pass
|
||
|
|
return msg,title
|
||
|
|
|
||
|
|
def send_msg(self,msg,to_user = 'default'):
|
||
|
|
"""
|
||
|
|
钉钉发送信息
|
||
|
|
@msg 消息正文
|
||
|
|
"""
|
||
|
|
|
||
|
|
if not self.__dingding_info :
|
||
|
|
return public.returnMsg(False, public.lang("DingTalk information is incorrectly configured."))
|
||
|
|
|
||
|
|
if isinstance(self.__dingding_info['user'],int):
|
||
|
|
return public.returnMsg(False, public.lang("DingTalk configuration error, please reconfigure the DingTalk robot."))
|
||
|
|
|
||
|
|
at_info = ''
|
||
|
|
for user in self.__dingding_info['user']:
|
||
|
|
if re.match("^[0-9]{11,11}$",str(user)): at_info += '@'+user+' '
|
||
|
|
|
||
|
|
msg,title = self.get_send_msg(msg)
|
||
|
|
|
||
|
|
if at_info: msg = msg + '\n\n>' + at_info
|
||
|
|
|
||
|
|
data = {
|
||
|
|
"msgtype": "markdown",
|
||
|
|
"markdown": {
|
||
|
|
"title": "server notification",
|
||
|
|
"text": msg
|
||
|
|
},
|
||
|
|
"at": {
|
||
|
|
"atMobiles": self.__dingding_info['user'],
|
||
|
|
"isAtAll": self.__dingding_info['isAtAll']
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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:
|
||
|
|
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":{}}
|
||
|
|
"""
|
||
|
|
return self.send_msg(data['msg'])
|
||
|
|
|
||
|
|
def __get_default_channel(self):
|
||
|
|
"""
|
||
|
|
@获取默认消息通道
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
if public.readFile(self.__default_pl) == self.__module_name:
|
||
|
|
return True
|
||
|
|
except:pass
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def uninstall(self):
|
||
|
|
if os.path.exists(self.conf_path):
|
||
|
|
os.remove(self.conf_path)
|