Initial commit

table_to_image.py

@@ -0,0 +1,155 @@
+
+import json
+import os
+import plotly.graph_objects as go
+from plotly.io import write_image, write_html
+from config import styles
+
+class PlotlyTableConverter:
+    """使用Plotly将表格数据转换为交互式表格或图片的类"""
+    def __init__(self):
+        """初始化表格样式配置"""
+        self.table_config = styles.get('table')
+
+    def load_table_data(self, file_path):
+        """从JSON文件加载表格数据"""
+        with open(file_path, 'r', encoding='utf-8') as file:
+            data = json.load(file)
+        return data.get('tables', [])
+
+    def create_table(self, table_data, table_index=0):
+        """使用Plotly创建表格"""
+        if not table_data:
+            return None
+
+        # 提取表头和表体
+        header = table_data[0]
+        cells = table_data[1:] if len(table_data) > 1 else []
+
+        # 配置表格样式
+        header_config = self.table_config['header']
+        cells_config = self.table_config['cells']
+        layout_config = self.table_config['layout']
+        border_config = self.table_config['border']
+
+        # 创建Plotly表格（优化边框和对齐方式）
+        table = go.Table(
+            header=dict(
+                values=header,
+                font=dict(
+                    size=header_config['font_size'],
+                    color=header_config['font_color'],
+                    family=header_config['font_family']
+                ),
+                fill_color=header_config['background_color'],
+                height=header_config['height'],
+                line=dict(width=border_config['width'], color=border_config['color']),
+                align=header_config.get('align', 'center')
+            ),
+            cells=dict(
+                values=list(zip(*cells)) if cells else [],  # 转置数据以匹配Plotly格式
+                font=dict(
+                    size=cells_config['font_size'],
+                    color=cells_config['font_color'],
+                    family=cells_config['font_family']
+                ),
+                fill_color=cells_config['background_color'],
+                height=cells_config['height'],
+                line=dict(width=border_config['width'], color=border_config['color']),
+                align=cells_config.get('align', 'left')
+            )
+        )
+
+        # 创建布局（优化标题和边距）
+        layout = go.Layout(
+            width=layout_config['width'],
+            height=layout_config['height'],
+            margin=dict(
+                t=layout_config['margin']['t'],
+                b=layout_config['margin']['b'],
+                l=layout_config['margin']['l'],
+                r=layout_config['margin']['r']
+            ),
+            title=dict(
+                text=f'项目评估表格 {table_index+1}',
+                font=dict(
+                    size=layout_config['title']['font']['size'],
+                    family=layout_config['title']['font'].get('family', 'LXGWWenKai-Medium, SimHei, Arial')
+                )
+            ),
+            showlegend=layout_config.get('showlegend', False)
+        )
+
+        # 创建Figure
+        fig = go.Figure(data=[table], layout=layout)
+        return fig
+
+    def save_as_html(self, fig, output_path):
+        """保存为交互式HTML"""
+        write_html(fig, output_path, auto_open=False)
+        return output_path
+
+    def save_as_image(self, fig, output_path):
+        """保存为图片"""
+        image_config = self.table_config['image']
+        write_image(
+            fig,
+            file=output_path,
+            format=image_config['format'],
+            width=image_config['width'],
+            height=image_config['height'],
+            scale=image_config['scale']
+        )
+        return output_path
+
+    def convert_tables(self, json_path, output_dir='temp'):
+        """将JSON中的所有表格转换为HTML和图片"""
+        os.makedirs(output_dir, exist_ok=True)
+        tables = self.load_table_data(json_path)
+        output_paths = {'html': [], 'image': []}
+
+        for i, table in enumerate(tables):
+            fig = self.create_table(table, i)
+            if fig:
+                # 保存HTML
+                # html_path = os.path.join(output_dir, f"table_{i+1}.html")
+                # html_path = self.save_as_html(fig, html_path)
+                # output_paths['html'].append(html_path)
+
+                # 保存图片
+                image_path = os.path.join(output_dir, f"table_{i+1}.png")
+                image_path = self.save_as_image(fig, image_path)
+                output_paths['image'].append(image_path)
+
+        return output_paths
+
+def generate_table_images():
+    converter = PlotlyTableConverter()
+    # 示例JSON数据格式应与图片中的表格结构一致
+    sample_data = {
+        "tables": [
+            [
+                ["维度", "项目A/Project A", "项目B/Project B", "项目C/Project C", "评估标准/Evaluation"],
+                ["进度 Progress", "按时完成90%的关键里程碑,延期完成10%", "提前两周完成所有阶段", "频繁延期,关键任务延误超过一个月", "按时完成率≥90%为优秀,80%-89%为良好,<80%为需改进"],
+                ["质量 Quality", "客户满意度调查得分92分", "产品缺陷率低至0.5%", "多次因质量问题返工", "客户满意度≥90%,产品缺陷率≤1%为高质量"],
+                ["成本 Cost", "超支5%,在可接受范围内", "预算内完成,成本控制出色,节约3%", "成本超支20%", "成本控制在预算+5%内为优秀,+5%-10%为良好,>10%为需改进"],
+                ["风险 Risk", "无重大损失", "风险识别全面", "风险管理不足", "风险识别率≥90%,有效缓解率≥80%为优秀"],
+                ["沟通 Communication", "定期召开项目会议,沟通记录详尽,反馈及时", "建立了高效的沟通机制使用管理工具促进协作", "沟通不畅,信息延误导致决策失误", "沟通满意度≥90%,信息传递准确率≥95%为高效沟通"]
+            ]
+        ]
+    }
+
+    # 保存示例JSON数据
+    os.makedirs("temp", exist_ok=True)
+    with open("temp/table.json", "w", encoding="utf-8") as f:
+        json.dump(sample_data, f, ensure_ascii=False, indent=4)
+
+    # 执行转换
+    output_paths = converter.convert_tables("temp/table.json")
+
+    print("表格图片已保存至:")
+    for path in output_paths['image']:
+        print(f"- {path}")
+
+if __name__ == "__main__":
+    generate_table_images()