113 lines
3.1 KiB
Python
113 lines
3.1 KiB
Python
|
# flake8: noqa
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import multiprocessing
|
|||
|
|
from time import sleep
|
|||
|
|
from datetime import datetime, time
|
|||
|
|
from logging import DEBUG
|
|||
|
|
|
|||
|
|
# 将repostory的目录i，作为根目录，添加到系统环境中。
|
|||
|
|
ROOT_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
|
|||
|
|
if ROOT_PATH not in sys.path:
|
|||
|
|
sys.path.append(ROOT_PATH)
|
|||
|
|
print(f'append {ROOT_PATH} into sys.path')
|
|||
|
|
|
|||
|
|
from vnpy.event import EventEngine, EVENT_TIMER
|
|||
|
|
from vnpy.trader.setting import SETTINGS
|
|||
|
|
from vnpy.trader.engine import MainEngine
|
|||
|
|
from vnpy.trader.utility import load_json
|
|||
|
|
# from vnpy.gateway.gj import GjGateway
|
|||
|
|
from vnpy.app.stock_screener import ScreenerApp
|
|||
|
|
# from vnpy.app.cta_stock import CtaStockApp
|
|||
|
|
# from vnpy.app.cta_crypto.base import EVENT_CTA_LOG
|
|||
|
|
from vnpy.app.rpc_service import RpcServiceApp
|
|||
|
|
# from vnpy.app.algo_broker import AlgoBrokerApp
|
|||
|
|
from vnpy.app.account_recorder import AccountRecorderApp
|
|||
|
|
from vnpy.trader.util_pid import update_pid
|
|||
|
|
|
|||
|
|
# from vnpy.trader.util_monitor import OrderMonitor, TradeMonitor, PositionMonitor, AccountMonitor, LogMonitor
|
|||
|
|
|
|||
|
|
SETTINGS["log.active"] = True
|
|||
|
|
SETTINGS["log.level"] = DEBUG
|
|||
|
|
SETTINGS["log.console"] = True
|
|||
|
|
SETTINGS["log.file"] = True
|
|||
|
|
|
|||
|
|
gateway_name = 'em02'
|
|||
|
|
|
|||
|
|
import types
|
|||
|
|
import traceback
|
|||
|
|
|
|||
|
|
|
|||
|
|
def excepthook(exctype: type, value: Exception, tb: types.TracebackType) -> None:
|
|||
|
|
"""
|
|||
|
|
Raise exception under debug mode
|
|||
|
|
"""
|
|||
|
|
sys.__excepthook__(exctype, value, tb)
|
|||
|
|
|
|||
|
|
msg = "".join(traceback.format_exception(exctype, value, tb))
|
|||
|
|
|
|||
|
|
print(msg, file=sys.stderr)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DaemonService(object):
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.event_engine = EventEngine()
|
|||
|
|
self.g_count = 0
|
|||
|
|
self.last_dt = datetime.now()
|
|||
|
|
|
|||
|
|
# 创建主引擎
|
|||
|
|
self.main_engine = MainEngine(self.event_engine)
|
|||
|
|
|
|||
|
|
self.save_data_time = None
|
|||
|
|
self.save_snapshot_time = None
|
|||
|
|
|
|||
|
|
# 注册定时器，用于判断重连
|
|||
|
|
self.event_engine.register(EVENT_TIMER, self.on_timer)
|
|||
|
|
|
|||
|
|
def on_timer(self, event):
|
|||
|
|
"""定时器执行逻辑，每十秒执行一次"""
|
|||
|
|
|
|||
|
|
# 60秒才执行一次检查
|
|||
|
|
self.g_count += 1
|
|||
|
|
if self.g_count <= 60:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
self.g_count = 0
|
|||
|
|
dt = datetime.now()
|
|||
|
|
|
|||
|
|
if dt.hour != self.last_dt.hour:
|
|||
|
|
self.last_dt = dt
|
|||
|
|
print(u'run_screener.py checkpoint:{0}'.format(dt))
|
|||
|
|
self.main_engine.write_log(u'run_screener.py checkpoint:{0}'.format(dt))
|
|||
|
|
|
|||
|
|
def start(self):
|
|||
|
|
"""
|
|||
|
|
Running in the child process.
|
|||
|
|
"""
|
|||
|
|
SETTINGS["log.file"] = True
|
|||
|
|
|
|||
|
|
# 添加选股引擎
|
|||
|
|
screen_engine = self.main_engine.add_app(ScreenerApp)
|
|||
|
|
screen_engine.init_engine()
|
|||
|
|
|
|||
|
|
self.main_engine.write_log("主引擎创建成功")
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
sleep(1)
|
|||
|
|
if screen_engine.get_all_completed_status():
|
|||
|
|
from vnpy.trader.util_wechat import send_wx_msg
|
|||
|
|
msg = f'{gateway_name}完成所有选股任务'
|
|||
|
|
send_wx_msg(content=msg)
|
|||
|
|
self.main_engine.write_log(msg)
|
|||
|
|
sleep(10)
|
|||
|
|
os._exit(0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
sys.excepthook = excepthook
|
|||
|
|
|
|||
|
|
s = DaemonService()
|
|||
|
|
s.start()
|