yaml 文件管理

yaml文件管理,包括读取yaml文件内容,写入yaml文件,清空yaml文件。

import pytest
import yaml


# 读取yaml文件
def read_yaml(file):
    with open(file, 'r', encoding='utf-8') as f:
        content = f.read()
    data = yaml.load(content, yaml.FullLoader)
    return data

# 追加写入yaml文件
def write_yaml(file, data):
    with open(file, 'a', encoding='utf-8') as f:
        yaml.dump(data, f)


# 清除yaml文件内容
def clean_yaml(file):
    with open(file, 'r+', encoding='utf-8') as f:
        f.seek(0)  # 把文件定位到第一个位置
        f.truncate()  # 清空文件
    print('执行测试前,先清空yaml文件内容~')


if __name__ == '__main__':
    # read_yaml('key.yml')
    key = {
        "exponent": "123456",
        "modulus": "12345678901234567890-"
    }
    write_yaml('key.yml', key)
    # clean_yaml('key.yml')
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容