YAML: use Loaders and safe_load (#11910)

PX4 · Jul 10, 2019 · 453ecfe · 453ecfe
1 parent 6da8ce9
commit 453ecfe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Tools/validate_yaml.py b/Tools/validate_yaml.py
@@ -40,7 +40,7 @@
 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

diff --git a/msg/tools/uorb_rtps_classifier.py b/msg/tools/uorb_rtps_classifier.py
@@ -94,7 +94,7 @@ def parse_yaml_msg_id_file(yaml_file):
         """
         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(

diff --git a/msg/tools/uorb_to_ros_rtps_ids.py b/msg/tools/uorb_to_ros_rtps_ids.py
@@ -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)