73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from _docx_common import (
|
|
SkillArgumentParser,
|
|
inspect_archive,
|
|
output_file,
|
|
pack_docx_directory,
|
|
publish_file,
|
|
run_cli,
|
|
)
|
|
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
parser = SkillArgumentParser(
|
|
description="把安全解包并编辑后的 OOXML 目录重新打包为 DOCX。"
|
|
)
|
|
parser.add_argument("--input-dir", required=True)
|
|
parser.add_argument("--output", required=True)
|
|
parser.add_argument("--overwrite", action="store_true")
|
|
return parser
|
|
|
|
|
|
def main() -> dict[str, Any]:
|
|
from docx import Document
|
|
|
|
args = build_parser().parse_args()
|
|
source_dir = Path(args.input_dir).expanduser().resolve()
|
|
destination = output_file(
|
|
args.output,
|
|
{".docx"},
|
|
overwrite=args.overwrite,
|
|
)
|
|
if destination == source_dir or source_dir in destination.parents:
|
|
raise ValueError("输出 DOCX 不能放在待打包目录内部")
|
|
descriptor, temp_name = tempfile.mkstemp(
|
|
prefix=f".{destination.stem}.",
|
|
suffix=".docx",
|
|
dir=str(destination.parent),
|
|
)
|
|
os.close(descriptor)
|
|
temp_path = Path(temp_name)
|
|
try:
|
|
archive = pack_docx_directory(source_dir, temp_path)
|
|
if archive["missing_required_parts"]:
|
|
raise ValueError(
|
|
"打包结果缺少必要部件："
|
|
+ "、".join(archive["missing_required_parts"])
|
|
)
|
|
Document(str(temp_path))
|
|
publish_file(temp_path, destination, overwrite=args.overwrite)
|
|
finally:
|
|
if temp_path.exists():
|
|
temp_path.unlink()
|
|
return {
|
|
"path": str(destination),
|
|
"input_dir": str(source_dir),
|
|
"archive": inspect_archive(destination),
|
|
"requires_validation": True,
|
|
"requires_visual_review": True,
|
|
}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(run_cli(main))
|