YAML文件操作

孟船长

发布于 2022-04-12 14:47:36

7210

发布于 2022-04-12 14:47:36

文章被收录于专栏：自动化测试实战自动化测试实战

什么是YAML

yaml是一种灵活的数据格式，支持注释、换行符、多行字符串、裸字符等。在自动化测试过程中，很多地方需要使用配置文件存储数据，比如测试环境、数据库、账号信息、日志报告等。

yaml是配置文件类型之一，相比.ini、conf配置文件来说，yaml更简洁、操作更简单，还可以存放不同类型的数据。后缀yaml或yml。

编写格式

大小写敏感
使用缩进表示层级关系
缩进用空格，相同层级元素左对齐即可
#表示注释

认识yaml格式

列表

- zs
- ls
- wr

数据：

['zs', 'ls', 'wr']

字典

usrname: zs
passwd: 12345

数据：

{'usrname': 'zs', 'passwd': 12345}

字典列表

- usrname: zs
  passwd: 1234
- db: usrs

数据

[{'usrname': 'zs', 'passwd': 1234}, {'db': 'usrs'}]

字典列表2

- usrinfo:
    usrname: zs
    passwd: 1234
- key: wr

数据

[{'usrinfo': {'usrname': 'zs', 'passwd': 1234}}, {'key': 'wr'}]

列表字典

usrinfo:
    usrname: zs
    passwd:
      - 1
      - 2
db: usrs

数据：

{'usrinfo': {'usrname': 'zs', 'passwd': [1, 2]}, 'db': 'usrs'}

yaml文件读取

读

# 安装 PyYAML库
import yaml
import os

with open(os.path.dirname(__file__)+'/config.yaml', mode='r') as f:
    d = yaml.load(stream=f, Loader=yaml.FullLoader)
    print(d)

写入

info = {'access_token': '440', 'unico': '中文字符'}
with open(os.path.dirname(__file__)+'/config.yaml', mode='w+') as f:
    yaml.dump(stream=f, data=info, allow_unicode=True)

删除/清空

with open(os.path.dirname(__file__)+'/config.yaml', mode='w') as f:
    f.truncate()

文件操作模式

模式	解释
r	以只读方式打开文件。文件的指针将会放在文件的开头。这是默认模式。
rb	以二进制格式打开一个文件用于只读。文件指针将会放在文件的开头。这是默认模式。一般用于非文本文件如图片等。
r+	打开一个文件用于读写。文件指针将会放在文件的开头。
rb+	以二进制格式打开一个文件用于读写。文件指针将会放在文件的开头。一般用于非文本文件如图片等。
w	打开一个文件只用于写入。如果该文件已存在则打开文件，并从开头开始编辑，即原有内容会被删除。如果该文件不存在，创建新文件。
wb	以二进制格式打开一个文件只用于写入。如果该文件已存在则打开文件，并从开头开始编辑，即原有内容会被删除。如果该文件不存在，创建新文件。一般用于非文本文件如图片等。
w+	打开一个文件用于读写。如果该文件已存在则打开文件，并从开头开始编辑，即原有内容会被删除。如果该文件不存在，创建新文件。
a	打开一个文件用于追加。如果该文件已存在，文件指针将会放在文件的结尾。也就是说，新的内容将会被写入到已有内容之后。如果该文件不存在，创建新文件进行写入。
a+	打开一个文件用于读写。如果该文件已存在，文件指针将会放在文件的结尾。文件打开时会是追加模式。如果该文件不存在，创建新文件用于读写。