提交 b9cc273b 编写于 作者: C changsh726 提交者: Liu Jiaming

Tools: replaced yaml.load() with yaml.safe_load()

上级 da62ec0f
......@@ -166,7 +166,7 @@ if __name__ == "__main__":
print("Usage:\npython %s your_car_parse_config_file.yml" % sys.argv[0])
sys.exit(0)
with open(sys.argv[1], 'r') as fp:
conf = yaml.load(fp)
conf = yaml.safe_load(fp)
dbc_file = conf["dbc_file"]
protocol_conf_file = conf["protocol_conf"]
car_type = conf["car_type"]
......
......@@ -66,5 +66,5 @@ if __name__ == "__main__":
print("usage:\npython %s some_config.yml" % sys.argv[0])
sys.exit(0)
with open(sys.argv[1], 'r') as fp:
conf = yaml.load(fp)
conf = yaml.safe_load(fp)
gen(conf)
......@@ -140,7 +140,7 @@ if __name__ == "__main__":
print("usage:\npython %s some_config.yml" % sys.argv[0])
sys.exit(0)
with open(sys.argv[1], 'r') as fp:
conf = yaml.load(fp)
conf = yaml.safe_load(fp)
protocol_conf = conf["protocol_conf"]
work_dir = conf["output_dir"] + "proto/"
......
......@@ -441,7 +441,7 @@ def gen_protocols(protocol_conf_file, protocol_dir):
if not os.path.exists(protocol_dir):
os.makedirs(protocol_dir)
with open(protocol_conf_file, 'r') as fp:
content = yaml.load(fp)
content = yaml.safe_load(fp)
protocols = content["protocols"]
car_type = content["car_type"]
for p_name in protocols:
......@@ -475,7 +475,7 @@ if __name__ == "__main__":
print("Usage:\npython %s some_config.yml" % sys.argv[0])
sys.exit(0)
with open(sys.argv[1], 'r') as fp:
conf = yaml.load(fp)
conf = yaml.safe_load(fp)
protocol_conf = conf["protocol_conf"]
protocol_dir = conf["output_dir"] + "vehicle/" + conf["car_type"].lower(
......
......@@ -220,7 +220,7 @@ def gen_build_file(content, output_dir):
def gen_vehicle_controller_and_manager(config_file, output_dir):
print("Generating controller and manager")
with open(config_file, 'r') as fp:
content = yaml.load(fp)
content = yaml.safe_load(fp)
gen_vehicle_controller_header(content, output_dir)
gen_vehicle_controller_cpp(content, output_dir)
gen_message_manager_header(content, output_dir)
......@@ -236,7 +236,7 @@ if __name__ == "__main__":
sys.exit(0)
with open(sys.argv[1], 'r') as fp:
conf = yaml.load(fp)
conf = yaml.safe_load(fp)
protocol_conf = conf["protocol_conf"]
output_dir = conf["output_dir"] + "vehicle/" + conf["car_type"].lower() + \
......
......@@ -48,7 +48,7 @@ def read_parameters(yaml_file):
function to read YAML parameter file and define output destinations
"""
with open(yaml_file, 'r') as f:
params = yaml.load(f)
params = yaml.safe_load(f)
# record file params
RECORD_FOLDER = params['records']['filepath']
parse_type = params['parse']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册