340 lines
11 KiB
Python
340 lines
11 KiB
Python
|
#coding: utf-8
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | YakPanel
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Author: hwliang <2020-05-18>
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | 消息提醒
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
import time
|
|||
|
|
import json
|
|||
|
|
import public
|
|||
|
|
try:
|
|||
|
|
from YakPanel import cache
|
|||
|
|
except :
|
|||
|
|
import cachelib
|
|||
|
|
cache = cachelib.SimpleCache()
|
|||
|
|
|
|||
|
|
class panelMessage:
|
|||
|
|
os = 'linux'
|
|||
|
|
|
|||
|
|
def set_send_status(self, id, data):
|
|||
|
|
'''
|
|||
|
|
@name 设置消息发送状态
|
|||
|
|
@author cjxin <2021-04-12>
|
|||
|
|
@param args dict_obj{
|
|||
|
|
id: 消息标识,
|
|||
|
|
data
|
|||
|
|
}
|
|||
|
|
@return dict
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
public.M('messages').where('id=?',id).update(data)
|
|||
|
|
return public.returnMsg(True, public.lang("Setup successful!"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
获取官网推送消息，一天获取一次
|
|||
|
|
"""
|
|||
|
|
def get_cloud_messages(self,args):
|
|||
|
|
|
|||
|
|
# yakpanel 暂时不用
|
|||
|
|
return public.returnMsg(True, public.lang("Synchronization success!"))
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
ret = cache.get('get_cloud_messages')
|
|||
|
|
if ret: return public.returnMsg(True, public.lang("同步成功1!"))
|
|||
|
|
data = {}
|
|||
|
|
data['version'] = public.version()
|
|||
|
|
data['os'] = self.os
|
|||
|
|
sUrl = public.GetConfigValue('home') + '/api/wpanel/get_messages'
|
|||
|
|
import http_requests
|
|||
|
|
http_requests.DEFAULT_TYPE = 'src'
|
|||
|
|
info = http_requests.post(sUrl,data).json()
|
|||
|
|
# info = json.loads(public.httpPost(sUrl,data))
|
|||
|
|
for x in info:
|
|||
|
|
count = public.M('messages').where('level=? and msg=?',(x['level'],x['msg'],)).count()
|
|||
|
|
if count: continue
|
|||
|
|
|
|||
|
|
pdata = {
|
|||
|
|
"level":x['level'],
|
|||
|
|
"msg":x['msg'],
|
|||
|
|
"state":1,
|
|||
|
|
"expire":int(time.time()) + (int(x['expire']) * 86400),
|
|||
|
|
"addtime": int(time.time())
|
|||
|
|
}
|
|||
|
|
public.M('messages').insert(pdata)
|
|||
|
|
cache.set('get_cloud_messages',86400)
|
|||
|
|
return public.returnMsg(True, public.lang("同步成功!"))
|
|||
|
|
except:
|
|||
|
|
return public.returnMsg(False, public.lang("同步失败!"))
|
|||
|
|
|
|||
|
|
def get_messages(self,args = None):
|
|||
|
|
'''
|
|||
|
|
@name 获取消息列表
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@return list
|
|||
|
|
'''
|
|||
|
|
ikey = 'get_message'
|
|||
|
|
data = cache.get(ikey)
|
|||
|
|
if not data:
|
|||
|
|
if not public.is_aarch():
|
|||
|
|
public.run_thread(self.get_cloud_messages,args=(args,))
|
|||
|
|
data = public.M('messages').where('state=? and expire>?',(1,int(time.time()))).order("id desc").select()
|
|||
|
|
cache.set(ikey,data,86400)
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def get_messages_all(self,args = None):
|
|||
|
|
'''
|
|||
|
|
@name 获取所有消息列表
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@return list
|
|||
|
|
'''
|
|||
|
|
public.run_thread(self.get_cloud_messages,args=(args,))
|
|||
|
|
data = public.M('messages').order("id desc").select()
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def get_message_find(self,args = None,id = None):
|
|||
|
|
'''
|
|||
|
|
@name 获取指定消息
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param args dict_obj{
|
|||
|
|
id: 消息标识
|
|||
|
|
}
|
|||
|
|
@return dict
|
|||
|
|
'''
|
|||
|
|
if args:
|
|||
|
|
id = int(args.id)
|
|||
|
|
data = public.M('messages').where('id=?',id).find()
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
|
|||
|
|
def create_message(self,args = None,level=None,msg=None,expire=None):
|
|||
|
|
'''
|
|||
|
|
@name 创建新的消息
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param args dict_obj{
|
|||
|
|
level: 消息级别(info/warning/danger/error),
|
|||
|
|
msg: 消息内容
|
|||
|
|
expire: 过期时间
|
|||
|
|
}
|
|||
|
|
@return dict
|
|||
|
|
'''
|
|||
|
|
if args:
|
|||
|
|
level = args.level
|
|||
|
|
msg = args.msg
|
|||
|
|
expire = args.expire
|
|||
|
|
pdata = {
|
|||
|
|
"level":level,
|
|||
|
|
"msg":msg,
|
|||
|
|
"state":1,
|
|||
|
|
"expire":int(time.time()) + (int(expire) * 86400),
|
|||
|
|
"addtime": int(time.time())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public.M('messages').insert(pdata)
|
|||
|
|
return public.returnMsg(True, public.lang("Created successfully!"))
|
|||
|
|
|
|||
|
|
def status_message(self,args = None,id = None,state = None):
|
|||
|
|
'''
|
|||
|
|
@name 设置消息状态
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param args dict_obj{
|
|||
|
|
id: 消息标识,
|
|||
|
|
state: 消息状态(0.已忽略, 1.正常)
|
|||
|
|
}
|
|||
|
|
@return dict
|
|||
|
|
'''
|
|||
|
|
if args:
|
|||
|
|
id = int(args.id)
|
|||
|
|
state = int(args.state)
|
|||
|
|
public.M('messages').where('id=?',id).setField('state',state)
|
|||
|
|
return public.returnMsg(True, public.lang("Set up successfully!"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def remove_message(self,args = None,id = None):
|
|||
|
|
'''
|
|||
|
|
@name 删除指定消息
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param args dict_obj{
|
|||
|
|
id: 消息标识
|
|||
|
|
}
|
|||
|
|
@return dict
|
|||
|
|
'''
|
|||
|
|
if args:
|
|||
|
|
id = int(args.id)
|
|||
|
|
public.M('messages').where('id=?',id).delete()
|
|||
|
|
return public.returnMsg(True, public.lang("Successfully deleted!"))
|
|||
|
|
|
|||
|
|
def remove_message_level(self,level):
|
|||
|
|
'''
|
|||
|
|
@name 删除指定消息
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param level string(指定级别或标识)
|
|||
|
|
@return bool
|
|||
|
|
'''
|
|||
|
|
public.M('messages').where('(level=? or level=? or level=? or level=?) and state=?',(level,level+'15',level+'7',level+'3',1)).delete()
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def remove_message_all(self):
|
|||
|
|
public.M('messages').where('state=?',(1,)).delete()
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def is_level(self,level):
|
|||
|
|
'''
|
|||
|
|
@name 指定消息是否忽略
|
|||
|
|
@author hwliang <2020-05-18>
|
|||
|
|
@param level string(指定级别或标识)
|
|||
|
|
@return bool
|
|||
|
|
'''
|
|||
|
|
if public.M('messages').where('level=? and state=?',(level,0)).count():
|
|||
|
|
return False
|
|||
|
|
else:
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def init_msg_module(self, module):
|
|||
|
|
"""
|
|||
|
|
初始化消息通道, 迁移自windows
|
|||
|
|
@module 消息通道模块名称
|
|||
|
|
@author lx
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
import os, sys
|
|||
|
|
if not os.path.exists('class/msg'): os.makedirs('class/msg')
|
|||
|
|
panelPath = "/www/server/panel"
|
|||
|
|
|
|||
|
|
sfile = 'class/msg/{}_msg.py'.format(module)
|
|||
|
|
if not os.path.exists(sfile): return False
|
|||
|
|
sys.path.insert(0, "{}/class/msg".format(panelPath))
|
|||
|
|
|
|||
|
|
msg_main = __import__('{}_msg'.format(module))
|
|||
|
|
try:
|
|||
|
|
public.reload_mod(msg_main)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
return eval('msg_main.{}_msg()'.format(module));
|
|||
|
|
except:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def get_default_channel(self, args=None):
|
|||
|
|
"""获取面板默认消息通道
|
|||
|
|
Returns:
|
|||
|
|
channel: str/None，没有安装消息通道的情况下返回None。
|
|||
|
|
"""
|
|||
|
|
default_channel_pl = "/www/server/panel/data/default_msg_channel.pl"
|
|||
|
|
default_channel = public.readFile(default_channel_pl)
|
|||
|
|
if default_channel:
|
|||
|
|
return public.returnMsg(True, default_channel)
|
|||
|
|
return public.returnMsg(False, public.lang(""))
|
|||
|
|
|
|||
|
|
# def get_default_channel(self):
|
|||
|
|
# """获取面板默认消息通道，默认是邮箱，其次默认选择已安装的第一个消息通道
|
|||
|
|
|
|||
|
|
# Returns:
|
|||
|
|
# channel: str/None，没有安装消息通道的情况下返回None。
|
|||
|
|
# """
|
|||
|
|
# from config import config
|
|||
|
|
# c = config()
|
|||
|
|
# get = public.dict_obj()
|
|||
|
|
# configs = c.get_msg_configs(get)
|
|||
|
|
# installed = []
|
|||
|
|
# for channel, obj in configs.items():
|
|||
|
|
# if "setup" in obj and obj["setup"]:
|
|||
|
|
# installed.append(channel)
|
|||
|
|
# if "default" in obj and obj["default"]:
|
|||
|
|
# return channel
|
|||
|
|
# if "mail" in installed:
|
|||
|
|
# return "mail"
|
|||
|
|
# if installed:
|
|||
|
|
# return installed[0]
|
|||
|
|
# return None
|
|||
|
|
|
|||
|
|
def notify(self, args):
|
|||
|
|
"""发送通知
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
args (dict):
|
|||
|
|
title: 消息标题
|
|||
|
|
msg: 消息内容
|
|||
|
|
channel: 消息通道
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
msg = ""
|
|||
|
|
if "msg" in args:
|
|||
|
|
body = args.msg
|
|||
|
|
title = ""
|
|||
|
|
if "title" in args:
|
|||
|
|
title = args.title
|
|||
|
|
sm_type = None
|
|||
|
|
if "sm_type" in args:
|
|||
|
|
sm_type = args.sm_type
|
|||
|
|
sm_args = {}
|
|||
|
|
if "sm_args" in args:
|
|||
|
|
sm_args = json.loads(args.sm_args)
|
|||
|
|
channel = None
|
|||
|
|
channels = []
|
|||
|
|
if "channel" in args:
|
|||
|
|
channel = args.channel
|
|||
|
|
if channel.find(",") != -1:
|
|||
|
|
channels = channel.split(",")
|
|||
|
|
else:
|
|||
|
|
channels = [channel]
|
|||
|
|
if not channel:
|
|||
|
|
channel_res = self.get_default_channel()
|
|||
|
|
if "msg" in channel_res:
|
|||
|
|
channels = [channel_res["msg"]]
|
|||
|
|
if not channels:
|
|||
|
|
return False
|
|||
|
|
try:
|
|||
|
|
from config import config
|
|||
|
|
c = config()
|
|||
|
|
get = public.dict_obj()
|
|||
|
|
msg_channels = c.get_msg_configs(get)
|
|||
|
|
|
|||
|
|
error_channel = []
|
|||
|
|
channel_data = {}
|
|||
|
|
for ch in channels:
|
|||
|
|
msg_data = {}
|
|||
|
|
# 根据不同的消息通道准备不同的内容
|
|||
|
|
if ch == "mail":
|
|||
|
|
# 如果邮箱通知，没有标题直接跳过
|
|||
|
|
if not title: continue
|
|||
|
|
msg_data = {
|
|||
|
|
"msg": body.replace("\n", "<br/>"),
|
|||
|
|
"title": title
|
|||
|
|
}
|
|||
|
|
if ch in ["dingding", "weixin", "feishu"]:
|
|||
|
|
# 钉钉类必须有消息内容
|
|||
|
|
if not body: continue
|
|||
|
|
msg_data["msg"] = body
|
|||
|
|
if ch in ["sms"]:
|
|||
|
|
# 短信必须指定短信模板名
|
|||
|
|
if not sm_type: continue
|
|||
|
|
msg_data["sm_type"] = sm_type
|
|||
|
|
msg_data["sm_args"] = sm_args
|
|||
|
|
if not msg_data:
|
|||
|
|
channel_data[ch] = args
|
|||
|
|
# print("channel data:")
|
|||
|
|
# print(channel_data)
|
|||
|
|
# 即时推送
|
|||
|
|
|
|||
|
|
from panelPush import panelPush
|
|||
|
|
pp = panelPush()
|
|||
|
|
error_count = 0
|
|||
|
|
push_res = pp.push_message_immediately(channel_data)
|
|||
|
|
if push_res["status"]:
|
|||
|
|
channel_res = push_res["msg"]
|
|||
|
|
for ch, res in channel_res.items():
|
|||
|
|
if not res["status"]:
|
|||
|
|
if ch in msg_channels:
|
|||
|
|
error_channel.append(msg_channels[ch]["title"])
|
|||
|
|
error_count +=1
|
|||
|
|
if error_count == len(channels):
|
|||
|
|
return False
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
return False
|