aishare-txt 2025.11.6.13__py3-none-any.whl

AIShareTxt/core/analyzer.py

@@ -0,0 +1,495 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+重构后的股票分析器主模块
+作为协调器，整合数据获取、指标计算和报告生成功能
+"""
+
+import warnings
+from .data_fetcher import StockDataFetcher
+from ..indicators.technical_indicators import TechnicalIndicators
+from .report_generator import ReportGenerator
+from ..utils.utils import Logger, LoggerManager, Utils, DataValidator, ErrorHandler
+from .config import IndicatorConfig as Config
+
+warnings.filterwarnings('ignore')
+
+
+class StockAnalyzer:
+    """重构后的股票分析器主类"""
+    
+    def __init__(self):
+        """初始化分析器组件"""
+        # 初始化日志系统
+        LoggerManager.setup_logging()
+        self.logger = LoggerManager.get_logger('stock_analyzer')
+        
+        self.config = Config()
+        self.data_fetcher = StockDataFetcher()
+        self.indicators_calculator = TechnicalIndicators()
+        self.report_generator = ReportGenerator()
+        self.utils = Utils()
+        
+        # 数据存储
+        self.stock_data = None
+        self.stock_code = None
+        self.stock_info = None
+        self.fund_flow_data = None
+        self.indicators = None
+        
+    def analyze_stock(self, stock_code, enable_performance_monitor=False):
+        """
+        分析指定股票的技术指标
+        
+        Args:
+            stock_code (str): 股票代码
+            enable_performance_monitor (bool): 是否启用性能监控
+            
+        Returns:
+            str: 分析报告
+        """
+        # 性能监控
+        monitor = None
+        if enable_performance_monitor:
+            monitor = PerformanceMonitor()
+            monitor.start()
+        
+        try:
+            self.logger.info(f"开始分析股票：{stock_code}")
+            
+            # 验证股票代码
+            if not self.utils.validate_stock_code(stock_code):
+                error_msg = "错误：股票代码格式不正确，请使用6位数字格式（如：000001）"
+                self.logger.error(error_msg)
+                return error_msg
+            
+            self.stock_code = stock_code
+            
+            # 步骤1：获取股票基本信息
+            self.logger.info("步骤1/4：获取股票基本信息...")
+            self.stock_info = self.data_fetcher.get_stock_basic_info(stock_code)
+            if monitor:
+                monitor.checkpoint("获取基本信息")
+            
+            # 步骤2：获取主力资金流数据
+            self.logger.info("步骤2/4：获取主力资金流数据...")
+            self.fund_flow_data = self.data_fetcher.get_fund_flow_data(stock_code)
+            if monitor:
+                monitor.checkpoint("获取资金流数据")
+            
+            # 步骤3：获取股票价格数据
+            self.logger.info("步骤3/4：获取股票价格数据...")
+            self.stock_data = self.data_fetcher.fetch_stock_data(stock_code)
+            
+            if self.stock_data is None:
+                return f"{self.config.ERROR_MESSAGES['no_data']}: {stock_code}"
+            
+            # 验证数据质量
+            is_valid, error_msg = DataValidator.validate_price_data(self.stock_data)
+            if not is_valid:
+                return f"数据质量验证失败: {error_msg}"
+            
+            # 检查数据长度
+            min_length = self.config.DATA_CONFIG.get('min_data_length', 250)
+            is_sufficient, length_msg = self.utils.check_data_quality(self.stock_data, min_length)
+            if not is_sufficient:
+                self.logger.warning(f"警告: {length_msg}")
+                
+            if monitor:
+                monitor.checkpoint("获取价格数据")
+            
+            # 步骤4：计算技术指标
+            self.logger.info("步骤4/4：计算技术指标...")
+            self.indicators = self.indicators_calculator.calculate_all_indicators(self.stock_data)
+            
+            if self.indicators is None:
+                return f"{self.config.ERROR_MESSAGES['calculation_failed']}: 可能是数据不足"
+            
+            # 验证指标数据
+            is_valid, error_msg = DataValidator.validate_indicators(self.indicators)
+            if not is_valid:
+                return f"指标验证失败: {error_msg}"
+            
+            # 将资金流数据添加到指标中
+            if self.fund_flow_data:
+                self.indicators.update(self.fund_flow_data)
+            
+            if monitor:
+                monitor.checkpoint("计算技术指标")
+            
+            # 生成报告
+            self.logger.info("生成分析报告...")
+            report = self.report_generator.generate_report(
+                stock_code, 
+                self.indicators, 
+                self.stock_info
+            )
+            
+            if monitor:
+                monitor.checkpoint("生成报告")
+                monitor.finish()
+            
+            return report
+            
+        except Exception as e:
+            error_context = f"分析股票 {stock_code}"
+            self.utils.log_error(e, error_context)
+            ErrorHandler.handle_api_error(e, "股票分析")
+            return f"分析过程中出错：{str(e)}"
+    
+    def quick_analyze(self, stock_code):
+        """
+        快速分析（不包含资金流数据，提高速度）
+        
+        Args:
+            stock_code (str): 股票代码
+            
+        Returns:
+            str: 分析报告
+        """
+        try:
+            self.logger.info(f"快速分析股票：{stock_code}")
+            
+            # 验证股票代码
+            if not self.utils.validate_stock_code(stock_code):
+                error_msg = "错误：股票代码格式不正确"
+                self.logger.error(error_msg)
+                return error_msg
+            
+            self.stock_code = stock_code
+            
+            # 获取股票价格数据
+            self.stock_data = self.data_fetcher.fetch_stock_data(stock_code)
+            
+            if self.stock_data is None:
+                return f"{self.config.ERROR_MESSAGES['no_data']}: {stock_code}"
+            
+            # 计算技术指标
+            self.indicators = self.indicators_calculator.calculate_all_indicators(self.stock_data)
+            
+            if self.indicators is None:
+                return f"{self.config.ERROR_MESSAGES['calculation_failed']}"
+            
+            # 生成报告（不包含基本信息和资金流）
+            report = self.report_generator.generate_report(
+                stock_code, 
+                self.indicators, 
+                None  # 不包含基本信息
+            )
+            
+            return report
+            
+        except Exception as e:
+            error_context = f"快速分析股票 {stock_code}"
+            self.utils.log_error(e, error_context)
+            return f"快速分析过程中出错：{str(e)}"
+    
+    def get_current_indicators(self):
+        """
+        获取当前计算的指标数据
+        
+        Returns:
+            dict: 指标数据字典
+        """
+        return self.indicators
+    
+    def get_current_stock_data(self):
+        """
+        获取当前的股票数据
+        
+        Returns:
+            pd.DataFrame: 股票数据
+        """
+        return self.stock_data
+    
+    def get_current_stock_info(self):
+        """
+        获取当前的股票基本信息
+        
+        Returns:
+            dict: 股票基本信息
+        """
+        return self.stock_info
+    
+    def export_indicators_to_dict(self):
+        """
+        导出指标数据为字典格式
+        
+        Returns:
+            dict: 包含所有数据的字典
+        """
+        if not self.indicators:
+            return None
+        
+        # 为了确保JSON序列化安全，对所有数据进行清理
+        export_data = {
+            'stock_code': self.stock_code,
+            'analysis_time': self.utils.get_current_time(),
+            'stock_info': self._serialize_data(self.stock_info),
+            'indicators': self._serialize_data(self.indicators),
+            'fund_flow': self._serialize_data(self.fund_flow_data)
+        }
+        
+        return export_data
+    
+    def _serialize_data(self, data):
+        """
+        将数据中的日期对象转换为字符串，确保JSON序列化安全
+        
+        Args:
+            data: 需要序列化的数据
+            
+        Returns:
+            序列化后的数据
+        """
+        if data is None:
+            return None
+            
+        if isinstance(data, dict):
+            result = {}
+            for key, value in data.items():
+                result[key] = self._serialize_data(value)
+            return result
+        elif isinstance(data, (list, tuple)):
+            return [self._serialize_data(item) for item in data]
+        elif hasattr(data, 'strftime'):  # 日期对象
+            return data.strftime('%Y-%m-%d')
+        elif hasattr(data, 'isoformat'):  # datetime对象
+            return data.isoformat()
+        else:
+            return data
+    
+    def validate_analysis_environment(self):
+        """
+        验证分析环境
+        
+        Returns:
+            tuple: (是否通过验证, 错误消息列表)
+        """
+        errors = []
+        
+        try:
+            # 检查必要的包
+            import akshare as ak
+            import talib
+            import pandas as pd
+            import numpy as np
+        except ImportError as e:
+            errors.append(f"缺少必要的包: {str(e)}")
+        
+        # 检查配置
+        if not hasattr(self.config, 'MA_PERIODS'):
+            errors.append("配置文件不完整")
+        
+        # 检查组件初始化
+        if not self.data_fetcher:
+            errors.append("数据获取器初始化失败")
+        
+        if not self.indicators_calculator:
+            errors.append("指标计算器初始化失败")
+        
+        if not self.report_generator:
+            errors.append("报告生成器初始化失败")
+        
+        return len(errors) == 0, errors
+    
+    def get_supported_indicators(self):
+        """
+        获取支持的指标列表
+        
+        Returns:
+            dict: 按类别组织的指标列表
+        """
+        indicators_info = {
+            '移动平均线': {
+                'MA': f"周期: {self.config.MA_PERIODS['short'] + self.config.MA_PERIODS['medium'] + self.config.MA_PERIODS['long']}",
+                'EMA': f"周期: {self.config.EMA_PERIODS}",
+                'WMA': f"周期: {self.config.WMA_PERIODS}"
+            },
+            '均线衍生指标': {
+                'BIAS': f"乖离率，周期: {self.config.BIAS_PERIODS}",
+                'MACD': f"参数: {self.config.MACD_CONFIG}",
+                'Bollinger Bands': f"参数: {self.config.BOLLINGER_BANDS_CONFIG}"
+            },
+            '量价指标': {
+                'VWAP': f"成交量加权平均价，周期: {self.config.VWAP_PERIOD}",
+                'OBV': "能量潮指标"
+            },
+            '趋势强度': {
+                'ADX/DMI': f"趋势方向指标，周期: {self.config.ADX_PERIOD}"
+            },
+            '动量振荡': {
+                'RSI': f"相对强弱指数，周期: {self.config.RSI_PERIODS}",
+                'Stochastic': f"随机振荡器，参数: {self.config.STOCH_CONFIG}"
+            },
+            '波动率': {
+                'ATR': f"平均真实波幅，周期: {self.config.ATR_PERIOD}"
+            },
+            '成交量': {
+                'Volume Ratio': "量比指标",
+                'Volume Trend': "成交量趋势"
+            },
+            '资金流': {
+                'Fund Flow': "主力资金流向",
+                '5-day Fund Flow': "5日累计资金流"
+            }
+        }
+        
+        return indicators_info
+    
+    def get_analysis_summary(self):
+        """
+        获取分析摘要
+        
+        Returns:
+            dict: 分析摘要信息
+        """
+        if not self.indicators:
+            return None
+        
+        summary = {
+            'stock_code': self.stock_code,
+            'current_price': self.indicators.get('current_price', 0),
+            'analysis_date': self.indicators.get('date', ''),
+            'data_points': len(self.stock_data) if self.stock_data is not None else 0,
+            'indicators_count': len(self.indicators),
+            'has_fund_flow': '主力净流入额' in self.indicators,
+            'has_basic_info': self.stock_info is not None
+        }
+        
+        return summary
+
+
+def quick_test(stock_code):
+    """快速测试单个股票的便捷函数"""
+    analyzer = StockAnalyzer()
+    logger = LoggerManager.get_logger('stock_analyzer')
+    
+    print(f"快速测试股票：{stock_code}")
+    print(Utils.create_separator())
+    logger.info(f"开始快速测试股票：{stock_code}")
+    
+    try:
+        # 验证分析环境
+        is_valid, errors = analyzer.validate_analysis_environment()
+        if not is_valid:
+            print("环境验证失败:")
+            logger.error("环境验证失败")
+            for error in errors:
+                print(f"  - {error}")
+                logger.error(f"环境错误: {error}")
+            return False
+        
+        # 执行分析
+        report = analyzer.analyze_stock(stock_code, enable_performance_monitor=True)
+        print("\n" + report)
+        
+        # 显示分析摘要
+        summary = analyzer.get_analysis_summary()
+        if summary:
+            print(f"\n分析摘要:")
+            print(f"  数据点数: {summary['data_points']}")
+            print(f"  指标数量: {summary['indicators_count']}")
+            print(f"  包含资金流: {'是' if summary['has_fund_flow'] else '否'}")
+            print(f"  包含基本信息: {'是' if summary['has_basic_info'] else '否'}")
+            logger.info(f"快速测试完成 - 数据点数: {summary['data_points']}, 指标数量: {summary['indicators_count']}")
+        
+        logger.info(f"快速测试股票 {stock_code} 成功完成")
+        return True
+    except Exception as e:
+        Utils.log_error(e, f"测试股票 {stock_code}")
+        return False
+
+
+def main():
+    """主函数"""
+    analyzer = StockAnalyzer()
+    utils = Utils()
+    logger = LoggerManager.get_logger('stock_analyzer')
+    
+    print("股票技术指标分析器 (重构版)")
+    print(utils.create_separator())
+    print("提示：可以在命令行直接运行 python stock_analyzer.py 000001 来快速测试")
+    logger.info("股票技术指标分析器启动")
+    
+    # 验证分析环境
+    print("\n验证分析环境...")
+    logger.info("开始验证分析环境")
+    is_valid, errors = analyzer.validate_analysis_environment()
+    if not is_valid:
+        print("环境验证失败:")
+        logger.error("环境验证失败")
+        for error in errors:
+            print(f"  ❌ {error}")
+            logger.error(f"环境错误: {error}")
+        return
+    else:
+        print("✅ 环境验证通过")
+        logger.info("环境验证通过")
+    
+    # 检查命令行参数
+    stock_code = utils.parse_command_line_args()
+    if stock_code:
+        logger.info(f"使用命令行参数启动快速测试: {stock_code}")
+        quick_test(stock_code)
+        return
+    
+    # 交互模式
+    while True:
+        try:
+            print(f"\n{utils.create_separator('-', 40)}")
+            stock_code = utils.get_user_input(
+                "请输入股票代码（如：000001，输入 'quit' 退出）：",
+                ['quit', 'exit', 'q']
+            )
+            
+            if stock_code is None:
+                print("谢谢使用！")
+                logger.info("用户退出程序")
+                break
+            
+            if not stock_code:
+                print("请输入有效的股票代码")
+                continue
+            
+            logger.info(f"用户输入股票代码: {stock_code}")
+            
+            # 询问分析模式
+            print("\n选择分析模式:")
+            print("1. 完整分析（包含基本信息和资金流）")
+            print("2. 快速分析（仅技术指标）")
+            
+            mode_choice = utils.get_user_input("请选择模式 (1/2): ")
+            
+            if mode_choice == '2':
+                print("\n执行快速分析...")
+                logger.info(f"开始快速分析股票: {stock_code}")
+                report = analyzer.quick_analyze(stock_code)
+            else:
+                print("\n执行完整分析...")
+                logger.info(f"开始完整分析股票: {stock_code}")
+                report = analyzer.analyze_stock(stock_code, enable_performance_monitor=True)
+            
+            print("\n" + report)
+            
+            # 显示分析摘要
+            summary = analyzer.get_analysis_summary()
+            if summary:
+                print(f"\n{utils.create_separator('-', 30)}")
+                print("分析摘要:")
+                print(f"  股票代码: {summary['stock_code']}")
+                print(f"  当前价格: {summary['current_price']:.2f}")
+                print(f"  数据点数: {summary['data_points']}")
+                print(f"  指标数量: {summary['indicators_count']}")
+                logger.info(f"分析完成 - {summary['stock_code']}: 数据点{summary['data_points']}, 指标{summary['indicators_count']}")
+            
+        except KeyboardInterrupt:
+            if utils.handle_keyboard_interrupt():
+                logger.info("程序被用户中断")
+                break
+        except Exception as e:
+            utils.log_error(e, "主程序")
+
+
+if __name__ == "__main__":
+    main()