71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
# MIT License. Part of the Hermes docx skill.
|
|
"""Fill {{placeholder}} tokens in a .docx from a JSON mapping.
|
|
|
|
Tokens are replaced everywhere: body paragraphs, tables (including nested
|
|
tables), headers and footers. Run formatting is preserved; tokens split
|
|
across runs are handled.
|
|
|
|
Usage:
|
|
docx_template.py template.docx values.json output.docx
|
|
docx_template.py template.docx values.json output.docx --strict
|
|
|
|
values.json: {"name": "Ada", "date": "2026-01-01"} fills {{name}}, {{date}}.
|
|
With --strict, exits 1 if any {{token}} remains unfilled after processing.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import re
|
|
import sys
|
|
|
|
from docx import Document
|
|
|
|
from docx_common import iter_all_paragraphs, replace_in_paragraph
|
|
|
|
TOKEN_RE = re.compile(r"\{\{\s*([A-Za-z0-9_.-]+)\s*\}\}")
|
|
|
|
|
|
def main() -> int:
|
|
ap = argparse.ArgumentParser(
|
|
description="Fill {{token}} placeholders in a .docx from JSON.")
|
|
ap.add_argument("template", help="input .docx with {{tokens}}")
|
|
ap.add_argument("values", help="JSON file of token -> value")
|
|
ap.add_argument("output", help="output .docx path")
|
|
ap.add_argument("--strict", action="store_true",
|
|
help="fail if any token remains unfilled")
|
|
args = ap.parse_args()
|
|
|
|
with open(args.values, encoding="utf-8") as f:
|
|
values = json.load(f)
|
|
|
|
doc = Document(args.template)
|
|
filled = {}
|
|
for para in iter_all_paragraphs(doc):
|
|
# Normalize whitespace variants like {{ name }} first.
|
|
for m in set(TOKEN_RE.findall(para.text)):
|
|
if m in values:
|
|
# Replace any spacing variant with canonical token, then fill.
|
|
for variant in set(
|
|
t.group(0) for t in TOKEN_RE.finditer(para.text)
|
|
if t.group(1) == m):
|
|
n = replace_in_paragraph(para, variant, str(values[m]))
|
|
filled[m] = filled.get(m, 0) + n
|
|
|
|
remaining = sorted({m for para in iter_all_paragraphs(doc)
|
|
for m in TOKEN_RE.findall(para.text)})
|
|
doc.save(args.output)
|
|
result = {"ok": True, "output": args.output, "filled": filled,
|
|
"unfilled_tokens": remaining}
|
|
if args.strict and remaining:
|
|
result["ok"] = False
|
|
print(json.dumps(result, ensure_ascii=False))
|
|
return 1
|
|
print(json.dumps(result, ensure_ascii=False))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|