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