JUKUAN/Lib/future/MAPatternStrategy_v001.py

# 导入函数库
from jqdata import *
from jqdata import finance
import pandas as pd
import numpy as np
from datetime import date, datetime, timedelta, time
import re

# 顺势交易策略 v001
# 基于均线走势（前提条件）+ K线形态（开盘价差、当天价差）的期货交易策略
#
# 核心逻辑：
# 1. 开盘时检查均线走势（MA30<=MA20<=MA10<=MA5为多头，反之为空头）
# 2. 检查开盘价差是否符合方向要求（多头>=0.5%，空头<=-0.5%）
# 3. 14:35和14:55检查当天价差（多头>0，空头<0），满足条件则开仓
# 4. 应用固定止损和动态追踪止盈
# 5. 自动换月移仓

# 设置以便完整打印 DataFrame
pd.set_option('display.max_rows', None)
pd.set_option('display.max_columns', None)
pd.set_option('display.width', None)
pd.set_option('display.max_colwidth', 20)

## 初始化函数，设定基准等等
def initialize(context):
    # 设定沪深300作为基准
    set_benchmark('000300.XSHG')
    # 开启动态复权模式(真实价格)
    set_option('use_real_price', True)
    # 输出内容到日志
    log.info('=' * 60)
    log.info('均线形态交易策略 v001 初始化开始')
    log.info('策略类型: 均线走势 + K线形态')
    log.info('=' * 60)

    ### 期货相关设定 ###
    # 设定账户为金融账户
    set_subportfolios([SubPortfolioConfig(cash=context.portfolio.starting_cash, type='index_futures')])
    # 期货类每笔交易时的手续费是: 买入时万分之0.23,卖出时万分之0.23,平今仓为万分之23
    set_order_cost(OrderCost(open_commission=0.000023, close_commission=0.000023, close_today_commission=0.0023), type='index_futures')
    
    # 设置期货交易的滑点
    set_slippage(StepRelatedSlippage(2))
    
    # 初始化全局变量
    g.usage_percentage = 0.8  # 最大资金使用比例
    g.max_margin_per_position = 20000  # 单个标的最大持仓保证金（元）
    
    # 均线策略参数
    g.ma_periods = [5, 10, 20, 30]  # 均线周期
    g.ma_historical_days = 60  # 获取历史数据天数（确保足够计算MA30）
    g.ma_open_gap_threshold = 0.002  # 方案1开盘价差阈值（0.2%）
    g.ma_pattern_lookback_days = 10  # 历史均线模式一致性检查的天数
    g.ma_pattern_consistency_threshold = 0.8  # 历史均线模式一致性阈值（80%）
    g.check_intraday_spread = False  # 是否检查日内价差（True: 检查, False: 跳过）
    g.ma_proximity_min_threshold = 8  # MA5与MA10贴近计数和的最低阈值
    
    # 均线价差策略方案选择
    g.ma_gap_strategy_mode = 2  # 策略模式选择（1: 原方案, 2: 新方案）
    g.ma_open_gap_threshold2 = 0.002  # 方案2开盘价差阈值（0.2%）
    g.ma_intraday_threshold_scheme2 = 0.005  # 方案2日内变化阈值（0.5%）
    
    # 止损止盈策略参数
    g.fixed_stop_loss_rate = 0.01  # 固定止损比率（1%）
    g.ma_offset_ratio_normal = 0.003  # 均线跟踪止盈常规偏移量（0.3%）
    g.ma_offset_ratio_close = 0.01  # 均线跟踪止盈收盘前偏移量（1%）
    g.days_for_adjustment = 4  # 持仓天数调整阈值
    
    # 输出策略参数
    log.info("均线形态策略参数:")
    log.info(f"  均线周期: {g.ma_periods}")
    log.info(f"  策略模式: 方案{g.ma_gap_strategy_mode}")
    log.info(f"  方案1开盘价差阈值: {g.ma_open_gap_threshold:.1%}")
    log.info(f"  方案2开盘价差阈值: {g.ma_open_gap_threshold2:.1%}")
    log.info(f"  方案2日内变化阈值: {g.ma_intraday_threshold_scheme2:.1%}")
    log.info(f"  历史均线模式检查天数: {g.ma_pattern_lookback_days}天")
    log.info(f"  历史均线模式一致性阈值: {g.ma_pattern_consistency_threshold:.1%}")
    log.info(f"  均线贴近计数阈值: {g.ma_proximity_min_threshold}")
    log.info(f"  是否检查日内价差: {g.check_intraday_spread}")
    log.info(f"  固定止损: {g.fixed_stop_loss_rate:.1%}")
    log.info(f"  均线跟踪止盈常规偏移: {g.ma_offset_ratio_normal:.1%}")
    log.info(f"  均线跟踪止盈收盘前偏移: {g.ma_offset_ratio_close:.1%}")
    log.info(f"  持仓天数调整阈值: {g.days_for_adjustment}天")
    
    # 期货品种完整配置字典
    g.futures_config = {
        # 贵金属
        'AU': {'has_night_session': True, 'margin_rate': {'long': 0.21, 'short': 0.21}, 'multiplier': 1000, 'trading_start_time': '21:00'},
        'AG': {'has_night_session': True, 'margin_rate': {'long': 0.22, 'short': 0.22}, 'multiplier': 15, 'trading_start_time': '21:00'},
        
        # 有色金属
        'CU': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'AL': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'ZN': {'has_night_session': True, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'PB': {'has_night_session': True, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'NI': {'has_night_session': True, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'SN': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'SS': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 5, 'trading_start_time': '21:00'},
        
        # 黑色系
        'RB': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'HC': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'I': {'has_night_session': True, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 100, 'trading_start_time': '21:00'},
        'JM': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 100, 'trading_start_time': '21:00'},
        'J': {'has_night_session': True, 'margin_rate': {'long': 0.25, 'short': 0.25}, 'multiplier': 60, 'trading_start_time': '21:00'},
        
        # 能源化工
        'SP': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'FU': {'has_night_session': True, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'BU': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'RU': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'BR': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'SC': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 1000, 'trading_start_time': '21:00'},
        'NR': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'LU': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'LC': {'has_night_session': False, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 1, 'trading_start_time': '09:00'},
        
        # 化工
        'FG': {'has_night_session': True, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 20, 'trading_start_time': '21:00'},
        'TA': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'MA': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'SA': {'has_night_session': True, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 20, 'trading_start_time': '21:00'},
        'L': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'V': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'EG': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'PP': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'EB': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'PG': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 20, 'trading_start_time': '21:00'},
        
        # 农产品
        'RM': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'OI': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'CF': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'SR': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'PF': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'C': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'CS': {'has_night_session': True, 'margin_rate': {'long': 0.11, 'short': 0.11}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'CY': {'has_night_session': True, 'margin_rate': {'long': 0.11, 'short': 0.11}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'A': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'B': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'M': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'Y': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'P': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '21:00'},
        
        # 无夜盘品种
        'IF': {'has_night_session': False, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 300, 'trading_start_time': '09:30'},
        'IH': {'has_night_session': False, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 300, 'trading_start_time': '09:30'},
        'IC': {'has_night_session': False, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 200, 'trading_start_time': '09:30'},
        'IM': {'has_night_session': False, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 200, 'trading_start_time': '09:30'},
        'AP': {'has_night_session': False, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 10, 'trading_start_time': '09:00'},
        'CJ': {'has_night_session': False, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 5, 'trading_start_time': '09:00'},
        'PK': {'has_night_session': False, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 5, 'trading_start_time': '09:00'},
        'JD': {'has_night_session': False, 'margin_rate': {'long': 0.11, 'short': 0.11}, 'multiplier': 10, 'trading_start_time': '09:00'},
        'LH': {'has_night_session': False, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 16, 'trading_start_time': '09:00'},
        'T': {'has_night_session': False, 'margin_rate': {'long': 0.03, 'short': 0.03}, 'multiplier': 1000000, 'trading_start_time': '09:30'},
        'PS': {'has_night_session': False, 'margin_rate': {'long': 0.16, 'short': 0.16}, 'multiplier': 3, 'trading_start_time': '09:00'},
        'UR': {'has_night_session': False, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 20, 'trading_start_time': '09:00'},
        'MO': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 100, 'trading_start_time': '21:00'},
        'LF': {'has_night_session': False, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 1, 'trading_start_time': '09:30'},
        'HO': {'has_night_session': False, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 100, 'trading_start_time': '09:30'},
        'LR': {'has_night_session': True, 'margin_rate': {'long': 0.21, 'short': 0.21}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'LG': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'FB': {'has_night_session': True, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'PX': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'PM': {'has_night_session': True, 'margin_rate': {'long': 0.2, 'short': 0.2}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'EC': {'has_night_session': False, 'margin_rate': {'long': 0.23, 'short': 0.23}, 'multiplier': 50, 'trading_start_time': '09:00'},
        'RR': {'has_night_session': True, 'margin_rate': {'long': 0.11, 'short': 0.11}, 'multiplier': 10, 'trading_start_time': '21:00'},
        'OP': {'has_night_session': False, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 40, 'trading_start_time': '09:00'},
        'IO': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 1, 'trading_start_time': '21:00'},
        'BC': {'has_night_session': True, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '21:00'},
        'WH': {'has_night_session': False, 'margin_rate': {'long': 0.2, 'short': 0.2}, 'multiplier': 20, 'trading_start_time': '09:00'},
        'SH': {'has_night_session': True, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 30, 'trading_start_time': '21:00'},
        'RI': {'has_night_session': False, 'margin_rate': {'long': 0.21, 'short': 0.21}, 'multiplier': 20, 'trading_start_time': '09:00'},
        'TS': {'has_night_session': False, 'margin_rate': {'long': 0.015, 'short': 0.015}, 'multiplier': 2000000, 'trading_start_time': '09:30'},
        'JR': {'has_night_session': False, 'margin_rate': {'long': 0.21, 'short': 0.21}, 'multiplier': 20, 'trading_start_time': '09:00'},
        'AD': {'has_night_session': False, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 10, 'trading_start_time': '09:00'},
        'BB': {'has_night_session': False, 'margin_rate': {'long': 0.19, 'short': 0.19}, 'multiplier': 500, 'trading_start_time': '09:00'},
        'PL': {'has_night_session': False, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 20, 'trading_start_time': '09:00'},
        'RS': {'has_night_session': False, 'margin_rate': {'long': 0.26, 'short': 0.26}, 'multiplier': 10, 'trading_start_time': '09:00'},
        'SI': {'has_night_session': False, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '09:00'},
        'ZC': {'has_night_session': True, 'margin_rate': {'long': 0.56, 'short': 0.56}, 'multiplier': 100, 'trading_start_time': '21:00'},
        'SM': {'has_night_session': False, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 5, 'trading_start_time': '09:00'},
        'AO': {'has_night_session': True, 'margin_rate': {'long': 0.17, 'short': 0.17}, 'multiplier': 20, 'trading_start_time': '21:00'},
        'TL': {'has_night_session': False, 'margin_rate': {'long': 0.045, 'short': 0.045}, 'multiplier': 1000000, 'trading_start_time': '09:00'},
        'SF': {'has_night_session': False, 'margin_rate': {'long': 0.14, 'short': 0.14}, 'multiplier': 5, 'trading_start_time': '09:00'},
        'WR': {'has_night_session': False, 'margin_rate': {'long': 0.15, 'short': 0.15}, 'multiplier': 10, 'trading_start_time': '09:00'},
        'PR': {'has_night_session': True, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 15, 'trading_start_time': '21:00'},
        'TF': {'has_night_session': False, 'margin_rate': {'long': 0.022, 'short': 0.022}, 'multiplier': 1000000, 'trading_start_time': '09:00'},
        'VF': {'has_night_session': False, 'margin_rate': {'long': 0.12, 'short': 0.12}, 'multiplier': 1, 'trading_start_time': '09:00'},
        'BZ': {'has_night_session': False, 'margin_rate': {'long': 0.13, 'short': 0.13}, 'multiplier': 30, 'trading_start_time': '09:00'},
    }
    
    # 策略品种选择策略配置
    # 方案1：全品种策略 - 考虑所有配置的期货品种
    g.strategy_focus_symbols = ['IC', 'LH']  # 空列表表示考虑所有品种
    
    # 方案2：精选品种策略 - 只交易流动性较好的特定品种（如需使用请取消下行注释）
    # g.strategy_focus_symbols = ['RM', 'CJ', 'CY', 'JD', 'L', 'LC', 'SF', 'SI']
    
    log.info(f"品种选择策略: {'全品种策略（覆盖所有配置品种）' if not g.strategy_focus_symbols else '精选品种策略（' + str(len(g.strategy_focus_symbols)) + '个品种）'}")
    
    # 交易记录和数据存储
    g.trade_history = {}  # 持仓记录 {symbol: {'entry_price': xxx, 'direction': xxx, ...}}
    g.daily_ma_candidates = {}  # 通过均线和开盘价差检查的候选品种 {symbol: {'direction': 'long'/'short', 'open_price': xxx, ...}}
    g.today_trades = []  # 当日交易记录
    g.excluded_contracts = {}  # 每日排除的合约缓存 {dominant_future: {'reason': 'ma_trend'/'open_gap', 'trading_day': xxx}}
    g.ma_checked_underlyings = {}  # 记录各品种在交易日的均线检查状态 {symbol: trading_day}
    g.last_ma_trading_day = None  # 最近一次均线检查所属交易日
    
    # 定时任务设置
    # 夜盘开始（21:05） - 均线和开盘价差检查
    run_daily(check_ma_trend_and_open_gap, time='21:05:00', reference_security='IF1808.CCFX')
    
    # 日盘开始 - 均线和开盘价差检查
    run_daily(check_ma_trend_and_open_gap, time='09:05:00', reference_security='IF1808.CCFX')
    run_daily(check_ma_trend_and_open_gap, time='09:35:00', reference_security='IF1808.CCFX')
    
    # 盘中价差检查和开仓（14:35和14:55）
    run_daily(check_intraday_price_diff, time='14:35:00', reference_security='IF1808.CCFX')
    run_daily(check_intraday_price_diff, time='14:55:00', reference_security='IF1808.CCFX')
    
    # 夜盘止损止盈检查
    run_daily(check_stop_loss_profit, time='21:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='21:35:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='22:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='22:35:00', reference_security='IF1808.CCFX')
    
    # 日盘止损止盈检查
    run_daily(check_stop_loss_profit, time='09:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='09:35:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='10:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='10:35:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='11:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='11:25:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='13:35:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='14:05:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='14:35:00', reference_security='IF1808.CCFX')
    run_daily(check_stop_loss_profit, time='14:55:00', reference_security='IF1808.CCFX')
    
    # 收盘后
    run_daily(after_market_close, time='15:30:00', reference_security='IF1808.CCFX')
    
    log.info('=' * 60)

############################ 主程序执行函数 ###################################

def get_current_trading_day(current_dt):
    """根据当前时间推断对应的期货交易日"""
    current_date = current_dt.date()
    current_time = current_dt.time()

    trade_days = get_trade_days(end_date=current_date, count=1)
    if trade_days and trade_days[0] == current_date:
        trading_day = current_date
    else:
        next_days = get_trade_days(start_date=current_date, count=1)
        trading_day = next_days[0] if next_days else current_date

    if current_time >= time(20, 59):
        next_trade_days = get_trade_days(start_date=trading_day, count=2)
        if len(next_trade_days) >= 2:
            return next_trade_days[1]
        if len(next_trade_days) == 1:
            return next_trade_days[0]
    return trading_day


def normalize_trade_day_value(value):
    """将交易日对象统一转换为 datetime.date"""
    if isinstance(value, date) and not isinstance(value, datetime):
        return value
    if isinstance(value, datetime):
        return value.date()
    if hasattr(value, 'to_pydatetime'):
        return value.to_pydatetime().date()
    try:
        return pd.Timestamp(value).date()
    except Exception:
        return value


def check_ma_trend_and_open_gap(context):
    """阶段一：开盘时均线走势和开盘价差检查（一天一次）"""
    log.info("=" * 60)
    current_trading_day = get_current_trading_day(context.current_dt)
    log.info(f"执行均线走势和开盘价差检查 - 时间: {context.current_dt}, 交易日: {current_trading_day}")
    log.info("=" * 60)
    
    # 先检查换月移仓
    position_auto_switch(context)
    
    # 检查是否进入新交易日，必要时清空缓存
    if g.last_ma_trading_day != current_trading_day:
        if g.excluded_contracts:
            log.info(f"交易日切换至 {current_trading_day}，清空上一交易日的排除缓存")
        g.excluded_contracts = {}
        g.ma_checked_underlyings = {}
        g.last_ma_trading_day = current_trading_day

    # 获取当前时间
    current_time = str(context.current_dt.time())[:5]  # HH:MM格式
    
    # 筛选可交易品种（根据交易开始时间判断）
    focus_symbols = g.strategy_focus_symbols if g.strategy_focus_symbols else list(g.futures_config.keys())
    tradable_symbols = []
    
    # 根据当前时间确定可交易的时段
    # 21:05 -> 仅接受21:00开盘的合约
    # 09:05 -> 接受09:00或21:00开盘的合约
    # 09:35 -> 接受所有时段（21:00, 09:00, 09:30）的合约
    for symbol in focus_symbols:
        trading_start_time = get_futures_config(symbol, 'trading_start_time', '09:05')
        should_trade = False
        
        if current_time == '21:05':
            # 夜盘开盘：仅接受21:00开盘的品种
            should_trade = trading_start_time.startswith('21:00')
        elif current_time == '09:05':
            # 日盘早盘：接受21:00和09:00开盘的品种
            should_trade = trading_start_time.startswith('21:00') or trading_start_time.startswith('09:00')
        elif current_time == '09:35':
            # 日盘晚开：接受所有品种（21:00, 09:00, 09:30）
            should_trade = True
        
        if should_trade:
            tradable_symbols.append(symbol)
    
    if not tradable_symbols:
        log.info(f"当前时间 {current_time} 无品种开盘，跳过检查")
        return
    
    log.info(f"当前时间 {current_time} 开盘品种: {tradable_symbols}")
    
    # 对每个品种执行均线和开盘价差检查
    for symbol in tradable_symbols:
        if g.ma_checked_underlyings.get(symbol) == current_trading_day:
            log.info(f"{symbol} 已在交易日 {current_trading_day} 完成均线检查，跳过本次执行")
            continue

        try:
            g.ma_checked_underlyings[symbol] = current_trading_day
            # 获取主力合约
            dominant_future = get_dominant_future(symbol)
            # log.debug(f"{symbol} 主力合约: {dominant_future}")
            if not dominant_future:
                log.info(f"{symbol} 未找到主力合约，跳过")
                continue
            
            # 检查是否在排除缓存中（当日已检查过但不符合条件）
            if dominant_future in g.excluded_contracts:
                excluded_info = g.excluded_contracts[dominant_future]
                if excluded_info['trading_day'] == current_trading_day:
                    # log.debug(f"{symbol} 在排除缓存中（原因: {excluded_info['reason']}），跳过")
                    continue
                else:
                    # 新的一天，从缓存中移除（会在after_market_close统一清理，这里也做兜底）
                    del g.excluded_contracts[dominant_future]
            
            # 检查是否已有持仓
            if check_symbol_prefix_match(dominant_future, set(g.trade_history.keys())):
                log.info(f"{symbol} 已有持仓，跳过")
                continue
            
            # 获取历史数据（需要足够计算MA30）
            # 使用get_price获取数据，可以正确处理夜盘品种
            # 注意：historical_data最后一行是昨天的数据，不包含今天的数据
            historical_data = get_price(dominant_future, end_date=context.current_dt, 
                                       frequency='1d', fields=['open', 'close', 'high', 'low'], 
                                       count=g.ma_historical_days)
            
            if historical_data is None or len(historical_data) < max(g.ma_periods):
                log.info(f"{symbol} 历史数据不足，跳过")
                continue

            previous_trade_days = get_trade_days(end_date=current_trading_day, count=2)
            previous_trade_days = [normalize_trade_day_value(d) for d in previous_trade_days]
            previous_trading_day = None
            if len(previous_trade_days) >= 2:
                previous_trading_day = previous_trade_days[-2]
            elif len(previous_trade_days) == 1 and previous_trade_days[0] < current_trading_day:
                previous_trading_day = previous_trade_days[0]

            if previous_trading_day is None:
                log.info(f"{symbol} 无法确定前一交易日，跳过")
                continue

            historical_dates = historical_data.index.date
            match_indices = np.where(historical_dates == previous_trading_day)[0]

            if len(match_indices) == 0:
                earlier_indices = np.where(historical_dates < previous_trading_day)[0]
                if len(earlier_indices) == 0:
                    log.info(f"{symbol} 历史数据缺少 {previous_trading_day} 之前的记录，跳过")
                    continue
                match_indices = [earlier_indices[-1]]

            data_upto_yesterday = historical_data.iloc[:match_indices[-1] + 1]
            # log.debug(f"data_upto_yesterday: {data_upto_yesterday}")
            yesterday_data = data_upto_yesterday.iloc[-1]
            yesterday_close = yesterday_data['close']
            
            # 获取今天的开盘价（使用get_current_data API）
            current_data = get_current_data()[dominant_future]
            today_open = current_data.day_open
            
            # log.info(f"  历史数据时间范围: {historical_data.index[0]} 至 {historical_data.index[-1]}")
            
            # 计算昨天的均线值（使用截至前一交易日的数据）
            ma_values = calculate_ma_values(data_upto_yesterday, g.ma_periods)
            ma_proximity_counts = calculate_ma_proximity_counts(data_upto_yesterday, g.ma_periods, g.ma_pattern_lookback_days)
            
            log.info(f"{symbol}({dominant_future}) 均线检查:")
            # log.debug(f"yesterday_data: {yesterday_data}")
            # log.info(f"  昨收: {yesterday_close:.2f}, 今开: {today_open:.2f}")
            # log.info(f"  昨日均线 - MA5: {ma_values['MA5']:.2f}, MA10: {ma_values['MA10']:.2f}, "
            #         f"MA20: {ma_values['MA20']:.2f}, MA30: {ma_values['MA30']:.2f}")
            log.info(f"  均线贴近统计: {ma_proximity_counts}")
            proximity_sum = ma_proximity_counts.get('MA5', 0) + ma_proximity_counts.get('MA10', 0)
            if proximity_sum < g.ma_proximity_min_threshold:
                log.info(f"  {symbol}({dominant_future}) ✗ 均线贴近计数不足，MA5+MA10={proximity_sum} < {g.ma_proximity_min_threshold}，跳过")
                g.excluded_contracts[dominant_future] = {
                    'reason': 'ma_proximity',
                    'trading_day': current_trading_day
                }
                continue
            
            # 判断均线走势（使用新的灵活模式检查）
            direction = None
            if check_ma_pattern(ma_values, 'long'):
                direction = 'long'
                # log.info(f"  {symbol}({dominant_future}) 均线走势判断: 多头排列")
            elif check_ma_pattern(ma_values, 'short'):
                direction = 'short'
                # log.info(f"  {symbol}({dominant_future}) 均线走势判断: 空头排列")
            else:
                # log.info(f"  均线走势判断: 不符合多头或空头排列，跳过")
                # 将不符合条件的合约加入排除缓存
                g.excluded_contracts[dominant_future] = {
                    'reason': 'ma_trend',
                    'trading_day': current_trading_day
                }
                continue
            
            # 检查历史均线模式一致性
            consistency_passed, consistency_ratio = check_historical_ma_pattern_consistency(
                historical_data, direction, g.ma_pattern_lookback_days, g.ma_pattern_consistency_threshold
            )
            
            if not consistency_passed:
                log.info(f"  {symbol}({dominant_future}) ✗ 历史均线模式一致性不足 "
                        f"({consistency_ratio:.1%} < {g.ma_pattern_consistency_threshold:.1%})，跳过")
                g.excluded_contracts[dominant_future] = {
                    'reason': 'ma_consistency',
                    'trading_day': current_trading_day
                }
                continue
            else:
                log.info(f"  {symbol}({dominant_future}) ✓ 历史均线模式一致性检查通过 "
                        f"({consistency_ratio:.1%} >= {g.ma_pattern_consistency_threshold:.1%})")
            
            # 计算开盘价差比例
            open_gap_ratio = (today_open - yesterday_close) / yesterday_close
            
            log.info(f"  开盘价差检查: 昨收 {yesterday_close:.2f}, 今开 {today_open:.2f}, "
                    f"价差比例 {open_gap_ratio:.2%}")
            
            # 检查开盘价差是否符合方向要求
            gap_check_passed = False
            
            if g.ma_gap_strategy_mode == 1:
                # 方案1：多头检查上跳，空头检查下跳
                if direction == 'long' and open_gap_ratio >= g.ma_open_gap_threshold:
                    log.info(f"  {symbol}({dominant_future}) ✓ 方案1多头开盘价差检查通过 ({open_gap_ratio:.2%} >= {g.ma_open_gap_threshold:.2%})")
                    gap_check_passed = True
                elif direction == 'short' and open_gap_ratio <= -g.ma_open_gap_threshold:
                    log.info(f"  {symbol}({dominant_future}) ✓ 方案1空头开盘价差检查通过 ({open_gap_ratio:.2%} <= {-g.ma_open_gap_threshold:.2%})")
                    gap_check_passed = True
            elif g.ma_gap_strategy_mode == 2:
                # 方案2：多头检查下跳，空头检查上跳
                if direction == 'long' and open_gap_ratio <= -g.ma_open_gap_threshold2:
                    log.info(f"  {symbol}({dominant_future}) ✓ 方案2多头开盘价差检查通过 ({open_gap_ratio:.2%} <= {-g.ma_open_gap_threshold2:.2%})")
                    gap_check_passed = True
                elif direction == 'short' and open_gap_ratio >= g.ma_open_gap_threshold2:
                    log.info(f"  {symbol}({dominant_future}) ✓ 方案2空头开盘价差检查通过 ({open_gap_ratio:.2%} >= {g.ma_open_gap_threshold2:.2%})")
                    gap_check_passed = True
            
            if not gap_check_passed:
                # log.info(f"  ✗ 开盘价差不符合方案{g.ma_gap_strategy_mode} {direction}方向要求，跳过")
                # 将不符合条件的合约加入排除缓存
                g.excluded_contracts[dominant_future] = {
                    'reason': 'open_gap',
                    'trading_day': current_trading_day
                }
                continue
            
            # 将通过检查的品种加入候选列表
            g.daily_ma_candidates[dominant_future] = {
                'symbol': symbol,
                'direction': direction,
                'open_price': today_open,
                'yesterday_close': yesterday_close,
                'ma_values': ma_values
            }
            
            log.info(f"  ✓✓ {symbol} 通过均线和开盘价差检查，加入候选列表")
            
        except Exception as e:
            g.ma_checked_underlyings.pop(symbol, None)
            log.warning(f"{symbol} 检查时出错: {str(e)}")
            continue
    
    log.info(f"候选列表更新完成，当前候选品种: {list(g.daily_ma_candidates.keys())}")
    log.info("=" * 60)

def check_intraday_price_diff(context):
    """阶段二：盘中价差检查和开仓（14:35和14:55）"""
    log.info("=" * 60)
    log.info(f"执行当天价差检查和开仓逻辑 - 时间: {context.current_dt}")
    log.info("=" * 60)
    
    # 先检查换月移仓
    position_auto_switch(context)
    
    if not g.daily_ma_candidates:
        log.info("当前无候选品种，跳过")
        return
    
    log.info(f"候选品种数量: {len(g.daily_ma_candidates)}")
    
    # 遍历候选品种
    candidates_to_remove = []
    
    for dominant_future, candidate_info in g.daily_ma_candidates.items():
        try:
            symbol = candidate_info['symbol']
            direction = candidate_info['direction']
            open_price = candidate_info['open_price']
            
            # 再次检查是否已有持仓
            if check_symbol_prefix_match(dominant_future, set(g.trade_history.keys())):
                log.info(f"{symbol} 已有持仓，从候选列表移除")
                candidates_to_remove.append(dominant_future)
                continue
            
            # 获取当前价格
            current_data = get_current_data()[dominant_future]
            current_price = current_data.last_price
            
            # 计算当天价差
            intraday_diff = current_price - open_price
            intraday_diff_ratio = intraday_diff / open_price  # 计算相对变化比例
            
            log.info(f"{symbol}({dominant_future}) 当天价差检查:")
            log.info(f"  方向: {direction}, 开盘价: {open_price:.2f}, 当前价: {current_price:.2f}, "
                    f"当天价差: {intraday_diff:.2f}, 变化比例: {intraday_diff_ratio:.2%}")
            
            # 判断是否满足开仓条件
            should_open = False
            
            if g.ma_gap_strategy_mode == 1:
                # 方案1：根据参数决定是否检查日内价差
                if not g.check_intraday_spread:
                    # 跳过日内价差检查，直接允许开仓
                    log.info(f"  方案1跳过日内价差检查（check_intraday_spread=False）")
                    should_open = True
                elif direction == 'long' and intraday_diff > 0:
                    log.info(f"  ✓ 方案1多头当天价差检查通过 ({intraday_diff:.2f} > 0)")
                    should_open = True
                elif direction == 'short' and intraday_diff < 0:
                    log.info(f"  ✓ 方案1空头当天价差检查通过 ({intraday_diff:.2f} < 0)")
                    should_open = True
                else:
                    log.info(f"  ✗ 方案1当天价差不符合{direction}方向要求")
            elif g.ma_gap_strategy_mode == 2:
                # 方案2：强制检查日内变化，使用专用阈值
                if direction == 'long' and intraday_diff_ratio >= g.ma_intraday_threshold_scheme2:
                    log.info(f"  ✓ 方案2多头日内变化检查通过 ({intraday_diff_ratio:.2%} >= {g.ma_intraday_threshold_scheme2:.2%})")
                    should_open = True
                elif direction == 'short' and intraday_diff_ratio <= -g.ma_intraday_threshold_scheme2:
                    log.info(f"  ✓ 方案2空头日内变化检查通过 ({intraday_diff_ratio:.2%} <= {-g.ma_intraday_threshold_scheme2:.2%})")
                    should_open = True
                else:
                    log.info(f"  ✗ 方案2日内变化不符合{direction}方向要求（阈值: ±{g.ma_intraday_threshold_scheme2:.2%}）")
            
            if should_open:
                # 执行开仓
                log.info(f"  准备开仓: {symbol} {direction}")
                target_hands = calculate_target_hands(context, dominant_future, direction)
                
                if target_hands > 0:
                    success = open_position(context, dominant_future, target_hands, direction, 
                                          f'均线形态开仓')
                    if success:
                        log.info(f"  ✓✓ {symbol} 开仓成功，从候选列表移除")
                        candidates_to_remove.append(dominant_future)
                    else:
                        log.warning(f"  ✗ {symbol} 开仓失败")
                else:
                    log.warning(f"  ✗ {symbol} 计算目标手数为0，跳过开仓")
                    
        except Exception as e:
            log.warning(f"{dominant_future} 处理时出错: {str(e)}")
            continue
    
    # 从候选列表中移除已开仓的品种
    for future in candidates_to_remove:
        if future in g.daily_ma_candidates:
            del g.daily_ma_candidates[future]
    
    log.info(f"剩余候选品种: {list(g.daily_ma_candidates.keys())}")
    log.info("=" * 60)

def check_stop_loss_profit(context):
    """阶段三：止损止盈检查（所有时间点）"""
    # 先检查换月移仓
    position_auto_switch(context)
    
    # 获取当前时间
    current_time = str(context.current_dt.time())[:2]
    
    # 判断是否为夜盘时间
    is_night_session = (current_time in ['21', '22', '23', '00', '01', '02'])
    
    # 遍历所有持仓进行止损止盈检查
    subportfolio = context.subportfolios[0]
    long_positions = list(subportfolio.long_positions.values())
    short_positions = list(subportfolio.short_positions.values())
    
    closed_count = 0
    skipped_count = 0
    
    for position in long_positions + short_positions:
        security = position.security
        underlying_symbol = security.split('.')[0][:-4]
        
        # 检查交易时间适配性
        has_night_session = get_futures_config(underlying_symbol, 'has_night_session', False)
        
        # 如果是夜盘时间，但品种不支持夜盘交易，则跳过
        if is_night_session and not has_night_session:
            skipped_count += 1
            continue
        
        # 执行止损止盈检查
        if check_position_stop_loss_profit(context, position):
            closed_count += 1
    
    if closed_count > 0:
        log.info(f"执行了 {closed_count} 次止损止盈")
    
    if skipped_count > 0:
        log.info(f"夜盘时间跳过 {skipped_count} 个日间品种的止损止盈检查")

def check_position_stop_loss_profit(context, position):
    """检查单个持仓的止损止盈"""
    security = position.security
    
    if security not in g.trade_history:
        return False
    
    trade_info = g.trade_history[security]
    direction = trade_info['direction']
    entry_price = trade_info['entry_price']
    entry_time = trade_info['entry_time']
    current_price = position.price
    
    # 计算当前盈亏比率
    if direction == 'long':
        profit_rate = (current_price - entry_price) / entry_price
    else:
        profit_rate = (entry_price - current_price) / entry_price
    
    # 检查固定止损
    if profit_rate <= -g.fixed_stop_loss_rate:
        log.info(f"触发固定止损 {security} {direction}, 当前亏损率: {profit_rate:.3%}, "
                f"成本价: {entry_price:.2f}, 当前价格: {current_price:.2f}")
        close_position(context, security, direction)
        return True
    
    # 检查是否启用均线跟踪止盈
    if not trade_info.get('ma_trailing_enabled', True):
        return False

    # 检查均线跟踪止盈
    # 获取持仓天数
    entry_date = entry_time.date()
    current_date = context.current_dt.date()
    all_trade_days = get_all_trade_days()
    holding_days = sum((entry_date <= d <= current_date) for d in all_trade_days)
    
    # 计算变化率
    today_price = get_current_data()[security].last_price
    avg_daily_change_rate = calculate_average_daily_change_rate(security)
    historical_data = attribute_history(security, 1, '1d', ['close'])
    yesterday_close = historical_data['close'].iloc[-1]
    today_change_rate = abs((today_price - yesterday_close) / yesterday_close)
    
    # 根据时间判断使用的偏移量
    current_time = context.current_dt.time()
    target_time = datetime.strptime('14:55:00', '%H:%M:%S').time()
    if current_time > target_time:
        offset_ratio = g.ma_offset_ratio_close
    else:
        offset_ratio = g.ma_offset_ratio_normal
    
    # 选择止损均线
    close_line = None
    if today_change_rate >= 1.5 * avg_daily_change_rate:
        close_line = 'ma5'  # 波动剧烈时用短周期
    elif holding_days <= g.days_for_adjustment:
        close_line = 'ma5'  # 持仓初期用短周期
    else:
        close_line = 'ma5' if today_change_rate >= 1.2 * avg_daily_change_rate else 'ma10'
    
    # 计算实时均线值
    ma_values = calculate_realtime_ma_values(security, [5, 10])
    ma_value = ma_values[close_line]
    
    # 应用偏移量
    if direction == 'long':
        adjusted_ma_value = ma_value * (1 - offset_ratio)
    else:
        adjusted_ma_value = ma_value * (1 + offset_ratio)
    
    # 判断是否触发均线止损
    if (direction == 'long' and today_price < adjusted_ma_value) or \
       (direction == 'short' and today_price > adjusted_ma_value):
        log.info(f"触发均线跟踪止盈 {security} {direction}, 止损均线: {close_line}, "
                f"均线值: {ma_value:.2f}, 调整后: {adjusted_ma_value:.2f}, "
                f"当前价: {today_price:.2f}, 持仓天数: {holding_days}")
        close_position(context, security, direction)
        return True
    
    return False

############################ 核心辅助函数 ###################################

def calculate_ma_values(data, periods):
    """计算均线值
    
    Args:
        data: DataFrame，包含'close'列的历史数据（最后一行是最新的数据）
        periods: list，均线周期列表，如[5, 10, 20, 30]
    
    Returns:
        dict: {'MA5': value, 'MA10': value, 'MA20': value, 'MA30': value}
        返回最后一行（最新日期）的各周期均线值
    """
    ma_values = {}
    
    for period in periods:
        if len(data) >= period:
            # 计算最后period天的均线值
            ma_values[f'MA{period}'] = data['close'].iloc[-period:].mean()
        else:
            ma_values[f'MA{period}'] = None
    
    return ma_values


def calculate_ma_proximity_counts(data, periods, lookback_days):
    """统计近 lookback_days 天收盘价贴近各均线的次数"""
    proximity_counts = {f'MA{period}': 0 for period in periods}

    if len(data) < lookback_days:
        return proximity_counts

    closes = data['close'].iloc[-lookback_days:]
    ma_series = {
        period: data['close'].rolling(window=period).mean().iloc[-lookback_days:]
        for period in periods
    }

    for idx, close_price in enumerate(closes):
        min_diff = None
        closest_period = None

        for period in periods:
            ma_value = ma_series[period].iloc[idx]
            if pd.isna(ma_value):
                continue
            diff = abs(close_price - ma_value)
            if min_diff is None or diff < min_diff:
                min_diff = diff
                closest_period = period

        if closest_period is not None:
            proximity_counts[f'MA{closest_period}'] += 1

    return proximity_counts


def check_ma_pattern(ma_values, direction):
    """检查均线排列模式是否符合方向要求
    
    Args:
        ma_values: dict，包含MA5, MA10, MA20, MA30的均线值
        direction: str，'long'或'short'
    
    Returns:
        bool: 是否符合均线排列要求
    """
    ma5 = ma_values['MA5']
    ma10 = ma_values['MA10']
    ma20 = ma_values['MA20']
    ma30 = ma_values['MA30']
    
    if direction == 'long':
        # 多头模式：MA30 <= MA20 <= MA10 <= MA5 或 MA30 <= MA20 <= MA5 <= MA10
        pattern1 = (ma30 <= ma20 <= ma10 <= ma5)
        pattern2 = (ma30 <= ma20 <= ma5 <= ma10)
        return pattern1 or pattern2
    elif direction == 'short':
        # 空头模式：MA10 <= MA5 <= MA20 <= MA30 或 MA5 <= MA10 <= MA20 <= MA30
        pattern1 = (ma10 <= ma5 <= ma20 <= ma30)
        pattern2 = (ma5 <= ma10 <= ma20 <= ma30)
        return pattern1 or pattern2
    else:
        return False

def check_historical_ma_pattern_consistency(historical_data, direction, lookback_days, consistency_threshold):
    """检查历史均线模式的一致性
    
    Args:
        historical_data: DataFrame，包含足够天数的历史数据
        direction: str，'long'或'short'
        lookback_days: int，检查过去多少天
        consistency_threshold: float，一致性阈值（0-1之间）
    
    Returns:
        tuple: (bool, float) - (是否通过一致性检查, 实际一致性比例)
    """
    if len(historical_data) < max(g.ma_periods) + lookback_days:
        # 历史数据不足
        return False, 0.0
    
    match_count = 0
    total_count = lookback_days
    
    # 检查过去lookback_days天的均线模式
    for i in range(lookback_days):
        # 获取倒数第(i+1)天的数据（i=0时是昨天，i=1时是前天，依此类推）
        end_idx = -(i + 1)
        if end_idx == -1:
            data_slice = historical_data
        else:
            data_slice = historical_data.iloc[:end_idx]
        
        # 计算该天的均线值
        ma_values = calculate_ma_values(data_slice, g.ma_periods)
        
        # 检查是否符合模式
        if check_ma_pattern(ma_values, direction):
            match_count += 1
    
    consistency_ratio = match_count / total_count
    passed = consistency_ratio >= consistency_threshold
    
    return passed, consistency_ratio

############################ 交易执行函数 ###################################

def open_position(context, security, target_hands, direction, reason=''):
    """开仓"""
    try:
        # 记录交易前的可用资金
        cash_before = context.portfolio.available_cash
        
        # 使用order_target按手数开仓
        order = order_target(security, target_hands, side=direction)
        
        if order is not None and order.filled > 0:
            # 记录交易后的可用资金
            cash_after = context.portfolio.available_cash
            
            # 计算实际资金变化
            cash_change = cash_before - cash_after
            
            # 获取订单价格和数量
            order_price = order.avg_cost if order.avg_cost else order.price
            order_amount = order.filled
            
            # 记录当日交易
            underlying_symbol = security.split('.')[0][:-4]
            g.today_trades.append({
                'security': security,
                'underlying_symbol': underlying_symbol,
                'direction': direction,
                'order_amount': order_amount,
                'order_price': order_price,
                'cash_change': cash_change,
                'time': context.current_dt
            })
            
            # 记录交易信息
            g.trade_history[security] = {
                'entry_price': order_price,
                'target_hands': target_hands,
                'actual_hands': order_amount,
                'actual_margin': cash_change,
                'direction': direction,
                'entry_time': context.current_dt
            }

            ma_trailing_enabled = True
            if direction == 'long':
                ma_values_at_entry = calculate_realtime_ma_values(security, [5])
                ma5_value = ma_values_at_entry.get('ma5')
                if ma5_value is not None and order_price < ma5_value:
                    ma_trailing_enabled = False
                    log.info(f"禁用均线跟踪止盈: {security} {direction}, 开仓价 {order_price:.2f} < MA5 {ma5_value:.2f}")

            g.trade_history[security]['ma_trailing_enabled'] = ma_trailing_enabled
            
            log.info(f"开仓成功: {security} {direction} {order_amount}手 @{order_price:.2f}, "
                    f"保证金: {cash_change:.0f}, 原因: {reason}")
            
            return True
            
    except Exception as e:
        log.warning(f"开仓失败 {security}: {str(e)}")
    
    return False

def close_position(context, security, direction):
    """平仓"""
    try:
        # 使用order_target平仓到0手
        order = order_target(security, 0, side=direction)
        
        if order is not None and order.filled > 0:
            underlying_symbol = security.split('.')[0][:-4]
            
            # 记录当日交易（平仓）
            g.today_trades.append({
                'security': security,
                'underlying_symbol': underlying_symbol,
                'direction': direction,
                'order_amount': -order.filled,
                'order_price': order.avg_cost if order.avg_cost else order.price,
                'cash_change': 0,
                'time': context.current_dt
            })
            
            log.info(f"平仓成功: {underlying_symbol} {direction} {order.filled}手")
            
            # 从交易历史中移除
            if security in g.trade_history:
                del g.trade_history[security]
            return True
            
    except Exception as e:
        log.warning(f"平仓失败 {security}: {str(e)}")
    
    return False

############################ 辅助函数 ###################################

def get_futures_config(underlying_symbol, config_key=None, default_value=None):
    """获取期货品种配置信息的辅助函数"""
    if underlying_symbol not in g.futures_config:
        if config_key and default_value is not None:
            return default_value
        return {}
    
    if config_key is None:
        return g.futures_config[underlying_symbol]
    
    return g.futures_config[underlying_symbol].get(config_key, default_value)

def get_margin_rate(underlying_symbol, direction, default_rate=0.10):
    """获取保证金比例的辅助函数"""
    return g.futures_config.get(underlying_symbol, {}).get('margin_rate', {}).get(direction, default_rate)

def get_multiplier(underlying_symbol, default_multiplier=10):
    """获取合约乘数的辅助函数"""
    return g.futures_config.get(underlying_symbol, {}).get('multiplier', default_multiplier)

def calculate_target_hands(context, security, direction):
    """计算目标开仓手数"""
    current_price = get_current_data()[security].last_price
    underlying_symbol = security.split('.')[0][:-4]
    
    # 使用保证金比例
    margin_rate = get_margin_rate(underlying_symbol, direction)
    multiplier = get_multiplier(underlying_symbol)
    
    # 计算单手保证金
    single_hand_margin = current_price * multiplier * margin_rate
    
    # 还要考虑可用资金限制
    available_cash = context.portfolio.available_cash * g.usage_percentage
    
    # 根据单个标的最大持仓保证金限制计算开仓数量
    max_margin = g.max_margin_per_position
    
    if single_hand_margin <= max_margin:
        # 如果单手保证金不超过最大限制，计算最大可开仓手数
        max_hands = int(max_margin / single_hand_margin)
        max_hands_by_cash = int(available_cash / single_hand_margin)
        
        # 取两者较小值
        actual_hands = min(max_hands, max_hands_by_cash)
        
        # 确保至少开1手
        actual_hands = max(1, actual_hands)
        
        log.info(f"单手保证金: {single_hand_margin:.0f}, 目标开仓手数: {actual_hands}")
        
        return actual_hands
    else:
        # 如果单手保证金超过最大限制，默认开仓1手
        actual_hands = 1
        
        log.info(f"单手保证金: {single_hand_margin:.0f} 超过最大限制: {max_margin}, 默认开仓1手")
        
        return actual_hands

def check_symbol_prefix_match(symbol, hold_symbols):
    """检查是否有相似的持仓品种"""
    symbol_prefix = symbol[:-9]
    
    for hold_symbol in hold_symbols:
        hold_symbol_prefix = hold_symbol[:-9] if len(hold_symbol) > 9 else hold_symbol
        
        if symbol_prefix == hold_symbol_prefix:
            return True
    return False

def calculate_average_daily_change_rate(security, days=30):
    """计算日均变化率"""
    historical_data = attribute_history(security, days + 1, '1d', ['close'])
    daily_change_rates = abs(historical_data['close'].pct_change()).iloc[1:]
    return daily_change_rates.mean()

def calculate_realtime_ma_values(security, ma_periods):
    """计算包含当前价格的实时均线值"""
    historical_data = attribute_history(security, max(ma_periods), '1d', ['close'])
    today_price = get_current_data()[security].last_price
    close_prices = historical_data['close'].tolist() + [today_price]
    ma_values = {f'ma{period}': sum(close_prices[-period:]) / period for period in ma_periods}
    return ma_values

def after_market_close(context):
    """收盘后运行函数"""
    log.info(str('函数运行时间(after_market_close):'+str(context.current_dt.time())))
    
    # 清空候选列表（每天重新检查）
    g.daily_ma_candidates = {}
    
    # 清空排除缓存（每天重新检查）
    excluded_count = len(g.excluded_contracts)
    if excluded_count > 0:
        log.info(f"清空排除缓存，共 {excluded_count} 个合约")
        g.excluded_contracts = {}
    
    # 只有当天有交易时才打印统计信息
    if g.today_trades:
        print_daily_trading_summary(context)
        
        # 清空当日交易记录
        g.today_trades = []
    
    log.info('##############################################################')

def print_daily_trading_summary(context):
    """打印当日交易汇总"""
    if not g.today_trades:
        return
    
    log.info("\n=== 当日交易汇总 ===")
    total_margin = 0
    
    for trade in g.today_trades:
        if trade['order_amount'] > 0:  # 开仓
            log.info(f"开仓 {trade['underlying_symbol']} {trade['direction']} {trade['order_amount']}手 "
                  f"价格:{trade['order_price']:.2f} 保证金:{trade['cash_change']:.0f}")
            total_margin += trade['cash_change']
        else:  # 平仓
            log.info(f"平仓 {trade['underlying_symbol']} {trade['direction']} {abs(trade['order_amount'])}手 "
                  f"价格:{trade['order_price']:.2f}")
    
    log.info(f"当日保证金占用: {total_margin:.0f}")
    log.info("==================\n")

########################## 自动移仓换月函数 #################################
def position_auto_switch(context, pindex=0, switch_func=None, callback=None):
    """期货自动移仓换月"""
    import re
    subportfolio = context.subportfolios[pindex]
    symbols = set(subportfolio.long_positions.keys()) | set(subportfolio.short_positions.keys())
    switch_result = []
    for symbol in symbols:
        match = re.match(r"(?P<underlying_symbol>[A-Z]{1,})", symbol)
        if not match:
            raise ValueError("未知期货标的: {}".format(symbol))
        else:
            dominant = get_dominant_future(match.groupdict()["underlying_symbol"])
            cur = get_current_data()
            symbol_last_price = cur[symbol].last_price
            dominant_last_price = cur[dominant].last_price
            
            if dominant > symbol:
                for positions_ in (subportfolio.long_positions, subportfolio.short_positions):
                    if symbol not in positions_.keys():
                        continue
                    else :
                        p = positions_[symbol]

                    if switch_func is not None:
                        switch_func(context, pindex, p, dominant)
                    else:
                        amount = p.total_amount
                        # 跌停不能开空和平多，涨停不能开多和平空
                        if p.side == "long":
                            symbol_low_limit = cur[symbol].low_limit
                            dominant_high_limit = cur[dominant].high_limit
                            if symbol_last_price <= symbol_low_limit:
                                log.warning("标的{}跌停，无法平仓。移仓换月取消。".format(symbol))
                                continue
                            elif dominant_last_price >= dominant_high_limit:
                                log.warning("标的{}涨停，无法开仓。移仓换月取消。".format(dominant))
                                continue
                            else:
                                log.info("进行移仓换月: ({0},long) -> ({1},long)".format(symbol, dominant))
                                order_old = order_target(symbol, 0, side='long')
                                if order_old != None and order_old.filled > 0:
                                    order_new = order_target(dominant, amount, side='long')
                                    if order_new != None and order_new.filled > 0:
                                        switch_result.append({"before": symbol, "after": dominant, "side": "long"})
                                        # 换月成功，更新交易记录
                                        if symbol in g.trade_history:
                                            g.trade_history[dominant] = g.trade_history[symbol]
                                            del g.trade_history[symbol]
                                    else:
                                        log.warning("标的{}交易失败，无法开仓。移仓换月失败。".format(dominant))
                        if p.side == "short":
                            symbol_high_limit = cur[symbol].high_limit
                            dominant_low_limit = cur[dominant].low_limit
                            if symbol_last_price >= symbol_high_limit:
                                log.warning("标的{}涨停，无法平仓。移仓换月取消。".format(symbol))
                                continue
                            elif dominant_last_price <= dominant_low_limit:
                                log.warning("标的{}跌停，无法开仓。移仓换月取消。".format(dominant))
                                continue
                            else:
                                log.info("进行移仓换月: ({0},short) -> ({1},short)".format(symbol, dominant))
                                order_old = order_target(symbol, 0, side='short')
                                if order_old != None and order_old.filled > 0:
                                    order_new = order_target(dominant, amount, side='short')
                                    if order_new != None and order_new.filled > 0:
                                        switch_result.append({"before": symbol, "after": dominant, "side": "short"})
                                        # 换月成功，更新交易记录
                                        if symbol in g.trade_history:
                                            g.trade_history[dominant] = g.trade_history[symbol]
                                            del g.trade_history[symbol]
                                    else:
                                        log.warning("标的{}交易失败，无法开仓。移仓换月失败。".format(dominant))
                        if callback:
                            callback(context, pindex, p, dominant)
    return switch_result