YAML: use Loaders and safe_load (#11910)

This commit is contained in:
Alexis Paques 2019-07-10 15:15:29 +02:00 committed by Beat Küng
parent 6da8ce94f1
commit 453ecfeb55
3 changed files with 3 additions and 3 deletions

View File

@ -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

View File

@ -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(

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)
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)