mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
Jack
a24f2e23c6
1. Added Mail Marketing Automation trigger tasks 2. Added Mail Marketing Groups Import, Export, Merge 3. Added Mail Marketing Subscribers to support paste import 4. Added Mail Marketing Template Import Export Duplicate 5. Added Mail server add timed automatic reply 6. Added HTTPS Protection function for Website (disables automatic HTTP to HTTPS redirection when enabled) 7. Optimize Website Interface Button Integration 8. Optimize the response speed of WP Toolkit interface
388 lines
14 KiB
Python
388 lines
14 KiB
Python
#coding: utf-8
|
|
#-------------------------------------------------------------------
|
|
# aaPanel
|
|
#-------------------------------------------------------------------
|
|
# Copyright (c) 2015-2099 aaPanel(www.aapanel.com) All rights reserved.
|
|
#-------------------------------------------------------------------
|
|
# Author: hwliang <hwl@aapanel.com>
|
|
#-------------------------------------------------------------------
|
|
|
|
#------------------------------
|
|
# 开机自启模块
|
|
#------------------------------
|
|
|
|
import os,sys,time,json,psutil,re
|
|
import public
|
|
import signal
|
|
|
|
class panelRun:
|
|
__panel_path = public.get_panel_path()
|
|
__run_config_path = '{}/config/run_config'.format(__panel_path)
|
|
__run_pids_path = '{}/logs/run_pids'.format(__panel_path)
|
|
__run_logs_path = '{}/logs/run_logs'.format(__panel_path)
|
|
__log_name = 'Startup items'
|
|
|
|
|
|
def __init__(self):
|
|
if not os.path.exists(self.__run_config_path):
|
|
os.makedirs(self.__run_config_path)
|
|
if not os.path.exists(self.__run_pids_path):
|
|
os.makedirs(self.__run_pids_path)
|
|
|
|
def get_run_list(self,get):
|
|
'''
|
|
@name 获取启动配置列表
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_type: string<启动类型>
|
|
}
|
|
@return list
|
|
'''
|
|
run_type = None
|
|
if 'run_type' in get:
|
|
run_type = get['run_type']
|
|
run_list = []
|
|
for run_name in os.listdir(self.__run_config_path):
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if not os.path.isfile(run_file):
|
|
continue
|
|
|
|
run_info = json.loads(public.readFile(run_file))
|
|
if run_type:
|
|
if run_info['run_type'] != run_type: continue
|
|
|
|
|
|
run_list.append(run_info)
|
|
|
|
return run_list
|
|
|
|
|
|
def get_run_info(self,get = None,run_name = None):
|
|
'''
|
|
@name 获取启动配置信息
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_name: string<启动项名称>
|
|
}
|
|
@return dict
|
|
'''
|
|
if get: run_name = get['run_name']
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if not os.path.isfile(run_file):
|
|
return public.return_msg_gettext(False, public.lang("Configuration file not exist"))
|
|
|
|
run_info = json.loads(public.readFile(run_file))
|
|
return run_info
|
|
|
|
|
|
def create_run(self,get):
|
|
'''
|
|
@name 创建启动配置
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_title: string<启动项显示标题>
|
|
run_name: string<启动项名称> 格式：\w
|
|
run_type: string<启动类型> python shell php node java等，也可以是一个可执行文件的路径 或直接为空
|
|
run_path: string<运行目录>
|
|
run_script: string<启动脚本>
|
|
run_script_args: string<启动脚本参数>
|
|
run_env: list<启动环境变量>
|
|
}
|
|
@return dict
|
|
'''
|
|
run_name = get['run_name']
|
|
run_title = get['run_title']
|
|
run_type = get['run_type']
|
|
run_path = get['run_path']
|
|
run_script = get['run_script']
|
|
run_script_args = get['run_script_args']
|
|
run_env = json.loads(get['run_env'])
|
|
if not os.path.exists(run_path):
|
|
return public.return_msg_gettext(False, public.lang("The specified run directory {} does not exist!", run_path))
|
|
|
|
if not re.match(r'^\w+$',run_name):
|
|
return public.return_msg_gettext(False, public.lang("The startup item name format is incorrect, support: [a-zA-Z0-9_]!"))
|
|
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if os.path.exists(run_file):
|
|
return public.return_msg_gettext(False, public.lang("Launch configuration already exists!"))
|
|
|
|
run_info = {
|
|
'run_title': run_title,
|
|
'run_name': run_name,
|
|
'run_path': run_path,
|
|
'run_script': run_script,
|
|
'run_env':run_env,
|
|
'run_status': 1
|
|
}
|
|
run_info = json.dumps(run_info)
|
|
public.writeFile(run_file,run_info)
|
|
public.write_log_gettext(self.__log_name,'Create startup item [] successful!'.format(run_title))
|
|
return public.return_msg_gettext(True, public.lang("Successfully created"))
|
|
|
|
|
|
def modify_run(self,get):
|
|
'''
|
|
@name 修改启动配置
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_name: string<启动项名称>
|
|
run_title: string<启动项显示标题>
|
|
run_type: string<启动类型>
|
|
run_path: string<启动路径>
|
|
run_script: string<启动脚本>
|
|
run_script_args: string<启动脚本参数>
|
|
}
|
|
@return dict
|
|
'''
|
|
|
|
run_name = get['run_name']
|
|
run_title = get['run_title']
|
|
run_type = get['run_type']
|
|
run_path = get['run_path']
|
|
run_script = get['run_script']
|
|
run_script_args = get['run_script_args']
|
|
run_env = json.loads(get['run_env'])
|
|
|
|
if not os.path.exists(run_path):
|
|
return public.return_msg_gettext(False,'The specified run directory {} does not exist!',(run_path,))
|
|
|
|
if not re.match(r'^\w+$',run_name):
|
|
return public.return_msg_gettext(False, public.lang("The startup item name format is incorrect, support: [a-zA-Z0-9_]!"))
|
|
|
|
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if not os.path.exists(run_file):
|
|
return public.return_msg_gettext(False, public.lang("The launch configuration does not exist!"))
|
|
|
|
run_info = json.loads(public.readFile(run_file))
|
|
run_info['run_title'] = run_title
|
|
run_info['run_path'] = run_path
|
|
run_info['run_script'] = run_script
|
|
run_info['run_env'] = run_env
|
|
run_info = json.dumps(run_info)
|
|
public.writeFile(run_file,run_info)
|
|
public.write_log_gettext(self.__log_name,'Modify startup item [{}] successful!',(run_title,))
|
|
return public.return_msg_gettext(True, public.lang("Successfully modified"))
|
|
|
|
|
|
def remove_run(self,get):
|
|
'''
|
|
@name 删除启动配置
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_name: string<启动项名称>
|
|
}
|
|
@return dict
|
|
'''
|
|
run_name = get['run_name']
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if not os.path.isfile(run_file):
|
|
return public.return_msg_gettext(False, public.lang("The launch configuration does not exist!"))
|
|
|
|
os.remove(run_file)
|
|
public.write_log_gettext(self.__log_name,'Delete startup item [{}] successful!',(run_name,))
|
|
return public.return_msg_gettext(True, public.lang("successfully deleted"))
|
|
|
|
def set_run_status(self,get):
|
|
'''
|
|
@name 设置启动项状态
|
|
@author hwliang<2021-08-06>
|
|
@param get<dict_obj>{
|
|
run_name: string<启动项名称>
|
|
run_status: int<启动项状态>
|
|
}
|
|
@return dict
|
|
'''
|
|
run_name = get['run_name']
|
|
run_status = get['run_status']
|
|
|
|
run_file = '{}/{}'.format(self.__run_config_path,run_name)
|
|
if not os.path.isfile(run_file):
|
|
return public.return_msg_gettext(False, public.lang("launch configuration does not exist!"))
|
|
|
|
run_info = json.loads(public.readFile(run_file))
|
|
run_info['run_status'] = run_status
|
|
run_info = json.dumps(run_info)
|
|
public.writeFile(run_file,run_info)
|
|
public.write_log_gettext(self.__log_name,'Setting startup item [{}] status succeeded!',(run_info['title'],))
|
|
return public.return_msg_gettext(True, public.lang("Setup successfully!"))
|
|
|
|
|
|
def stop_run(self,run_name = None):
|
|
'''
|
|
@name 关闭启动进程
|
|
@author hwliang<2021-08-06>
|
|
@param run_name: string<启动项名称>
|
|
@return dict
|
|
'''
|
|
pid = self.get_run_pid(run_name)
|
|
if not pid: return True
|
|
os.kill(pid,signal.SIGKILL)
|
|
public.write_log_gettext(self.__log_name,'Close startup item [{}] successful!',(run_name,))
|
|
return True
|
|
|
|
|
|
def pid_exists(self,pid):
|
|
'''
|
|
@name 检测PID是否存在
|
|
@author hwliang<2021-08-06>
|
|
@param pid int<PID>
|
|
@return bool
|
|
'''
|
|
if not isinstance(pid,int):
|
|
pid = int(pid)
|
|
if pid == 0:
|
|
return True
|
|
if not os.path.exists('/proc/{}'.format(pid)):
|
|
return False
|
|
return True
|
|
|
|
|
|
def get_run_pid(self,run_name):
|
|
'''
|
|
@name 获取启动项PID
|
|
@author hwliang<2021-08-06>
|
|
@param run_name string<启动项名称>
|
|
@return dict
|
|
'''
|
|
pid_file = '{}/{}.pid'.format(self.__run_pids_path,run_name)
|
|
if not os.path.exists(pid_file):
|
|
return None
|
|
|
|
run_pid = int(public.readFile(pid_file))
|
|
if run_pid is 0:
|
|
return None
|
|
|
|
if not self.pid_exists(run_pid):
|
|
return None
|
|
return run_pid
|
|
|
|
|
|
def get_run_status(self,run_name):
|
|
'''
|
|
@name 获取启动项状态
|
|
@author hwliang<2021-08-06>
|
|
@param run_name string<启动项名称>
|
|
@return dict
|
|
'''
|
|
pid = self.get_run_pid(run_name)
|
|
if not pid: return public.return_msg_gettext(False, public.lang("Not run"))
|
|
process_info = self.get_process_info(pid)
|
|
if not process_info: return public.return_msg_gettext(False, public.lang("Unable to get process information"))
|
|
return process_info
|
|
|
|
def get_process_info(self,pid):
|
|
'''
|
|
@name 获取进程信息
|
|
@author hwliang<2021-08-06>
|
|
@param pid int<PID>
|
|
@return dict
|
|
'''
|
|
process_info = {}
|
|
p = psutil.Process(pid)
|
|
status_ps = {'sleeping':'sleeping','running':'running'}
|
|
with p.oneshot():
|
|
p_mem = p.memory_full_info()
|
|
if p_mem.uss + p_mem.rss + p_mem.pss + p_mem.data == 0: return False
|
|
pio = p.io_counters()
|
|
p_cpus= p.cpu_times()
|
|
p_state = p.status()
|
|
if p_state in status_ps: p_state = status_ps[p_state]
|
|
process_info['exe'] = p.exe()
|
|
process_info['name'] = p.name()
|
|
process_info['pid'] = pid
|
|
process_info['ppid'] = p.ppid()
|
|
process_info['create_time'] = int(p.create_time())
|
|
process_info['status'] = p_state
|
|
process_info['user'] = p.username()
|
|
process_info['memory_used'] = p_mem.uss
|
|
# process_info['cpu_percent'] = self.get_cpu_percent(str(pid),p_cpus,self.new_info['cpu_time'])
|
|
process_info['io_write_bytes'] = pio.write_bytes
|
|
process_info['io_read_bytes'] = pio.read_bytes
|
|
# process_info['io_write_speed'] = self.get_io_write(str(pid),pio.write_bytes)
|
|
# process_info['io_read_speed'] = self.get_io_read(str(pid),pio.read_bytes)
|
|
process_info['connects'] = self.get_connects(pid)
|
|
process_info['threads'] = p.num_threads()
|
|
return process_info
|
|
|
|
|
|
def get_connects(self,pid):
|
|
'''
|
|
@name 获取进程连接数
|
|
@author hwliang<2021-08-06>
|
|
@param pid int<PID>
|
|
@return dict
|
|
'''
|
|
connects = 0
|
|
if pid == 1: return connects
|
|
tp = '/proc/' + str(pid) + '/fd/'
|
|
if not os.path.exists(tp): return connects
|
|
for d in os.listdir(tp):
|
|
fname = tp + d
|
|
if os.path.islink(fname):
|
|
l = os.readlink(fname)
|
|
if l.find('socket:') != -1: connects += 1
|
|
return connects
|
|
|
|
def is_run(self,run_name):
|
|
'''
|
|
@name 检测启动项是否在运行
|
|
@author hwliang<2021-08-06>
|
|
@param run_name string<启动项名称>
|
|
@return bool
|
|
'''
|
|
pid = self.get_run_pid(run_name)
|
|
if not pid: return False
|
|
return True
|
|
|
|
|
|
def get_script_pid(self,run_info):
|
|
'''
|
|
@name 获取脚本进程PID
|
|
@author hwliang<2021-08-06>
|
|
@param run_info dict<脚本文件路径>
|
|
@return int<PID>
|
|
'''
|
|
script_last = run_info['run_script'].split(' ')[0]
|
|
for pid in psutil.pids():
|
|
p = psutil.Process(pid)
|
|
if p.exe() == script_last and p.cwd() == run_info['run_path']:
|
|
return pid
|
|
return None
|
|
|
|
|
|
def start_run(self,run_name):
|
|
'''
|
|
@name 启动指定启动项
|
|
@author hwliang<2021-08-06>
|
|
@param run_name string<启动项名称>
|
|
@return bool
|
|
'''
|
|
run_info = self.get_run_info(run_name)
|
|
if not run_info: return False
|
|
|
|
log_file = '{}/{}.log'.format(self.__run_logs_path,run_name)
|
|
pid_file = '{}/{}.pid'.format(self.__run_pids_path,run_name)
|
|
public.ExecShell("nohup {} 2>&1 >> {} & $! > {}".format(run_info['run_script'],log_file,pid_file),cwd=run_info['run_path'],env=run_info['run_env'])[0]
|
|
time.sleep(1)
|
|
pid = self.get_script_pid(run_info)
|
|
public.writeFile(pid_file,str(pid))
|
|
public.write_log_gettext(self.__log_name, 'Startup {} successful, PID:{}',(run_name,pid,))
|
|
return True
|
|
|
|
|
|
def start(self):
|
|
'''
|
|
@name 启动所有启动项
|
|
@author hwliang<2021-08-06>
|
|
@param
|
|
@return bool
|
|
'''
|
|
run_list = self.get_run_list(public.dict_obj())
|
|
for run_name in run_list:
|
|
if not self.is_run(run_name):
|
|
self.start_run(run_name)
|
|
return True
|
|
|
|
|