Docx Contracts

#!/usr/bin/env python3
"""Extract template variables from a docx template and generate JSON schema."""
import sys
import json
from docxtpl import DocxTemplate

def extract_schema(template_path):
    """Extract variables and return JSON schema."""
    doc = DocxTemplate(template_path)
    variables = doc.get_undeclared_template_variables()
    
    schema = {
        "type": "object",
        "properties": {var: {"type": "string"} for var in variables},
        "required": list(variables)
    }
    
    return {
        "variables": list(variables),
        "schema": schema
    }

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python extract_schema.py <template.docx>")
        sys.exit(1)
    
    result = extract_schema(sys.argv[1])
    print(json.dumps(result, indent=2, ensure_ascii=False))

#!/usr/bin/env python3
"""Fill docx template with provided data."""
import sys
import json
from docxtpl import DocxTemplate

def fill_template(template_path, data_json, output_path):
    """Fill template with data and save result."""
    doc = DocxTemplate(template_path)
    data = json.loads(data_json) if isinstance(data_json, str) else data_json
    doc.render(data)
    doc.save(output_path)
    return output_path

if __name__ == "__main__":
    if len(sys.argv) != 4:
        print("Usage: python fill_template.py <template.docx> <data.json> <output.docx>")
        sys.exit(1)
    
    with open(sys.argv[2], 'r', encoding='utf-8') as f:
        data = json.load(f)
    
    result = fill_template(sys.argv[1], data, sys.argv[3])
    print(f"✓ Created: {result}")