Python YAML 模块实战：接口测试参数存储与配置

2. 读取 YAML 文件数据

根据键名获取对应的值，适合提取配置项。

import yaml
import os

def read_yaml(filename, key):
    file_path = os.getcwd() + "/data/" + filename
    with open(file_path, encoding="utf-8", mode="r") as f:
        data = yaml.safe_load(f)
        return data[key]

3. 清空 YAML 文件

直接覆盖文件内容即可实现清空。

import os

def clear_yaml(filename):
    file_path = os.getcwd() + "/data/" + filename
    with open(file_path, encoding="utf-8", mode="w") as f:
        f.truncate()

完整示例

将上述功能整合到一个脚本中，模拟实际测试场景：

import yaml
import os

def write_yaml(filename, data):
    file_path = os.getcwd() + "/data/" + filename
    with open(file_path, encoding="utf-8", mode="a+") as f:
        yaml.safe_dump(data, stream=f, allow_unicode=True)

def read_yaml(filename, key):
    file_path = os.getcwd() + "/data/" + filename
    with open(file_path, encoding="utf-8", mode="r") as f:
        data = yaml.safe_load(f)
        return data[key]

def clear_yaml(filename):
    file_path = os.getcwd() + "/data/" + filename
    with open(file_path, encoding="utf-8", mode="w") as f:
        f.truncate()

def test_yml():
    # 1. 写入字典数据
    data = {"str": "12345", "user": "zhangsan", "password": "123456"}
    write_yaml('test.yml', data)

    # 2. 读取特定键的值
    ret = read_yaml('test.yml', "str")
    print("读取结果:", ret)

    # 3. 清空文件
    clear_yaml('test.yml')

if __name__ == "__main__":
    test_yml()

关键注意事项

• 安全加载：务必使用 yaml.safe_load() 和 yaml.safe_dump()，避免使用存在反序列化风险的 yaml.load()；
• 文件路径：示例默认依赖 data 文件夹，运行前请确保该目录已存在，否则可能报错；
• 编码格式：显式指定 encoding="utf-8" 可有效避免中文乱码问题；
• 缩进规范：YAML 对缩进极其敏感，建议使用 2 个或 4 个空格，切勿混用 Tab。