diff --git a/process-templates b/process-templates index 703acc74..8f24d921 100755 --- a/process-templates +++ b/process-templates @@ -1,22 +1,18 @@ #!/usr/bin/env python3 from jinja2 import FileSystemLoader, Environment +from pathlib import Path import os import sys -ROOT_DIR = sys.argv[1] +ROOT_DIR = Path(sys.argv[1]) TEMPLATE_PATH_LIST = [ROOT_DIR, "templates/"] loader = FileSystemLoader(searchpath=TEMPLATE_PATH_LIST) environment = Environment(loader=loader, autoescape=True) -template_file_list = [] -for dirpath, dirnames, filenames in os.walk(ROOT_DIR): - for filename in filenames: - if filename.endswith(".html"): - template_file_list.append( - (os.path.join(dirpath, filename)).split(sep=os.path.sep, maxsplit=1)[1] - ) +file_path_list = ROOT_DIR.glob("**/*.html") +template_file_list = [os.fspath(p.relative_to(ROOT_DIR)) for p in file_path_list] for template_file in template_file_list: template = environment.get_template(template_file)