Browse Source

YAML: use Loaders and safe_load (#11910)

sbg
Alexis Paques 6 years ago committed by Beat Küng
parent
commit
453ecfeb55
  1. 2
      Tools/validate_yaml.py
  2. 2
      msg/tools/uorb_rtps_classifier.py
  3. 2
      msg/tools/uorb_to_ros_rtps_ids.py

2
Tools/validate_yaml.py

@ -40,7 +40,7 @@ verbose = args.verbose @@ -40,7 +40,7 @@ verbose = args.verbose
def load_yaml_file(file_name):
with open(file_name, 'r') as stream:
try:
return yaml.load(stream)
return yaml.safe_load(stream)
except yaml.YAMLError as exc:
print(exc)
raise

2
msg/tools/uorb_rtps_classifier.py

@ -94,7 +94,7 @@ class Classifier(): @@ -94,7 +94,7 @@ class Classifier():
"""
try:
with open(yaml_file, 'r') as f:
return yaml.load(f)
return yaml.load(f, Loader=yaml.FullLoader)
except OSError as e:
if e.errno == errno.ENOENT:
raise IOError(errno.ENOENT, os.strerror(

2
msg/tools/uorb_to_ros_rtps_ids.py

@ -74,7 +74,7 @@ def load_yaml_file(file): @@ -74,7 +74,7 @@ def load_yaml_file(file):
with open(file, 'r') as f:
if verbose:
print("--\t[Step 1] %s yaml file loaded!" % file)
return yaml.load(f)
return yaml.load(f, Loader=yaml.FullLoader)
except OSError as e:
if e.errno == errno.ENOENT:
raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), file)

Loading…
Cancel
Save