with open+safe_load
使用 with open 打开文件,safe_load 读取 yaml 文件数据
示例
import yaml
def read_yaml(file_name, key_word=None):
"""
读取 yaml 文件函数
:param file_name: 文件路径
:param key_word: 用例关键字
:return:
"""
logger.info(f"r开始读取文件:{file_name},key_word:{key_word}")
try:
with open(file=file_name, encoding="utf-8") as f:
# 使用 safe_load 读取
values = yaml.safe_load(f)
logger.info(f"r读取文件:{file_name}成功!!!")
# 有关键字的时候检测输出所有该关键字的用例
if key_word:
new_values = []
for val in values:
# 匹配关键字
if val.get('keywords') == key_word:
new_values.append(val)
return new_values
else:
return values
except Exception as e:
logger.warning(f"r读取文件:{file_name},key_word:{key_word} 失败!!!")
raise RuntimeError(f"用例数据读取失败: {e}") from e
评论 (0)