Fix yaml 5.1+ support with unsafe loading

Fixes pytroll#70
djhoese · Apr 24, 2019 · 40fa39e · 40fa39e
1 parent 897c86a
commit 40fa39e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyspectral/config.py b/pyspectral/config.py
@@ -31,6 +31,11 @@
 from collections import Mapping
 import pkg_resources
 
+try:
+    from yaml import UnsafeLoader
+except ImportError:
+    from yaml import Loader as UnsafeLoader
+
 
 LOG = logging.getLogger(__name__)
 
@@ -72,7 +77,7 @@ def get_config():
 
     config = {}
     with open(configfile, 'r') as fp_:
-        config = recursive_dict_update(config, yaml.load(fp_))
+        config = recursive_dict_update(config, yaml.load(fp_, Loader=UnsafeLoader))
 
     app_dirs = AppDirs('pyspectral', 'pytroll')
     user_datadir = app_dirs.user_data_dir