Class DataReportGenerator

cli/static/report_template.py:20–150

数据报告生成器

import re

class DataReportGenerator:
    """数据报告生成器"""
    
    def __init__(self, factors_data: Dict):
        self.factors_data = factors_data
        self.console = Console()
        self.market_type = os.environ.get('CONTEST_TRADE_MARKET', 'CN-Stock')
        
    def get_text(self, cn_text: str, en_text: str) -> str:
        return en_text if self.market_type == 'US-Stock' else cn_text
        
    def generate_markdown_report(self, save_path: Path) -> str:
        """生成数据报告的Markdown格式"""
        
        # 获取触发时间
        trigger_time = self.factors_data.get('trigger_time', 'N/A')
        
        # 统计数据源数量和代理数量
        total_agents = len(self.factors_data.get('agents', {}))
        
        report_content = f"""# ContestTrade {self.get_text('数据分析报告', 'Data Analysis Report')}

## 📊 {self.get_text('数据摘要', 'Data Summary')}

**{self.get_text('分析时间', 'Analysis Time')}**: {trigger_time}  
**{self.get_text('分析状态', 'Analysis Status')}**: ✅ {self.get_text('完成', 'Completed')}  
**{self.get_text('数据代理数量', 'Data Agent Count')}**: {total_agents}  

---

## 🔍 {self.get_text('数据源分析详情', 'Data Source Analysis Details')}

"""
        
        # 遍历每个代理的数据
        for agent_name, agent_data in self.factors_data.get('agents', {}).items():
            report_content += f"### 📈 {agent_name.replace('_', ' ').title()}\n\n"
            
            # 只获取context_string字段
            context_string = agent_data.get('context_string', '')
            
            if context_string:
                # 清洗掉 [Batch X] 标记
                cleaned_context = re.sub(r'\[Batch \d+\]', '', context_string).strip()
                report_content += f"{cleaned_context}\n\n"
            else:
                report_content += f"**{self.get_text('暂无分析内容', 'No analysis content available')}**\n\n"
            
            report_content += "---\n\n"
        
        # 免责声明
        report_content += f"## ⚠️ {self.get_text('免责声明', 'Disclaimer')}\n\n"
        report_content += self.get_text("本报告由ContestTrade数据分析系统生成，数据来源于各个数据代理的分析结果，仅供参考。\n\n", "This report is generated by the ContestTrade data analysis system. The data comes from the analysis results of various data agents and is for reference only.\n\n")
        report_content += f"**{self.get_text('报告生成时间', 'Report Generation Time')}**: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
        report_content += f"**{self.get_text('系统版本', 'System Version')}**: ContestTrade v1.1.0\n"
        
        # 保存到文件
        with open(save_path, 'w', encoding='utf-8') as f: