81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
|
import sys
|
|||
|
|
import argparse
|
|||
|
|
from pathlib import Path
|
|||
|
|
from core.manager import StrategyManager, print_status_table
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="策略管理系统 - 批量管理你的交易策略",
|
|||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|||
|
|
epilog="""
|
|||
|
|
示例:
|
|||
|
|
# 查看所有策略状态
|
|||
|
|
python start.py status
|
|||
|
|
|
|||
|
|
# 启动所有策略
|
|||
|
|
python start.py start --all
|
|||
|
|
|
|||
|
|
# 启动单个策略（使用标识符）
|
|||
|
|
python start.py start -n DualModeTrendlineHawkesStrategy2_FG
|
|||
|
|
|
|||
|
|
# 停止单个策略
|
|||
|
|
python start.py stop -n DualModeTrendlineHawkesStrategy2_FG
|
|||
|
|
|
|||
|
|
# 重启策略
|
|||
|
|
python start.py restart -n DualModeTrendlineHawkesStrategy2_FG
|
|||
|
|
|
|||
|
|
# 查看日志（最近50行）
|
|||
|
|
python start.py logs -n DualModeTrendlineHawkesStrategy2_FG -t 50
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
parser.add_argument("action", choices=["start", "stop", "restart", "status", "logs"])
|
|||
|
|
parser.add_argument("-n", "--name", help="策略标识符（策略名_品种）")
|
|||
|
|
parser.add_argument("-a", "--all", action="store_true", help="对所有策略执行操作")
|
|||
|
|
parser.add_argument("-c", "--config", default="config/main.json", help="主配置文件路径")
|
|||
|
|
parser.add_argument("-t", "--tail", type=int, default=30, help="查看日志末尾行数")
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
manager = StrategyManager(args.config)
|
|||
|
|
|
|||
|
|
if args.action == "status":
|
|||
|
|
status = manager.get_status()
|
|||
|
|
print_status_table(status)
|
|||
|
|
|
|||
|
|
elif args.action in ["start", "stop"]:
|
|||
|
|
if args.all:
|
|||
|
|
manager.start_all() if args.action == "start" else manager.stop_all()
|
|||
|
|
elif args.name:
|
|||
|
|
manager.start_strategy(args.name) if args.action == "start" else manager.stop_strategy(args.name)
|
|||
|
|
else:
|
|||
|
|
print("❌ 错误: 请指定策略名称 (-n) 或使用 --all")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
elif args.action == "restart":
|
|||
|
|
if not args.name:
|
|||
|
|
print("❌ 错误: 重启操作必须指定策略名称 (-n)")
|
|||
|
|
sys.exit(1)
|
|||
|
|
manager.restart_strategy(args.name)
|
|||
|
|
|
|||
|
|
elif args.action == "logs":
|
|||
|
|
if not args.name:
|
|||
|
|
print("❌ 错误: 查看日志必须指定策略名称 (-n)")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
log_file = Path("logs") / f"{args.name}.log"
|
|||
|
|
if not log_file.exists():
|
|||
|
|
print(f"⚠️ 日志文件不存在: {log_file}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
print(f"\n📄 {args.name} 的日志 (最近{args.tail}行):")
|
|||
|
|
print("=" * 80)
|
|||
|
|
with open(log_file, 'r') as f:
|
|||
|
|
lines = f.readlines()[-args.tail:]
|
|||
|
|
for line in lines:
|
|||
|
|
print(line.rstrip())
|
|||
|
|
print("=" * 80)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|