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

3. 清空文件

直接以写模式打开并截断即可。

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()

关键注意事项

• 安全性：务必使用 yaml.safe_load() 和 yaml.safe_dump()，避免使用 yaml.load() 以防代码执行风险。
• 编码：显式指定 encoding="utf-8" 避免中文乱码。
• 路径：示例默认将文件保存在项目根目录的 data 文件夹下，请确保该目录已存在。

完整示例

以下是一个整合了写入、读取和清空的测试脚本：

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():
    data = {"str": "12345", "user": "zhangsan", "password": "123456"}
    write_yaml('test.yml', data)
    ret = read_yaml('test.yml', "str")
    print("读取结果:", ret)
    clear_yaml('test.yml')

if __name__ == "__main__":
    test_yml()