python读取yaml文件

yaml文件可以存储多种不同类型的数据（字符串、数字、元组、列表、字典等）；

所有的数据都是以冒号和缩进的形式进行存储，严格区分大小写、中英文状态；
yaml文件示例：

# yaml文件中通过#号进行注释
# ①存储字典格式数据
test1:
  login:
    user: buxiubuzhi
    password: 123456

# 或者
test2: {login:{user: buxiubuzhi,password: 123456}}
# 读取到的数据：
# {
#'test1': {'login': {'user': 'buxiubuzhi', 'password': 123456}},
#'test2': {'login': {'user': 'buxiubuzhi', 'password': 123456}}
#}
# 读取到的数据的数据类型： <class 'dict'>

# ②存储列表格式数据

- test: 第一个测试用例
- test: 第二个测试用例
- test: 第三个测试用例

#读取到的数据： [{'test': '第一个测试用例'}, {'test': '第二个测试用例'}, {'test': '第三个测试用例'}]
#读取到的数据的数据类型： <class 'list'>


# 或者
[test: 第一个测试用例,test: 第二个测试用例, test: 第三个测试用例]
# 读取到的数据： [{'test': '第一个测试用例'}, {'test': '第二个测试用例'}, {'test': '第三个测试用例'}]
# 读取到的数据的数据类型： <class 'list'>

# ③ 数据的锚点  通过&创建锚点
userInfo: &userInfo
    username: buxiubuzhi
    password: 123456
    phoneNumber: 19999999999

# 注册时需要用到用户信息  通过*引用锚点  << 代表锚点插入的位置
register:
    <<: *userInfo

#输出结果
# {
# 'userInfo': {'username': 'buxiubuzhi', 'password': 123456, 'phoneNumber': 19999999999},
# 'register': {'username': 'buxiubuzhi', 'password': 123456, 'phoneNumber': 19999999999}
# }

userInfo: &userInfo
    username: &username buxiubuzhi
    password: &password 123456
    phoneNumber: 19999999999

# 登录时只需要引用用户名和密码
login:
    username: *username
    password: *password

# 输出结果
#{
#  'userInfo': {'username': 'buxiubuzhi', 'password': 123456, 'phoneNumber': 19999999999},
#  'login': {'username': 'buxiubuzhi', 'password': 123456}
#}

1.python读取一个yaml文件

# 导包
import yaml

# 定义yaml文件路径
yaml_path = r'D:Python_Script
ew_frameworksource_filelogin_page.yaml'

# 获取文件流对象
with open(yaml_path,'rb')as fileOpen:
    # 通过safe_load 读取文件流
    value = yaml.safe_load(fileOpen)
    print('读取到的数据：',value)
    print('读取到的数据的数据类型：',type(value))

2.将数据写入到yaml文件

# 定义yaml文件路径
yaml_path = r'./source_file/test1.yaml'

# 将复杂类型写入到yaml文件中
data = {
    "terminal" : "Chrome",
    "URL": "www.baidu.com",
    "login":{"username":"buxiubuzhi","password":"123456"},
    "test":["test_01","test_02","test_03"]
}

# 写入文件,如果文件不存在将会自动创建
with open(yaml_path,mode="w",encoding="utf-8") as fp:
    yaml.dump(data,fp)

写入结果：

虽然数据成功写入了，但是有一个问题，写入之后的顺序发生了改变，有时候我们希望数据有一定的关联性数据顺序不能发生改变，那该怎么样呢？

第二种写入数据的方法，yaml模块的升级版：ruamel.yaml

# 安装模块：ruamel.yaml
# pip install ruamel.yaml

# 导包
from ruamel import yaml

# 定义yaml文件路径
yaml_path = r'./source_file/test1.yaml'

# 将复杂类型写入到yaml文件中
data = {
    "terminal" : "Chrome",
    "URL": "www.baidu.com",
    "login":{"username":"buxiubuzhi","password":"123456"},
    "test":["test_01","test_02","test_03"]
}

# 写入文件,如果文件不存在将会自动创建
with open(yaml_path,mode="w",encoding="utf-8") as fp:
    yaml.dump(data,fp,Dumper=yaml.RoundTripDumper) # 只需要添加参数：Dumper=yaml.RoundTripDumper

写入结果：