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):
|
def load_yaml_file(file_name):
|
||||||
with open(file_name, 'r') as stream:
|
with open(file_name, 'r') as stream:
|
||||||
try:
|
try:
|
||||||
return yaml.load(stream)
|
return yaml.safe_load(stream)
|
||||||
except yaml.YAMLError as exc:
|
except yaml.YAMLError as exc:
|
||||||
print(exc)
|
print(exc)
|
||||||
raise
|
raise
|
||||||
|
|||||||
@@ -94,7 +94,7 @@ class Classifier():
|
|||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
with open(yaml_file, 'r') as f:
|
with open(yaml_file, 'r') as f:
|
||||||
return yaml.load(f)
|
return yaml.load(f, Loader=yaml.FullLoader)
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
if e.errno == errno.ENOENT:
|
if e.errno == errno.ENOENT:
|
||||||
raise IOError(errno.ENOENT, os.strerror(
|
raise IOError(errno.ENOENT, os.strerror(
|
||||||
|
|||||||
@@ -74,7 +74,7 @@ def load_yaml_file(file):
|
|||||||
with open(file, 'r') as f:
|
with open(file, 'r') as f:
|
||||||
if verbose:
|
if verbose:
|
||||||
print("--\t[Step 1] %s yaml file loaded!" % file)
|
print("--\t[Step 1] %s yaml file loaded!" % file)
|
||||||
return yaml.load(f)
|
return yaml.load(f, Loader=yaml.FullLoader)
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
if e.errno == errno.ENOENT:
|
if e.errno == errno.ENOENT:
|
||||||
raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), file)
|
raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), file)
|
||||||
|
|||||||
Reference in New Issue
Block a user