ci: Fix Loader parameter in yaml.loader()

This commit is contained in:
Chen Yu Dong 2021-11-01 11:36:13 +08:00
parent e0e1fab0ab
commit 678275045f
6 changed files with 8 additions and 8 deletions

View File

@ -17,7 +17,7 @@ GITLAB_CONFIG_FILE = os.path.join(os.getenv("IDF_PATH"), ".gitlab-ci.yml")
def check_artifacts_expire_time(): def check_artifacts_expire_time():
with open(GITLAB_CONFIG_FILE, "r") as f: with open(GITLAB_CONFIG_FILE, "r") as f:
config = yaml.load(f) config = yaml.load(f, Loader=yaml.Loader)
errors = [] errors = []

View File

@ -110,7 +110,7 @@ class UnitTestAssignTest(CIAssignTest.AssignTest):
try: try:
with open(test_case_path, "r") as f: with open(test_case_path, "r") as f:
raw_data = yaml.load(f) raw_data = yaml.load(f, Loader=yaml.Loader)
test_cases = raw_data["test cases"] test_cases = raw_data["test cases"]
except IOError: except IOError:
print("Test case path is invalid. Should only happen when use @bot to skip unit test.") print("Test case path is invalid. Should only happen when use @bot to skip unit test.")

View File

@ -52,7 +52,7 @@ class Config(object):
""" """
try: try:
with open(config_file) as f: with open(config_file) as f:
configs = yaml.load(f)[env_name] configs = yaml.load(f, Loader=yaml.Loader)[env_name]
except (OSError, TypeError, IOError): except (OSError, TypeError, IOError):
configs = dict() configs = dict()
return configs return configs

View File

@ -45,7 +45,7 @@ class TestCase(object):
""" """
doc_string = self.test_method.__doc__ doc_string = self.test_method.__doc__
try: try:
doc = yaml.load(doc_string) doc = yaml.load(doc_string, Loader=yaml.Loader)
except (AttributeError, OSError, UnicodeDecodeError): except (AttributeError, OSError, UnicodeDecodeError):
doc = self.DEFAULT_CASE_DOC doc = self.DEFAULT_CASE_DOC
doc.update(self.test_method.env_args) doc.update(self.test_method.env_args)

View File

@ -150,7 +150,7 @@ class AssignTest(object):
def _parse_gitlab_ci_config(self, ci_config_file): def _parse_gitlab_ci_config(self, ci_config_file):
with open(ci_config_file, "r") as f: with open(ci_config_file, "r") as f:
ci_config = yaml.load(f) ci_config = yaml.load(f, Loader=yaml.Loader)
job_list = list() job_list = list()
for job_name in ci_config: for job_name in ci_config:

View File

@ -49,9 +49,9 @@ class Parser(object):
self.unit_jobs = {} self.unit_jobs = {}
self.file_name_cache = {} self.file_name_cache = {}
self.idf_path = idf_path self.idf_path = idf_path
self.tag_def = yaml.load(open(os.path.join(idf_path, self.TAG_DEF_FILE), "r")) self.tag_def = yaml.load(open(os.path.join(idf_path, self.TAG_DEF_FILE), "r"), Loader=yaml.Loader)
self.module_map = yaml.load(open(os.path.join(idf_path, self.MODULE_DEF_FILE), "r")) self.module_map = yaml.load(open(os.path.join(idf_path, self.MODULE_DEF_FILE), "r"), Loader=yaml.Loader)
self.config_dependencies = yaml.load(open(os.path.join(idf_path, self.CONFIG_DEPENDENCY_FILE), "r")) self.config_dependencies = yaml.load(open(os.path.join(idf_path, self.CONFIG_DEPENDENCY_FILE), "r"), Loader=yaml.Loader)
# used to check if duplicated test case names # used to check if duplicated test case names
self.test_case_names = set() self.test_case_names = set()
self.parsing_errors = [] self.parsing_errors = []