task.mk/generate.py

#!/usr/bin/env python3

import sys
from pathlib import Path
import jinja2

py_script_names = [
    "help",
    "print",
    "print-ansi",
    "vars",
    "confirm",
    "utils",
]


def get_jinja_env():
    templateLoader = jinja2.FileSystemLoader(searchpath=Path(__file__).parent / "src")
    return jinja2.Environment(
        loader=templateLoader,
        block_start_string="#%",
        block_end_string="%#",
        variable_start_string="##-",
        variable_end_string="-##",
        comment_start_string="###-",
        comment_end_string="-###",
    )


def render(env, template, **kwargs):
    template = env.get_template(template)
    return template.render(**kwargs)


def dropnewlines(text):
    return "\n".join([line for line in text.splitlines() if line])


def main():
    if len(sys.argv) == 2:
        version = sys.argv[1]
    else:
        version = "dev"

    env = get_jinja_env()
    py_scripts = [dropnewlines(render(env, f"{name}.py")) for name in py_script_names]
    print(render(env, "task.mk", py_scripts=py_scripts, version=version))


if __name__ == "__main__":
    main()