Radicale/documentation-generator/filter.py

#!/usr/bin/env python3

"""
Filter program for transforming the Pandoc AST
"""

import json
import sys

TITLE = "Documentation"


def main():
    data = json.load(sys.stdin)

    level1_headings = [
        (i, content) for i, (level, _, content)
        in ((i, b["c"]) for i, b in enumerate(data["blocks"])
            if b["t"] == "Header")
        if level == 1]
    if (len(level1_headings) != 1
            or level1_headings[0][1] != [{"t": "Str", "c": TITLE}]):
        print(("ERROR: Document must contain single level 1 heading "
               "with content %r") % TITLE, file=sys.stderr)
        exit(1)
    for i, _ in reversed(level1_headings):
        del data["blocks"][i]
    data["meta"]["title"] = {"t": "MetaInlines", "c": level1_headings[0][1]}

    json.dump(data, sys.stdout)


if __name__ == "__main__":
    main()