generate.py: get it working with PyYAML 5.3.1 and Jinja2 2.11.2

This commit is contained in:
Tee-Kiah Chia
2020-12-18 15:40:17 +00:00
parent 5fb307a428
commit 139dfec83b

View File

@@ -17,7 +17,7 @@ os.makedirs(protos_dir, exist_ok=True)
proto_template = env.get_template("proto.j2") proto_template = env.get_template("proto.j2")
import yaml import yaml
protos = yaml.load(open("protos.yaml")) protos = yaml.safe_load(open("protos.yaml"))
for proto in protos: for proto in protos:
with open(os.path.join(protos_dir, proto["name"] + ".h"), 'w') as header: 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))