""" PPT 渲染器 - 从 JSON 重建 PPTX 文件 从解析器生成的 JSON 重建完整的 PPT """ from pptx import Presentation from pptx.util import Inches, Pt, Emu from pptx.enum.text import PP_ALIGN from pathlib import Path from typing import Dict, List import json import pandas as pd from ..chart_builder import create_combo_chart from ..chart_builder.layout import ChartLayoutConfig, LegendConfig, ValueAxisConfig from ..chart_builder.styles import StyleConfig from ..chart_builder.date_axis import DateAxisConfig from pptx.enum.chart import XL_LEGEND_POSITION class PPTRenderer: """PPT 渲染器""" def __init__(self, json_data: Dict): """ 初始化渲染器 Args: json_data: 解析器生成的 JSON 数据 """ self.json_data = json_data self.prs = Presentation() # ⭐ 设置幻灯片尺寸为 16:9（从元数据中读取） slide_width = json_data.get("metadata", {}).get("slide_width") slide_height = json_data.get("metadata", {}).get("slide_height") if slide_width and slide_height: from pptx.util import Emu self.prs.slide_width = Emu(slide_width) self.prs.slide_height = Emu(slide_height) print(f"✅ 幻灯片尺寸: {Emu(slide_width).inches:.2f}\" × {Emu(slide_height).inches:.2f}\"") else: # 默认使用 16:9 from pptx.util import Inches self.prs.slide_width = Inches(13.33) # 16:9 标准宽度 self.prs.slide_height = Inches(7.5) print(f"✅ 使用默认 16:9 尺寸") def render(self) -> Presentation: """ 从 JSON 渲染 PPT Returns: Presentation 对象 """ for slide_data in self.json_data["slides"]: self._render_slide(slide_data) return self.prs def _render_slide(self, slide_data: Dict): """渲染单个幻灯片""" # 创建空白幻灯片 slide = self.prs.slides.add_slide(self.prs.slide_layouts[6]) # 空白布局 for element in slide_data["elements"]: self._render_element(slide, element) def _render_element(self, slide, element: Dict): """渲染单个元素""" element_type = element.get("type") position = element.get("position", {}) data = element.get("data", {}) if element_type == "chart": self._render_chart(slide, data, position) elif element_type == "table": self._render_table(slide, data, position) elif element_type == "textbox": self._render_textbox(slide, data, position) elif element_type == "group": self._render_group(slide, data, position) def _render_chart(self, slide, chart_data: Dict, position: Dict): """渲染图表""" try: # 重建 DataFrame df = pd.DataFrame(chart_data["data"]) # 转换日期列 categories_col = chart_data["categories_col"] if categories_col in df.columns: df[categories_col] = pd.to_datetime(df[categories_col]) series_config = chart_data["series_config"] layout_info = chart_data.get("layout_info", {}) # 提取数据范围信息 data_cols = [col for col in df.columns if col != categories_col] # 配置样式 style_config = StyleConfig( color_scheme="aim00", line_width_pt=2.0, marker_style="none", ) # 配置布局 legend_pos_map = { "-4160": XL_LEGEND_POSITION.TOP, "-4107": XL_LEGEND_POSITION.BOTTOM, "-4131": XL_LEGEND_POSITION.RIGHT, "-4129": XL_LEGEND_POSITION.LEFT, } legend_pos_str = str(layout_info.get("legend", {}).get("position", -4160)) legend_position = legend_pos_map.get(legend_pos_str, XL_LEGEND_POSITION.TOP) layout_config = ChartLayoutConfig( title=chart_data.get("title", ""), legend_config=LegendConfig( position=legend_position, font_size_pt=9, font_name="黑体", ), value_axis_config=ValueAxisConfig( number_format="0%", font_size_pt=9, font_name="黑体", has_major_gridlines=False, ), secondary_value_axis_config=ValueAxisConfig( number_format="#,##0", font_size_pt=9, font_name="黑体", has_major_gridlines=False, ), ) # 日期轴配置 label_interval = len(df) // 7 date_axis_config = DateAxisConfig( base_unit='days', major_unit=label_interval, number_format='yyyy/mm', ) layout_config.date_axis_config = date_axis_config # 创建图表 chart = create_combo_chart( slide=slide, df=df, categories_col=categories_col, series_config=series_config, position=(Emu(position["left"]), Emu(position["top"])), size=(Emu(position["width"]), Emu(position["height"])), style_config=style_config, layout_config=layout_config, ) print(f" ✅ 图表已渲染: {chart_data.get('title', '未命名')}") except Exception as e: print(f" ❌ 图表渲染失败: {e}") import traceback traceback.print_exc() def _render_table(self, slide, table_data: Dict, position: Dict): """渲染表格""" try: rows = table_data["rows"] cols = table_data["columns"] data = table_data["data"] # 创建表格 table_shape = slide.shapes.add_table( rows, cols, Emu(position["left"]), Emu(position["top"]), Emu(position["width"]), Emu(position["height"]) ) table = table_shape.table # 填充数据 for i, row_data in enumerate(data): for j, cell_text in enumerate(row_data): cell = table.cell(i, j) cell.text = cell_text # 设置字体 cell.text_frame.paragraphs[0].font.name = "黑体" cell.text_frame.paragraphs[0].font.size = Pt(9) print(f" ✅ 表格已渲染: {rows}行 × {cols}列") except Exception as e: print(f" ❌ 表格渲染失败: {e}") def _render_textbox(self, slide, textbox_data: Dict, position: Dict): """渲染文本框""" try: # 创建文本框 textbox = slide.shapes.add_textbox( Emu(position["left"]), Emu(position["top"]), Emu(position["width"]), Emu(position["height"]) ) text_frame = textbox.text_frame # 添加段落 paragraphs = textbox_data.get("paragraphs", []) for i, para_data in enumerate(paragraphs): if i == 0: # 使用默认的第一个段落 para = text_frame.paragraphs[0] else: para = text_frame.add_paragraph() # 设置段落级别 if para_data.get("level"): para.level = para_data["level"] # 设置对齐方式 alignment_str = para_data.get("alignment") if alignment_str: try: # 尝试解析对齐方式 if "CENTER" in alignment_str: para.alignment = PP_ALIGN.CENTER elif "RIGHT" in alignment_str: para.alignment = PP_ALIGN.RIGHT elif "LEFT" in alignment_str: para.alignment = PP_ALIGN.LEFT except: pass # 添加 runs（保留原始格式） runs = para_data.get("runs", []) if runs: for run_data in runs: run = para.add_run() run.text = run_data["text"] # ⭐ 应用原始字体信息 if run_data.get("font_name"): # 如果原始字体不可用，回退到黑体 try: run.font.name = run_data["font_name"] except: run.font.name = "黑体" else: run.font.name = "黑体" if run_data.get("font_size"): run.font.size = Pt(run_data["font_size"]) if run_data.get("bold"): run.font.bold = True if run_data.get("italic"): run.font.italic = True if run_data.get("underline"): run.font.underline = True # 应用颜色 if run_data.get("color"): try: from pptx.util import RGBColor color_hex = run_data["color"].lstrip('#') r = int(color_hex[0:2], 16) g = int(color_hex[2:4], 16) b = int(color_hex[4:6], 16) run.font.color.rgb = RGBColor(r, g, b) except: pass else: # 如果没有 runs，直接设置段落文本 para.text = para_data["text"] para.font.name = "黑体" para.font.size = Pt(9) preview = textbox_data["full_text"][:30] + "..." if len(textbox_data["full_text"]) > 30 else textbox_data["full_text"] print(f' ✅ 文本框已渲染: "{preview}"') except Exception as e: print(f" ❌ 文本框渲染失败: {e}") import traceback traceback.print_exc() def _render_group(self, slide, group_data: Dict, position: Dict): """渲染组合形状（递归处理内部元素）""" # 注意：python-pptx 不直接支持创建组合形状 # 我们将组合形状的元素展开，直接渲染到幻灯片上 for element in group_data.get("elements", []): self._render_element(slide, element) print(f" ✅ 组合形状已渲染: {group_data.get('element_count', 0)} 个子元素") def save(self, output_path: Path): """ 保存 PPT 文件 Args: output_path: 输出路径 """ self.prs.save(str(output_path)) print(f"\n✅ PPT 已保存: {output_path}") def render_from_json_file(json_path: Path, output_pptx_path: Path): """ 从 JSON 文件渲染 PPT Args: json_path: JSON 文件路径 output_pptx_path: 输出的 PPTX 文件路径 """ with open(json_path, 'r', encoding='utf-8') as f: json_data = json.load(f) renderer = PPTRenderer(json_data) renderer.render() renderer.save(output_pptx_path) if __name__ == "__main__": # 测试 from pathlib import Path json_path = Path(__file__).parent.parent.parent / "output" / "aim01_parsed.json" output_path = Path(__file__).parent.parent.parent / "output" / "aim01_rebuilt.pptx" render_from_json_file(json_path, output_path)