from core.trade.orb_trade import ORBStrategy

def main():
    # 初始化ORB策略
    orb_strategy = ORBStrategy(
        initial_capital=25000,
        max_leverage=4,
        risk_per_trade=0.01,
        commission_per_share=0.0005,
    )

    # 1. 获取QQQ的5分钟日内数据（2024-2025，注意：yfinance免费版可能限制历史日内数据，建议用专业数据源）
    orb_strategy.fetch_intraday_data(
        symbol="ETH-USDT", start_date="2025-05-15", end_date="2025-08-20", interval="5m"
    )

    # 2. 生成ORB策略信号
    orb_strategy.generate_orb_signals()

    # 3. 回测策略（盈利目标10R）
    orb_strategy.backtest(profit_target_multiple=10)

    # 4. 绘制净值曲线
    orb_strategy.plot_equity_curve()


if __name__ == "__main__":
    main()