359 lines
13 KiB
Python
359 lines
13 KiB
Python
|
#coding: utf-8
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | YakPanel
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Copyright (c) 2015-2020 YakPanel(https://www.yakpanel.com) All rights reserved.
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
# | Author: 沐落 <cjx@yakpanel.com>
|
|||
|
|
# | Author: lx
|
|||
|
|
# +-------------------------------------------------------------------
|
|||
|
|
import sys,os,time,psutil,re
|
|||
|
|
panelPath = "/www/server/panel"
|
|||
|
|
os.chdir(panelPath)
|
|||
|
|
sys.path.insert(0, "class/")
|
|||
|
|
import public,time,panelPush,public
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from YakPanel import cache
|
|||
|
|
except :
|
|||
|
|
from cachelib import SimpleCache
|
|||
|
|
cache = SimpleCache()
|
|||
|
|
|
|||
|
|
class panel_push:
|
|||
|
|
|
|||
|
|
__push = None
|
|||
|
|
pids = None
|
|||
|
|
def __init__(self):
|
|||
|
|
self.__push = panelPush.panelPush()
|
|||
|
|
|
|||
|
|
#-----------------------------------------------------------start 添加推送 ------------------------------------------------------
|
|||
|
|
def get_version_info(self,get):
|
|||
|
|
"""
|
|||
|
|
获取版本信息
|
|||
|
|
"""
|
|||
|
|
data = {}
|
|||
|
|
data['ps'] = ''
|
|||
|
|
data['version'] = '1.2'
|
|||
|
|
data['date'] = '2022-09-20'
|
|||
|
|
data['author'] = 'YakPanel'
|
|||
|
|
data['help'] = 'https://www.yakpanel.com/forum'
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
#名取PID
|
|||
|
|
def getPid(self,pname):
|
|||
|
|
try:
|
|||
|
|
if not self.pids: self.pids = psutil.pids()
|
|||
|
|
for pid in self.pids:
|
|||
|
|
if psutil.Process(pid).name() == pname: return True
|
|||
|
|
return False
|
|||
|
|
except: return True
|
|||
|
|
|
|||
|
|
#检测指定进程是否存活
|
|||
|
|
def checkProcess(self,pid):
|
|||
|
|
try:
|
|||
|
|
if not self.pids: self.pids = psutil.pids()
|
|||
|
|
if int(pid) in self.pids: return True
|
|||
|
|
return False
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
#检查是否启动
|
|||
|
|
def check_run(self,name):
|
|||
|
|
if name == "php-fpm":
|
|||
|
|
status = False
|
|||
|
|
base_path = "/www/server/php"
|
|||
|
|
if not os.path.exists(base_path):
|
|||
|
|
return status
|
|||
|
|
for p in os.listdir(base_path):
|
|||
|
|
pid_file = os.path.join(base_path, p, "var/run/php-fpm.pid")
|
|||
|
|
if os.path.exists(pid_file):
|
|||
|
|
php_pid = int(public.readFile(pid_file))
|
|||
|
|
status = self.checkProcess(php_pid)
|
|||
|
|
if status:
|
|||
|
|
return status
|
|||
|
|
return status
|
|||
|
|
elif name == 'nginx':
|
|||
|
|
status = False
|
|||
|
|
if os.path.exists('/etc/init.d/nginx'):
|
|||
|
|
pidf = '/www/server/nginx/logs/nginx.pid'
|
|||
|
|
if os.path.exists(pidf):
|
|||
|
|
try:
|
|||
|
|
pid = public.readFile(pidf)
|
|||
|
|
status = self.checkProcess(pid)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
return status
|
|||
|
|
elif name == 'apache':
|
|||
|
|
status = False
|
|||
|
|
if os.path.exists('/etc/init.d/httpd'):
|
|||
|
|
pidf = '/www/server/apache/logs/httpd.pid'
|
|||
|
|
if os.path.exists(pidf):
|
|||
|
|
pid = public.readFile(pidf)
|
|||
|
|
status = self.checkProcess(pid)
|
|||
|
|
return status
|
|||
|
|
elif name == 'mysql':
|
|||
|
|
res = public.ExecShell("service mysqld status")
|
|||
|
|
if res and not re.search(r"not\s+running", res[0]):
|
|||
|
|
return True
|
|||
|
|
return False
|
|||
|
|
elif name == 'tomcat':
|
|||
|
|
status = False
|
|||
|
|
if os.path.exists('/www/server/tomcat/logs/catalina-daemon.pid'):
|
|||
|
|
if self.getPid('jsvc'): status = True
|
|||
|
|
if not status:
|
|||
|
|
if self.getPid('java'): status = True
|
|||
|
|
return status
|
|||
|
|
elif name == 'pure-ftpd':
|
|||
|
|
pidf = '/var/run/pure-ftpd.pid'
|
|||
|
|
status = False
|
|||
|
|
if os.path.exists(pidf):
|
|||
|
|
pid = public.readFile(pidf)
|
|||
|
|
status = self.checkProcess(pid)
|
|||
|
|
return status
|
|||
|
|
elif name == 'redis':
|
|||
|
|
status = False
|
|||
|
|
pidf = '/www/server/redis/redis.pid'
|
|||
|
|
if os.path.exists(pidf):
|
|||
|
|
pid = public.readFile(pidf)
|
|||
|
|
status = self.checkProcess(pid)
|
|||
|
|
return status
|
|||
|
|
elif name == 'memcached':
|
|||
|
|
status = False
|
|||
|
|
pidf = '/var/run/memcached.pid'
|
|||
|
|
if os.path.exists(pidf):
|
|||
|
|
pid = public.readFile(pidf)
|
|||
|
|
status = self.checkProcess(pid)
|
|||
|
|
return status
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def get_server_status(self, server_name):
|
|||
|
|
status = self.check_run(server_name)
|
|||
|
|
if status:
|
|||
|
|
return 1
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
@获取推送模块配置
|
|||
|
|
"""
|
|||
|
|
def get_module_config(self,get):
|
|||
|
|
|
|||
|
|
data = []
|
|||
|
|
|
|||
|
|
item = self.__push.format_push_data(push = ["mail",'dingding','weixin',"feishu"],project = 'ssl',type = 'ssl')
|
|||
|
|
item['cycle'] = 30
|
|||
|
|
item['title'] = 'SSL到期提醒'
|
|||
|
|
item['helps'] = ['SSL到期提醒一天只发送一次','证书夹内【剩余天数不足30天】的所有证书.']
|
|||
|
|
data.append(item)
|
|||
|
|
|
|||
|
|
item = self.__push.format_push_data(push = ["mail",'dingding','weixin',"feishu"],project = 'endtime',type = 'endtime')
|
|||
|
|
item['cycle'] = 5
|
|||
|
|
item['title'] = '专业版/企业版到期提醒'
|
|||
|
|
item['helps'] = ['']
|
|||
|
|
data.append(item)
|
|||
|
|
|
|||
|
|
services = ['nginx','apache',"pure-ftpd",'mysql','php-fpm','memcached','redis']
|
|||
|
|
channels = ['mail', 'dingding','weixin','sms',"feishu"]
|
|||
|
|
for x in services:
|
|||
|
|
item = {}
|
|||
|
|
item['project'] = x
|
|||
|
|
item['title'] = '{}服务停止通知'.format(x)
|
|||
|
|
|
|||
|
|
if x == 'other':
|
|||
|
|
item['title'] = '自定义服务停止通知'
|
|||
|
|
else:
|
|||
|
|
ser_name = self.__get_service_name(x)
|
|||
|
|
if self.get_server_status(ser_name) == -1: continue
|
|||
|
|
|
|||
|
|
item['type'] = 'services'
|
|||
|
|
item['keys'] = []
|
|||
|
|
item['interval'] = 300
|
|||
|
|
for x in ['stop']:
|
|||
|
|
item['keys'].append({'key':x,'val':'停止'})
|
|||
|
|
|
|||
|
|
item['push'] = channels
|
|||
|
|
item['helps'] = ['部分服务停止可能会造成业务中断.','短信推送需提前购买，如需帮助请联系 [<a href="https://www.yakpanel.com/forum" target="_blank" class="btlink">YakPanel 社区</a>]']
|
|||
|
|
data.append(item)
|
|||
|
|
import json
|
|||
|
|
public.writeFile('data/push/config/panel_push.json',json.dumps(data))
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def get_push_cycle(self,data):
|
|||
|
|
"""
|
|||
|
|
@获取执行周期
|
|||
|
|
"""
|
|||
|
|
result = {}
|
|||
|
|
for skey in data:
|
|||
|
|
result[skey] = data[skey]
|
|||
|
|
|
|||
|
|
m_cycle =[]
|
|||
|
|
m_type = data[skey]['type']
|
|||
|
|
if m_type in ['endtime','ssl']:
|
|||
|
|
m_cycle.append('剩余{}天时，每天1次'.format(data[skey]['cycle']))
|
|||
|
|
elif m_type in ['services']:
|
|||
|
|
m_cycle.append('服务停止时，每{}秒1次'.format(data[skey]['interval']))
|
|||
|
|
if len(m_cycle) > 0:
|
|||
|
|
result[skey]['m_cycle'] = ''.join(m_cycle)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
#-----------------------------------------------------------end 添加推送 ------------------------------------------------------
|
|||
|
|
"""
|
|||
|
|
@获取服务真实名称
|
|||
|
|
"""
|
|||
|
|
def __get_service_name(self,name):
|
|||
|
|
slist = {"FTP服务端":'pure-ftpd'}
|
|||
|
|
if name in slist: name = slist[name]
|
|||
|
|
|
|||
|
|
return name
|
|||
|
|
"""
|
|||
|
|
@获取推送栏目
|
|||
|
|
"""
|
|||
|
|
def get_total(self):
|
|||
|
|
return True;
|
|||
|
|
|
|||
|
|
def get_push_data(self,data,total):
|
|||
|
|
if data['type'] == 'services':
|
|||
|
|
ser_name = data['project']
|
|||
|
|
ser_name = self.__get_service_name(ser_name)
|
|||
|
|
|
|||
|
|
status = self.get_server_status(ser_name)
|
|||
|
|
if status > 0:
|
|||
|
|
return public.returnMsg(False, public.lang("状态正常，跳过."))
|
|||
|
|
else:
|
|||
|
|
if status == 0:
|
|||
|
|
return self.__get_service_result(data)
|
|||
|
|
return public.returnMsg(False, public.lang("服务未安装，跳过."))
|
|||
|
|
|
|||
|
|
elif data['type'] in ['ssl']:
|
|||
|
|
|
|||
|
|
if time.time() < data['index'] + 86400:
|
|||
|
|
return public.returnMsg(False, public.lang("一天推送一次，跳过."))
|
|||
|
|
|
|||
|
|
import panelSSL
|
|||
|
|
ssl = panelSSL.panelSSL()
|
|||
|
|
clist = []
|
|||
|
|
for x in ssl.GetCertList(None):
|
|||
|
|
timeArray = time.strptime(x['notAfter'], "%Y-%m-%d")
|
|||
|
|
endtime = time.mktime(timeArray)
|
|||
|
|
day = int((endtime - time.time()) / 86400)
|
|||
|
|
if day > data['cycle']: continue
|
|||
|
|
clist.append(x)
|
|||
|
|
|
|||
|
|
return self.__get_ssl_result(data,clist)
|
|||
|
|
|
|||
|
|
elif data['type'] in ['endtime']:
|
|||
|
|
if time.time() < data['index'] + 86400:
|
|||
|
|
return public.returnMsg(False, public.lang("一天推送一次，跳过."))
|
|||
|
|
|
|||
|
|
from pluginAuth import Plugin
|
|||
|
|
softs = Plugin(False).get_plugin_list(True)
|
|||
|
|
if softs['pro'] == 0: return public.returnMsg(False, public.lang("永久专业版，跳过."))
|
|||
|
|
|
|||
|
|
if softs['ltd'] == -2 and softs['pro'] == -2:
|
|||
|
|
pass
|
|||
|
|
else:
|
|||
|
|
pro_data = {}
|
|||
|
|
if softs['ltd'] > 0:
|
|||
|
|
pro_data['endtime'] = softs['ltd']
|
|||
|
|
pro_data['name'] = "Linux企业版"
|
|||
|
|
pro_data['affect'] = '全部企业版插件'
|
|||
|
|
elif softs['pro'] > 0:
|
|||
|
|
pro_data['endtime'] = softs['pro']
|
|||
|
|
pro_data['name'] = "Linux专业版"
|
|||
|
|
pro_data['affect'] = '全部专业版插件'
|
|||
|
|
|
|||
|
|
pro_data['day'] = int((pro_data['endtime'] - time.time()) / 86400)
|
|||
|
|
|
|||
|
|
if pro_data['day'] <= data['cycle']:
|
|||
|
|
return self.__get_ltd_result(data,pro_data)
|
|||
|
|
|
|||
|
|
return public.returnMsg(False, public.lang("未达到阈值，跳过."))
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
@企业版到期提醒
|
|||
|
|
"""
|
|||
|
|
def __get_ltd_result(self,data,pro_data):
|
|||
|
|
result = {'index':time.time()}
|
|||
|
|
|
|||
|
|
for m_module in data['module'].split(','):
|
|||
|
|
result[m_module] = self.__push.format_msg_data()
|
|||
|
|
newline = ""
|
|||
|
|
if m_module in ['dingding','weixin',"feishu","mail"]:
|
|||
|
|
if m_module in ["mail"]:
|
|||
|
|
newline="<br/>"
|
|||
|
|
result[m_module]["title"] = "YakPanel 业务到期提醒"
|
|||
|
|
else:
|
|||
|
|
newline = "\n\n"
|
|||
|
|
result[m_module]['msg'] = "".join((
|
|||
|
|
"#### YakPanel 业务到期提醒"+newline,
|
|||
|
|
">服务器 ："+ public.GetLocalIp() + newline,
|
|||
|
|
">剩余天数："+ str(pro_data['day'] + 1) +" 天"+newline,
|
|||
|
|
">到期产品："+ pro_data['name'] +newline,
|
|||
|
|
">到期时间："+ public.format_date(times =pro_data['endtime']) +newline,
|
|||
|
|
">影响业务："+ pro_data['affect'] +newline,
|
|||
|
|
">通知时间：" + public.format_date() + newline))
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
@ssl到期返回
|
|||
|
|
"""
|
|||
|
|
def __get_ssl_result(self,data,clist):
|
|||
|
|
if len(clist) == 0:
|
|||
|
|
return public.returnMsg(False, public.lang("未找到到期证书，跳过."))
|
|||
|
|
|
|||
|
|
result = {'index':time.time() }
|
|||
|
|
for m_module in data['module'].split(','):
|
|||
|
|
|
|||
|
|
result[m_module] = self.__push.format_msg_data()
|
|||
|
|
newline = ""
|
|||
|
|
if m_module in ['dingding','weixin',"feishu","mail"]:
|
|||
|
|
if m_module in ["mail"]:
|
|||
|
|
newline="<br/>"
|
|||
|
|
result[m_module]["title"] = "YakPanel SSL到期提醒"
|
|||
|
|
else:
|
|||
|
|
newline = "\n\n"
|
|||
|
|
|
|||
|
|
p_msg = "";
|
|||
|
|
for x in clist: p_msg+= " 到期：{} 域名：{}".format(x['notAfter'],x['subject']) + newline
|
|||
|
|
|
|||
|
|
result[m_module]['msg'] ="".join((
|
|||
|
|
"#### YakPanel SSL到期提醒" + newline,
|
|||
|
|
">服务器 ："+ public.GetLocalIp() +newline,
|
|||
|
|
">检测时间：" + public.format_date() +newline,
|
|||
|
|
">About to expire: "+ str(len(clist)) +" "+newline,
|
|||
|
|
p_msg))
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
@服务停止返回
|
|||
|
|
"""
|
|||
|
|
def __get_service_result(self,data):
|
|||
|
|
|
|||
|
|
s_idx = int(time.time())
|
|||
|
|
if s_idx < data['index'] + data['interval']:
|
|||
|
|
return public.returnMsg(False, public.lang("The interval is not reached, skip."))
|
|||
|
|
|
|||
|
|
result = {'index':s_idx}
|
|||
|
|
|
|||
|
|
for m_module in data['module'].split(','):
|
|||
|
|
result[m_module] = self.__push.format_msg_data()
|
|||
|
|
newline = ""
|
|||
|
|
if m_module in ['dingding','weixin',"feishu","mail"]:
|
|||
|
|
if m_module in ["mail"]:
|
|||
|
|
newline="<br/>"
|
|||
|
|
result[m_module]["title"] = "堡塔服务停止告警"
|
|||
|
|
else:
|
|||
|
|
newline = "\n\n"
|
|||
|
|
result[m_module]['msg'] = "".join((
|
|||
|
|
"#### 堡塔服务停止告警" + newline,
|
|||
|
|
">服务器 ："+ public.GetLocalIp() +newline + " ",
|
|||
|
|
">Type of service: "+ data["project"] +newline + " ",
|
|||
|
|
">Service state: 已停止"+newline+" ",
|
|||
|
|
">检测时间："+ public.format_date()))
|
|||
|
|
elif m_module in ['sms']:
|
|||
|
|
result[m_module]['sm_type'] = 'servcies'
|
|||
|
|
result[m_module]['sm_args'] = { 'name':'{}'.format(public.GetConfigValue('title')), 'product':data["project"],'product1':data["project"]}
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
|