make_templates.py

import collections
import json
import os.path
from typing import Any
from typing import Dict

import mako.lookup


template_lookup = mako.lookup.TemplateLookup(
    directories=['.'],
    default_filters=['html_escape'],
    imports=['from mako.filters import html_escape'],
)


ALL_TEMPLATES = [
    filename for filename in os.listdir('.')
    if filename.endswith('.mako') and filename != 'base.mako'
]


def get_env() -> Dict[str, Any]:
    all_hooks = json.loads(
        open('all-hooks.json').read(),
        object_pairs_hook=collections.OrderedDict,
    )
    all_types = {
        hook_type
        for properties in all_hooks.values()
        for hook_type in (
            properties[0].get("types", []) + properties[0].get("types_or", [])
        )
    }
    return {'all_hooks': all_hooks, 'all_types': all_types}


def main() -> int:
    env = get_env()
    for template in ALL_TEMPLATES:
        template_name, _ = os.path.splitext(template)
        env['template_name'] = template_name
        with open(f'{template_name}.html', 'w') as html_file:
            template_obj = template_lookup.get_template(template)
            html_file.write(template_obj.render(**env))
    return 0


if __name__ == '__main__':
    exit(main())