在myhhub/stock项目中实现自定义选股策略的技术方案

项目背景

myhhub/stock是一个开源的股票分析系统，提供了丰富的股票数据分析和策略回测功能。系统内置了多种技术指标和选股策略，同时也支持用户进行自定义扩展。本文将详细介绍如何在项目中实现自定义选股策略并将其结果保存到策略表中。

核心问题分析

在实际使用过程中，用户经常需要基于现有股票数据进行二次筛选和自定义策略开发。系统虽然提供了综合选股功能，但如何将选股结果持久化到策略表中以便后续分析和回测，是一个常见的需求场景。

技术实现方案

1. 策略表结构设计

系统采用模块化设计，策略表存储在TABLE_CN_STOCK_STRATEGIES中。要添加新的策略表，需要遵循以下步骤：

1. 在数据库配置中定义新表结构
2. 确保表结构与现有策略表兼容
3. 添加相应的数据访问层代码

2. 自定义选股逻辑实现

自定义选股逻辑可以参照系统内置的indicators_data_daily_job.py中的guess_sell功能模块。该模块展示了如何：

• 从原始数据中提取所需指标
• 应用自定义算法进行筛选
• 将结果格式化为策略表所需的结构

3. 数据持久化流程

将选股结果保存到策略表的关键步骤包括：

1. 建立数据库连接
2. 准备插入或更新语句
3. 批量处理选股结果
4. 处理可能出现的异常情况

4. 常见问题解决

在实现过程中，可能会遇到"股票代码未在策略表中定义"的错误。这通常是由于：

• 策略表结构定义不完整
• 数据字段映射错误
• 缺少必要的预处理步骤

解决方案是检查表结构定义和数据转换逻辑，确保所有必填字段都正确填充。

实现示例

以下是一个简化的实现示例，展示如何将综合选股结果保存到自定义策略表：

def save_custom_strategy(stock_list, strategy_name):
    """
    将自定义选股结果保存到策略表
    :param stock_list: 选中的股票列表
    :param strategy_name: 策略名称
    """
    # 准备批量插入数据
    values = []
    for stock in stock_list:
        values.append((
            stock['code'],
            strategy_name,
            datetime.now().strftime('%Y-%m-%d'),
            # 其他必要字段...
        ))
    
    # 执行批量插入
    with get_db_connection() as conn:
        cursor = conn.cursor()
        try:
            cursor.executemany(
                "INSERT INTO your_strategy_table VALUES (%s, %s, %s, ...)",
                values
            )
            conn.commit()
        except Exception as e:
            conn.rollback()
            raise e

注意事项

1. 数据一致性：确保策略表中的股票代码与基础数据表一致
2. 性能考虑：对于大量股票数据，建议使用批量操作
3. 错误处理：完善异常处理机制，避免部分失败导致数据不一致
4. 历史数据：考虑策略结果的历史版本管理需求

系统改进建议

根据用户反馈，系统存在一些小的界面文字错误（如"母子县"应为"母子线"）。建议：

1. 建立术语词典，统一专业术语表达
2. 增加界面文字的自动化检查机制
3. 鼓励用户社区参与纠错和改进

总结

在myhhub/stock项目中实现自定义选股策略并持久化结果，需要理解系统的数据架构和策略表设计规范。通过合理利用现有模块（如guess_sell功能）作为参考，开发者可以高效地扩展系统功能，满足个性化的选股需求。同时，注意数据一致性和性能优化，可以构建更加健壮的策略分析系统。