标签搜索

python 封装yaml读取

kirei
2025-05-15 / 0 评论 / 2 阅读 / 正在检测是否收录...

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

评论 (0)

取消