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