From 139dfec83b42f0772023bea45bdd8089f93f846b Mon Sep 17 00:00:00 2001 From: Tee-Kiah Chia Date: Fri, 18 Dec 2020 15:40:17 +0000 Subject: [PATCH] generate.py: get it working with PyYAML 5.3.1 and Jinja2 2.11.2 --- generate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generate.py b/generate.py index 259e0c7..3825c8c 100755 --- a/generate.py +++ b/generate.py @@ -17,7 +17,7 @@ os.makedirs(protos_dir, exist_ok=True) proto_template = env.get_template("proto.j2") import yaml -protos = yaml.load(open("protos.yaml")) +protos = yaml.safe_load(open("protos.yaml")) for proto in protos: with open(os.path.join(protos_dir, proto["name"] + ".h"), 'w') as header: - header.write(template.render(proto_template)) + header.write(proto_template.render(proto))