92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
import sys
|
|
import json
|
|
import signal
|
|
from datetime import timedelta
|
|
from pathlib import Path
|
|
import importlib
|
|
|
|
# ==================== 动态路径配置 ====================
|
|
from core.path_utils import add_project_root_to_path
|
|
|
|
# 添加项目根路径到sys.path
|
|
PROJECT_ROOT = add_project_root_to_path()
|
|
print(f"[INFO] 项目根路径: {PROJECT_ROOT}")
|
|
print(f"[INFO] Python路径: {sys.path[:3]}")
|
|
|
|
|
|
# ====================================================
|
|
|
|
def load_strategy_class(class_path: str):
|
|
"""动态加载策略类"""
|
|
try:
|
|
# class_path: "futures_trading_strategies.FG.TrendlineBreakoutStrategy.DualModeTrendlineHawkesStrategy2.DualModeTrendlineHawkesStrategy"
|
|
module_path, class_name = class_path.rsplit('.', 1)
|
|
module = importlib.import_module(module_path)
|
|
return getattr(module, class_name)
|
|
except Exception as e:
|
|
print(f"[ERROR] 加载策略类失败 {class_path}: {e}")
|
|
print(f"[ERROR] 请检查项目根路径是否正确: {PROJECT_ROOT}")
|
|
print(f"[ERROR] 当前sys.path: {sys.path}")
|
|
sys.exit(1)
|
|
|
|
|
|
def run_strategy(config_path: str):
|
|
"""通过配置文件运行策略"""
|
|
# 1. 加载配置
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
config = json.load(f)
|
|
|
|
print(f"[INFO] [{config['name']}] 正在启动...")
|
|
|
|
# 2. 动态加载策略类
|
|
strategy_class = load_strategy_class(config["strategy_class"])
|
|
|
|
# 3. 创建API
|
|
from tqsdk import TqApi, TqAuth, TqKq
|
|
api = TqApi(TqKq(), auth=TqAuth("emanresu", "dfgvfgdfgg"))
|
|
|
|
# 4. 准备策略参数
|
|
strategy_params = config["strategy_params"].copy()
|
|
strategy_params["main_symbol"] = config["engine_params"]["symbol"].split(".")[-1]
|
|
|
|
# 5. 创建引擎
|
|
from src.tqsdk_real_engine import TqsdkEngine
|
|
|
|
engine = TqsdkEngine(
|
|
strategy_class=strategy_class,
|
|
strategy_params=strategy_params,
|
|
api=api,
|
|
symbol=config["engine_params"]["symbol"],
|
|
duration_seconds=config["engine_params"]["duration_seconds"],
|
|
roll_over_mode=config["engine_params"]["roll_over_mode"],
|
|
history_length=config["engine_params"]["history_length"],
|
|
close_bar_delta=timedelta(**config["engine_params"]["close_bar_delta"])
|
|
)
|
|
|
|
# 6. 信号处理
|
|
def signal_handler(sig, frame):
|
|
print(f"\n[INFO] [{config['name']}] 收到停止信号 {sig}，正在关闭...")
|
|
api.close()
|
|
sys.exit(0)
|
|
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
# 7. 运行
|
|
try:
|
|
print(f"[INFO] [{config['name']}] 开始运行")
|
|
engine.run()
|
|
except Exception as e:
|
|
print(f"[ERROR] [{config['name']}] 运行出错: {e}")
|
|
sys.exit(1)
|
|
finally:
|
|
api.close()
|
|
print(f"[INFO] [{config['name']}] 已停止")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 3 or sys.argv[1] != "--config":
|
|
print("使用方法: python launcher.py --config <config_file>")
|
|
sys.exit(1)
|
|
|
|
run_strategy(sys.argv[2]) |