395 lines
12 KiB
Python
395 lines
12 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
# python 3 环境
|
||
# 激活 activate.sh (激活py35 env，启动运行程序
|
||
|
||
import sys
|
||
import time
|
||
from datetime import datetime
|
||
# import commands
|
||
import os
|
||
import subprocess
|
||
import psutil
|
||
|
||
# 将repostory的目录i，作为根目录，添加到系统环境中。
|
||
ROOT_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
|
||
sys.path.append(ROOT_PATH)
|
||
|
||
from vnpy.trader.util_wechat import send_wx_msg
|
||
|
||
# python容器文件
|
||
python_path = '/home/trade/anaconda3/envs/py37/bin/python'
|
||
|
||
# shell 文件，不使用sh
|
||
bash = "/bin/bash"
|
||
|
||
# 配置内容
|
||
# 运行时间段
|
||
# 是否7X24小时运行（数字货币）
|
||
IS_7x24 = False
|
||
# 是否激活夜盘
|
||
ACTIVE_NIGHT = False
|
||
# python 脚本，这里要和activate.sh里面得PROGRAM_NAME 一致
|
||
PROGRAM_NAME = './run_service.py'
|
||
|
||
# 日志目录
|
||
log_path = os.path.abspath(os.path.join(os.getcwd(), 'log'))
|
||
if os.path.isdir(log_path):
|
||
# 如果工作目录下，存在logs子目录，就使用logs子目录
|
||
base_path = os.getcwd()
|
||
else:
|
||
# 使用service.py所在得目录
|
||
base_path = os.path.abspath(os.path.dirname(os.path.realpath(__file__)))
|
||
|
||
# 进程组id保存文件
|
||
gpid_file = os.path.abspath(os.path.join(base_path, 'log', 'gpid.txt'))
|
||
|
||
tmp_cron_file = os.path.abspath(os.path.join(base_path, 'log', 'cron.tmp'))
|
||
|
||
program_file = os.path.join(base_path, 'activate.sh')
|
||
log_file = os.path.abspath(os.path.join(base_path, 'log', 'service.log'))
|
||
error_file = os.path.abspath(os.path.join(base_path, 'log', 'service-error.log'))
|
||
cron_log_file = os.path.abspath(os.path.join(base_path, 'log', 'cron.log'))
|
||
cron_error_file = os.path.abspath(os.path.join(base_path, 'log', 'cron-error.log'))
|
||
null_file = "/dev/null"
|
||
|
||
cron_content = "* * * * * {} {} schedule >>{} 2>>{}".format(python_path, os.path.realpath(__file__), cron_log_file,
|
||
cron_error_file)
|
||
program_command = "nohup {} {} >>{} 2>>{} &".format(bash, program_file, log_file, error_file)
|
||
|
||
USE_GPID = False
|
||
|
||
|
||
def _check_gpid(gpid):
|
||
"""
|
||
检查进程(组)ID
|
||
:param gpid:
|
||
:return: True, 正在运行/ False: 没有运行
|
||
"""
|
||
if not USE_GPID:
|
||
int_gpid = int(gpid)
|
||
return psutil.pid_exists(int_gpid)
|
||
|
||
try:
|
||
# 通过系统子进程，打开ps命令，找到gpid下得所有进程
|
||
p = subprocess.Popen(["ps", "-A", "-o", "pgrp="], stdin=subprocess.PIPE,
|
||
stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False)
|
||
returncode = p.wait()
|
||
except OSError as e:
|
||
print('找不到shell运行命令ps', file=sys.stderr)
|
||
exit(1)
|
||
|
||
# print('returncode1:{}'.format(returncode))
|
||
try:
|
||
p2 = subprocess.Popen("uniq", stdin=p.stdout, stdout=subprocess.PIPE,
|
||
stderr=subprocess.PIPE, shell=False)
|
||
returncode = p2.wait()
|
||
except OSError as e:
|
||
print(u'找不到shell运行命令uniq', file=sys.stderr)
|
||
exit(1)
|
||
|
||
# print('returncode2:{}'.format(returncode))
|
||
for i in p2.stdout.readlines():
|
||
# print (u'p2.line:{}'.format(i))
|
||
if i.decode().strip() == gpid:
|
||
print(u'找到gpid:{}'.format(gpid))
|
||
return True
|
||
|
||
print(u'找不到gpid:{}'.format(gpid))
|
||
return False
|
||
|
||
|
||
def _status():
|
||
"""
|
||
查询当前状态
|
||
:return:
|
||
"""
|
||
print(u'检查{}'.format(gpid_file))
|
||
if os.path.exists(gpid_file):
|
||
with open(gpid_file, 'r') as f:
|
||
gpid = f.read().strip()
|
||
print(u'gpid={}'.format(gpid))
|
||
if gpid != "" and _check_gpid(gpid):
|
||
return gpid
|
||
|
||
return None
|
||
|
||
|
||
def trade_off():
|
||
"""检查现在是否为非交易时间"""
|
||
now = datetime.now()
|
||
|
||
# 数字货币
|
||
if IS_7x24:
|
||
if now.hour == 12 and now.minute == 0:
|
||
return True
|
||
else:
|
||
return False
|
||
|
||
# 国内期货/股票
|
||
a = datetime.now().replace(hour=2, minute=35, second=0, microsecond=0)
|
||
b = datetime.now().replace(hour=9, minute=0, second=0, microsecond=0)
|
||
c = datetime.now().replace(hour=15, minute=30, second=0, microsecond=0)
|
||
d = datetime.now().replace(hour=20, minute=45, second=0, microsecond=0)
|
||
|
||
# 国内期货有夜盘
|
||
if ACTIVE_NIGHT:
|
||
weekend = (now.isoweekday() == 6 and now >= a) or (now.isoweekday() == 7) or (
|
||
now.isoweekday() == 1 and now <= a)
|
||
off = (a <= now <= b) or (c <= now <= d) or weekend
|
||
return off
|
||
else:
|
||
weekend = now.isoweekday() in [6, 7]
|
||
off = now <= b or c <= now or weekend
|
||
return off
|
||
|
||
|
||
def _start():
|
||
"""
|
||
启动服务
|
||
:return:
|
||
"""
|
||
# 获取进程组id
|
||
gpid = _status()
|
||
if trade_off():
|
||
# 属于停止运行期间
|
||
if gpid:
|
||
print(u'现在属于停止运行时间，进程组ID存在,将杀死服务进程:[gpid={}]'.format(gpid))
|
||
import signal
|
||
if USE_GPID:
|
||
# 杀死进程组
|
||
os.killpg(int(gpid), signal.SIGKILL)
|
||
else:
|
||
os.kill(int(gpid), signal.SIGKILL)
|
||
i = 0
|
||
while _status():
|
||
time.sleep(1)
|
||
i += 1
|
||
print(u'杀死进程中，等待{}秒'.format(i))
|
||
if i > 30:
|
||
print(u'杀死进程失败，退出')
|
||
exit(1)
|
||
|
||
print('进程组已停止运行[gpid={}]'.format(gpid))
|
||
send_wx_msg('进程组{}已停止运行[{}]'.format(gpid, base_path))
|
||
else:
|
||
print(u'{} 现在属于停止运行时间，不启动服务'.format(datetime.now()))
|
||
else:
|
||
# 属于运行时间
|
||
if not gpid:
|
||
print(u'{}属于运行时间,将启动服务:{}'.format(datetime.now(), program_command))
|
||
if os.path.isfile(gpid_file):
|
||
print(u'{0}文件存在，先执行删除'.format(gpid_file))
|
||
try:
|
||
os.remove(gpid_file)
|
||
except:
|
||
pass
|
||
|
||
os.popen(program_command)
|
||
i = 0
|
||
while True:
|
||
gpid = _status()
|
||
if gpid:
|
||
print('{}属于运行时间,成功启动服务[gpid={}]'.format(datetime.now(), gpid))
|
||
send_wx_msg('{}属于运行时间,成功启动服务[{},gpid={}]'.format(datetime.now(), base_path, gpid))
|
||
break
|
||
|
||
i += 1
|
||
print(u'启动进程中，等待{}秒'.format(i))
|
||
if i > 30:
|
||
print(u'启动进程失败，退出')
|
||
exit(1)
|
||
time.sleep(1)
|
||
|
||
|
||
else:
|
||
print(u'{}属于运行时间,{}服务已运行'.format(datetime.now(), base_path))
|
||
|
||
|
||
def schedule():
|
||
"""
|
||
crontab 计划执行
|
||
:return:
|
||
"""
|
||
print('======schedule========')
|
||
_start()
|
||
|
||
|
||
def status():
|
||
"""查看状态"""
|
||
print('======status========')
|
||
|
||
gpid = _status()
|
||
if gpid:
|
||
print('{}服务进程[gpid={}]正在运行'.format(base_path, gpid))
|
||
else:
|
||
print('{}服务进程没有运行.'.format(base_path))
|
||
|
||
check_pids_in_cwd(gpid)
|
||
|
||
|
||
# operate的可选字符串为：add, del
|
||
def operate_crontab(operate):
|
||
"""
|
||
操作crontab
|
||
:param operate: add , del
|
||
:return:
|
||
"""
|
||
|
||
try:
|
||
# 从系统命令中，获取定时任务
|
||
p = subprocess.Popen(["crontab", "-l"], stdin=subprocess.PIPE,
|
||
stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False)
|
||
returncode = p.wait()
|
||
except OSError as e:
|
||
print(u"找不到shell运行命令crontab", file=sys.stderr)
|
||
exit(1)
|
||
|
||
remain_cron_list = []
|
||
exist_flag = False
|
||
old_cron_content = ''
|
||
for i in p.stdout.readlines():
|
||
if i.decode("utf-8").find(os.path.realpath(__file__) + " schedule") >= 0:
|
||
old_cron_content = i.decode("utf-8")
|
||
exist_flag = True
|
||
else:
|
||
remain_cron_list.append(i.decode("utf-8"))
|
||
|
||
if operate == "add" and not exist_flag:
|
||
remain_cron_list.append(cron_content)
|
||
remain_cron_list.append("\n")
|
||
with open(tmp_cron_file, 'wb') as f:
|
||
for i in remain_cron_list:
|
||
f.write(i.encode("utf-8"))
|
||
os.popen("crontab {}".format(tmp_cron_file))
|
||
print(u'添加crontab项: {}'.format(cron_content), file=sys.stderr)
|
||
|
||
if operate == "del" and exist_flag:
|
||
with open(tmp_cron_file, 'wb') as f:
|
||
for i in remain_cron_list:
|
||
f.write(i.encode("utf-8"))
|
||
os.popen("crontab {}".format(tmp_cron_file))
|
||
print(u'删除crontab item: {}'.format(old_cron_content), file=sys.stderr)
|
||
|
||
# os.remove(tmp_cron_file)
|
||
|
||
|
||
def check_pids_in_cwd(gpid=None):
|
||
print('检查{}路径下运行得python {}进程'.format(base_path, PROGRAM_NAME))
|
||
|
||
runing_pids = []
|
||
for pid in psutil.pids():
|
||
try:
|
||
p = psutil.Process(pid)
|
||
p_name = p.name()
|
||
if not p_name.endswith('python'):
|
||
continue
|
||
p_cwd = p.cwd()
|
||
if p_cwd != base_path:
|
||
continue
|
||
p_cmdline = p.cmdline()
|
||
if PROGRAM_NAME not in p_cmdline:
|
||
continue
|
||
|
||
runing_pids.append(pid)
|
||
|
||
except:
|
||
pass
|
||
|
||
if len(runing_pids) > 1:
|
||
if gpid is not None:
|
||
if gpid in runing_pids:
|
||
print(u'排除其他pid')
|
||
runing_pids.remove(gpid)
|
||
else:
|
||
print(u'gpid，不在运行清单中，排除首个pid')
|
||
runing_pids.pop(0)
|
||
else:
|
||
print(u'gpid为空，排除首个pid')
|
||
runing_pids.pop(0)
|
||
|
||
for pid in runing_pids:
|
||
try:
|
||
p = psutil.Process(pid)
|
||
print(u'pid:{},name:{},bin:{},path:{},cmd:{}，被终止运行'
|
||
.format(pid, p.name, p.exe(), p.cwd(), p.cmdline()))
|
||
import signal
|
||
os.kill(int(pid), signal.SIGKILL)
|
||
except:
|
||
pass
|
||
|
||
|
||
def start():
|
||
print(u'======start========')
|
||
# 往任务表增加定时计划
|
||
operate_crontab("add")
|
||
print(u'任务表增加定时计划完毕')
|
||
# 执行启动
|
||
# _start()
|
||
print(u'启动{}服务执行完毕'.format(base_path))
|
||
|
||
|
||
def _stop():
|
||
print(u'======stop========')
|
||
# 在任务表删除定时计划
|
||
operate_crontab("del")
|
||
|
||
# 查询进程组id
|
||
gpid = _status()
|
||
if gpid:
|
||
# 进程组存在，杀死进程
|
||
import signal
|
||
# 杀死进程组
|
||
if USE_GPID:
|
||
# 杀死进程组
|
||
os.killpg(int(gpid), signal.SIGKILL)
|
||
else:
|
||
os.kill(int(gpid), signal.SIGKILL)
|
||
i = 0
|
||
while _status():
|
||
time.sleep(1)
|
||
i += 1
|
||
print(u'等待{}秒'.format(i))
|
||
|
||
print(u'{}成功停止{}服务[gpid={}]'.format(datetime.now(), base_path, gpid))
|
||
send_wx_msg(u'{}成功停止{}服务[gpid={}]'.format(datetime.now(), base_path, gpid))
|
||
else:
|
||
print(u'{}服务进程没有运行'.format(base_path))
|
||
|
||
|
||
def stop():
|
||
"""
|
||
停止服务
|
||
:return:
|
||
"""
|
||
_stop()
|
||
print(u'执行停止{}服务完成'.format(base_path))
|
||
|
||
|
||
def restart():
|
||
print(u'======restart========')
|
||
_stop()
|
||
_start()
|
||
print('执行重启{}服务完成'.format(base_path))
|
||
|
||
|
||
if __name__ == '__main__':
|
||
if len(sys.argv) >= 2:
|
||
fun = sys.argv[1]
|
||
else:
|
||
fun = ''
|
||
if fun == 'status':
|
||
status()
|
||
elif fun == 'start':
|
||
start()
|
||
elif fun == 'stop':
|
||
stop()
|
||
elif fun == 'restart':
|
||
restart()
|
||
elif fun == 'schedule':
|
||
schedule()
|
||
else:
|
||
print(u'Usage: {} (status|start|stop|restart)'.format(os.path.basename(__file__)))
|
||
status()
|