diff options
| -rwxr-xr-x | config/utils/generate_config.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/config/utils/generate_config.py b/config/utils/generate_config.py index f5721c8d..f02acf5a 100755 --- a/config/utils/generate_config.py +++ b/config/utils/generate_config.py @@ -21,10 +21,14 @@ from jinja2 import Environment, FileSystemLoader PARSER = argparse.ArgumentParser() PARSER.add_argument("--yaml", "-y", type=str, required=True) PARSER.add_argument("--jinja2", "-j", type=str, required=True) +PARSER.add_argument("--includesdir", "-i", type=str, action='append') ARGS = PARSER.parse_args() LOADER = yaml.CSafeLoader if yaml.__with_libyaml__ else yaml.SafeLoader -ENV = Environment(loader=FileSystemLoader(os.path.dirname(ARGS.jinja2))) +ENV = Environment( + loader=FileSystemLoader([os.path.dirname(ARGS.jinja2)] + ARGS.includesdir), + extensions=['jinja2.ext.do'] +) gen_config_lib.load_custom_filters(ENV) # Run `eyaml decrypt` on the whole file, but only if PDF data is encrypted |