lanyuanxiaoyao
2fd8bc1b4a
添加可选的description字段用于文档目的，不影响渲染输出。 主要更改： - core/presentation.py: 添加metadata.description属性 - core/template.py: 添加template.description属性 - tests: 添加16个新测试用例验证description功能 - docs: 更新README.md和README_DEV.md文档 - specs: 新增page-description规范文件
140 lines
4.8 KiB
Python
140 lines
4.8 KiB
Python
"""
|
||
演示文稿模块
|
||
|
||
管理整个演示文稿的生成流程。
|
||
"""
|
||
|
||
from pathlib import Path
|
||
from loaders.yaml_loader import load_yaml_file, validate_presentation_yaml, YAMLError
|
||
from core.template import Template
|
||
from core.elements import create_element
|
||
|
||
|
||
class Presentation:
|
||
"""演示文稿类，管理整个演示文稿的生成流程"""
|
||
|
||
def __init__(self, pres_file, templates_dir=None):
|
||
"""
|
||
初始化演示文稿
|
||
|
||
Args:
|
||
pres_file: 演示文稿 YAML 文件路径
|
||
templates_dir: 模板目录
|
||
"""
|
||
self.pres_file = Path(pres_file)
|
||
self.templates_dir = templates_dir
|
||
|
||
# 加载演示文稿文件
|
||
self.data = load_yaml_file(pres_file)
|
||
validate_presentation_yaml(self.data, str(pres_file))
|
||
|
||
# 获取演示文稿尺寸
|
||
metadata = self.data.get("metadata", {})
|
||
self.size = metadata.get("size", "16:9")
|
||
self.dpi = metadata.get("dpi", 96)
|
||
self.description = metadata.get("description") # 可选的描述字段
|
||
|
||
# 验证尺寸值
|
||
if not isinstance(self.size, str):
|
||
raise ValueError(
|
||
f"无效的尺寸值: {self.size}，尺寸必须是字符串（如 '16:9' 或 '4:3'）"
|
||
)
|
||
|
||
# 模板缓存
|
||
self.template_cache = {}
|
||
|
||
# 解析并保存内联模板
|
||
self.inline_templates = self.data.get('templates', {})
|
||
|
||
def get_template(self, template_name):
|
||
"""
|
||
获取模板（优先检查内联模板，检测同名冲突）
|
||
|
||
Args:
|
||
template_name: 模板名称
|
||
|
||
Returns:
|
||
Template 对象
|
||
|
||
Raises:
|
||
YAMLError: 内联和外部模板同名
|
||
"""
|
||
# 1. 先检查内联模板
|
||
if template_name in self.inline_templates:
|
||
# 2. 检查外部模板是否也存在同名
|
||
if self._external_template_exists(template_name):
|
||
raise YAMLError(
|
||
f"模板名称冲突: '{template_name}' 同时存在于内联模板和外部模板目录\n"
|
||
f"请使用不同的模板名称以避免冲突"
|
||
)
|
||
inline_data = self.inline_templates[template_name]
|
||
return Template.from_data(inline_data, template_name)
|
||
|
||
# 3. 回退到外部模板
|
||
if template_name not in self.template_cache:
|
||
self.template_cache[template_name] = Template(
|
||
template_name, self.templates_dir
|
||
)
|
||
return self.template_cache[template_name]
|
||
|
||
def _external_template_exists(self, template_name):
|
||
"""检查外部模板文件是否存在"""
|
||
if not self.templates_dir:
|
||
return False
|
||
template_path = Path(self.templates_dir) / f"{template_name}.yaml"
|
||
return template_path.exists()
|
||
def render_slide(self, slide_data):
|
||
"""
|
||
渲染单个幻灯片（支持混合模式）
|
||
|
||
Args:
|
||
slide_data: 幻灯片数据字典
|
||
|
||
Returns:
|
||
dict: 包含 background 和 elements 的字典
|
||
|
||
支持三种模式：
|
||
1. 纯模板模式：只有 template 字段
|
||
2. 纯自定义模式：只有 elements 字段
|
||
3. 混合模式：同时有 template 和 elements 字段
|
||
"""
|
||
has_template = "template" in slide_data
|
||
has_custom_elements = slide_data.get("elements")
|
||
|
||
elements_from_template = []
|
||
vars_values = {}
|
||
|
||
# 步骤1：渲染模板（如果有）
|
||
if has_template:
|
||
template_name = slide_data["template"]
|
||
template = self.get_template(template_name)
|
||
vars_values = slide_data.get("vars", {})
|
||
elements_from_template = template.render(vars_values)
|
||
|
||
# 步骤2：处理自定义元素（如果有）
|
||
elements_from_custom = []
|
||
if has_custom_elements:
|
||
custom_elements = slide_data["elements"]
|
||
if has_template:
|
||
# 混合模式：使用模板变量解析自定义元素
|
||
template = self.get_template(slide_data["template"])
|
||
elements_from_custom = [
|
||
template.resolve_element(elem, vars_values)
|
||
for elem in custom_elements
|
||
]
|
||
else:
|
||
# 纯自定义模式（原有行为）
|
||
elements_from_custom = custom_elements
|
||
|
||
# 步骤3：合并元素（模板元素在前，自定义元素在后）
|
||
final_elements = elements_from_template + elements_from_custom
|
||
|
||
# 步骤4：转换为元素对象
|
||
element_objects = [create_element(elem) for elem in final_elements]
|
||
|
||
return {
|
||
"background": slide_data.get("background"),
|
||
"elements": element_objects,
|
||
"description": slide_data.get("description"), # 保留幻灯片描述
|
||
}
|