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()
|