yaml: use safe_load(f) instead of load(f, Loader=yaml.FullLoader)

yaml.FullLoader is not always defined
This commit is contained in:
Beat Küng 2019-07-11 08:18:12 +02:00
parent 27ef1899c7
commit 3743e6d8fb
2 changed files with 2 additions and 2 deletions

View File

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

View 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, Loader=yaml.FullLoader)
return yaml.safe_load(f)
except OSError as e:
if e.errno == errno.ENOENT:
raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), file)