mirror of
https://gitee.com/xiaohuolufeihua/bizhang_-obav.git
synced 2026-05-22 01:12:31 +00:00
YAML: use Loaders and safe_load (#11910)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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(
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user