lanyuanxiaoyao
66472cbd86
重构命令行接口，建立清晰的子命令架构，提升用户体验和代码可维护性。 主要变更： - 移除传统模式，统一使用子命令架构（check/convert/preview） - 将 preview 从 convert 的标志独立为子命令，职责分离 - 重命名参数：--no-check → --skip-validation - 新增 --force/-f：convert 命令支持强制覆盖已存在文件 - 新增 --host：preview 命令支持配置主机地址（局域网预览） - 新增 --no-browser：preview 命令支持不自动打开浏览器 - 优化 --port 默认值：从固定端口改为随机端口（30000-40000） 破坏性变更： - 不再支持传统模式（yaml2pptx.py input.yaml output.pptx） - convert 命令不再支持 --preview 参数，需使用 preview 子命令 文档更新： - 更新 README.md 和 README_DEV.md 的所有使用示例 - 更新命令行选项说明表格 - 新增 CLI 接口规范文档 OpenSpec： - 创建 cli-interface 规范（新能力） - 更新 browser-preview-server 规范（修改的能力） - 归档 refactor-cli-args change（45/45 任务完成）
227 lines
7.0 KiB
Python
227 lines
7.0 KiB
Python
#!/usr/bin/env python3
|
||
# /// script
|
||
# requires-python = ">=3.8"
|
||
# dependencies = [
|
||
# "python-pptx",
|
||
# "pyyaml",
|
||
# "flask",
|
||
# "watchdog",
|
||
# ]
|
||
# ///
|
||
|
||
"""
|
||
YAML to PPTX Converter
|
||
将 YAML 格式的演示文稿源文件转换为 PPTX 文件
|
||
|
||
使用方法:
|
||
# 验证 YAML 文件
|
||
uv run yaml2pptx.py check input.yaml
|
||
|
||
# 转换为 PPTX
|
||
uv run yaml2pptx.py convert input.yaml output.pptx
|
||
uv run yaml2pptx.py convert input.yaml # 自动生成 input.pptx
|
||
|
||
# 启动预览服务器
|
||
uv run yaml2pptx.py preview input.yaml
|
||
"""
|
||
|
||
import sys
|
||
import argparse
|
||
import random
|
||
from pathlib import Path
|
||
|
||
# 导入模块
|
||
from utils import log_info, log_success, log_error, log_progress
|
||
from loaders.yaml_loader import YAMLError
|
||
from core.presentation import Presentation
|
||
from renderers.pptx_renderer import PptxGenerator
|
||
from preview.server import start_preview_server
|
||
from validators import Validator
|
||
|
||
|
||
def parse_args():
|
||
"""解析命令行参数"""
|
||
parser = argparse.ArgumentParser(
|
||
description="将 YAML 格式的演示文稿源文件转换为 PPTX 文件"
|
||
)
|
||
subparsers = parser.add_subparsers(dest='command', required=True)
|
||
|
||
# check 子命令
|
||
check_parser = subparsers.add_parser('check', help='验证 YAML 文件')
|
||
check_parser.add_argument("input", type=str, help="输入的 YAML 文件路径")
|
||
check_parser.add_argument("--template-dir", type=str, default=None, help="模板文件目录路径")
|
||
|
||
# convert 子命令
|
||
convert_parser = subparsers.add_parser('convert', help='转换 YAML 为 PPTX')
|
||
convert_parser.add_argument("input", type=str, help="输入的 YAML 文件路径")
|
||
convert_parser.add_argument("output", type=str, nargs="?", help="输出的 PPTX 文件路径（可选，默认为输入文件名.pptx）")
|
||
convert_parser.add_argument("--template-dir", type=str, default=None, help="模板文件目录路径")
|
||
convert_parser.add_argument("--skip-validation", action="store_true", help="跳过自动验证")
|
||
convert_parser.add_argument("-f", "--force", action="store_true", help="强制覆盖已存在文件")
|
||
|
||
# preview 子命令
|
||
preview_parser = subparsers.add_parser('preview', help='启动预览服务器')
|
||
preview_parser.add_argument("input", type=str, help="输入的 YAML 文件路径")
|
||
preview_parser.add_argument("--template-dir", type=str, default=None, help="模板文件目录路径")
|
||
preview_parser.add_argument("--port", type=int, default=None, help="服务器端口（默认：随机端口 30000-40000）")
|
||
preview_parser.add_argument("--host", type=str, default="127.0.0.1", help="主机地址（默认：127.0.0.1）")
|
||
preview_parser.add_argument("--no-browser", action="store_true", help="不自动打开浏览器")
|
||
|
||
return parser.parse_args()
|
||
|
||
|
||
def handle_check(args):
|
||
"""处理 check 子命令"""
|
||
try:
|
||
input_path = Path(args.input)
|
||
|
||
# 处理模板目录
|
||
template_dir = None
|
||
if args.template_dir:
|
||
template_dir = Path(args.template_dir)
|
||
|
||
# 执行验证
|
||
validator = Validator()
|
||
result = validator.validate(input_path, template_dir)
|
||
|
||
# 输出验证结果
|
||
print(result.format_output())
|
||
|
||
# 返回退出码
|
||
if result.has_errors():
|
||
sys.exit(1)
|
||
else:
|
||
sys.exit(0)
|
||
|
||
except Exception as e:
|
||
log_error(f"验证失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
|
||
def handle_preview(args):
|
||
"""处理 preview 子命令"""
|
||
# 检查输入文件是否存在
|
||
input_path = Path(args.input)
|
||
if not input_path.exists():
|
||
log_error(f"输入文件不存在: {input_path}")
|
||
sys.exit(1)
|
||
|
||
# 处理端口：如果未指定，随机选择 30000-40000 范围内的端口
|
||
port = args.port
|
||
if port is None:
|
||
port = random.randint(30000, 40000)
|
||
|
||
# 处理模板目录
|
||
template_dir = Path(args.template_dir) if args.template_dir else None
|
||
|
||
# 启动预览服务器
|
||
start_preview_server(
|
||
yaml_file=input_path,
|
||
template_dir=template_dir,
|
||
port=port,
|
||
host=args.host,
|
||
open_browser=not args.no_browser
|
||
)
|
||
|
||
|
||
def main():
|
||
"""主函数：加载 YAML → 渲染幻灯片 → 生成 PPTX"""
|
||
try:
|
||
args = parse_args()
|
||
|
||
# 路由到对应的处理函数
|
||
if args.command == 'check':
|
||
handle_check(args)
|
||
elif args.command == 'convert':
|
||
handle_convert(args)
|
||
elif args.command == 'preview':
|
||
handle_preview(args)
|
||
|
||
except YAMLError as e:
|
||
log_error(str(e))
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
log_error(f"未知错误: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
|
||
def handle_convert(args):
|
||
"""处理 convert 子命令"""
|
||
# 检查是否提供了输入文件
|
||
if not args.input:
|
||
log_error("错误: 缺少输入文件参数")
|
||
sys.exit(1)
|
||
|
||
# 处理输入文件路径
|
||
input_path = Path(args.input)
|
||
|
||
# 检查输入文件是否存在
|
||
if not input_path.exists():
|
||
log_error(f"输入文件不存在: {input_path}")
|
||
sys.exit(1)
|
||
|
||
# 处理输出文件路径
|
||
if args.output:
|
||
output_path = Path(args.output)
|
||
else:
|
||
# 自动生成输出文件名
|
||
output_path = input_path.with_suffix(".pptx")
|
||
|
||
# 检查输出文件是否已存在（除非使用 --force）
|
||
if output_path.exists() and not args.force:
|
||
log_error(f"输出文件已存在: {output_path}")
|
||
log_error("使用 --force 或 -f 强制覆盖")
|
||
sys.exit(1)
|
||
|
||
log_info(f"开始转换: {input_path}")
|
||
log_info(f"输出文件: {output_path}")
|
||
|
||
# 自动验证（除非使用 --skip-validation）
|
||
if not args.skip_validation:
|
||
log_info("验证 YAML 文件...")
|
||
template_dir = Path(args.template_dir) if args.template_dir else None
|
||
validator = Validator()
|
||
result = validator.validate(input_path, template_dir)
|
||
|
||
# 如果有错误，输出并终止
|
||
if result.has_errors():
|
||
print("\n" + result.format_output())
|
||
log_error("验证失败，转换已终止")
|
||
sys.exit(1)
|
||
|
||
# 如果有警告，输出但继续
|
||
if result.warnings:
|
||
print("\n" + result.format_output())
|
||
print() # 空行
|
||
|
||
# 1. 加载演示文稿
|
||
log_info("加载演示文稿...")
|
||
pres = Presentation(input_path, templates_dir=args.template_dir)
|
||
|
||
# 2. 创建 PPTX 生成器
|
||
log_info(f"创建演示文稿 ({pres.size})...")
|
||
generator = PptxGenerator(pres.size)
|
||
|
||
# 3. 渲染所有幻灯片
|
||
slides_data = pres.data.get('slides', [])
|
||
total_slides = len(slides_data)
|
||
|
||
for i, slide_data in enumerate(slides_data, 1):
|
||
log_progress(i, total_slides, f"处理幻灯片")
|
||
rendered_slide = pres.render_slide(slide_data)
|
||
generator.add_slide(rendered_slide, input_path.parent)
|
||
|
||
# 4. 保存 PPTX 文件
|
||
log_info("保存 PPTX 文件...")
|
||
generator.save(output_path)
|
||
|
||
log_success(f"转换完成: {output_path}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|