python读取yaml文件的内容_Python3 yaml文件读写操作

简介

yaml作为配置文件是非常友好的一种格式，前面一篇讲了yaml的一些基础语法和读取方法，本篇继续讲yaml文件写入方法

用yaml模块写入字典嵌套字典这种复杂的数据，会出现大括号{ }，不是真正的yaml文件数据，可以用ruamel模块就解决。

安装方法：

pip install yaml

yaml文件写入

1.用原生的yaml模块写入这种字典嵌套字典的复杂数据

importosimportyaml#将字典写入到yaml

desired_caps ={'platformName': 'Android','platformVersion': '7.0','deviceName': 'A5RNW18316011440','appPackage': 'com.tencent.mm','appActivity': '.ui.LauncherUI','automationName': 'Uiautomator2','unicodeKeyboard': [True,"hh"],'resetKeyboard': True,'noReset': True,'chromeOptions': {'androidProcess': 'com.tencent.mm:tools'}

}

curpath= os.path.dirname(os.path.realpath(__file__))

yamlpath= os.path.join(curpath, "caps.yaml")#写入到yaml文件

with open(yamlpath, "w", encoding="utf-8") as f:

yaml.dump(desired_caps, f)

2.运行完之后，在当前脚本文件夹目录会生成一个caps.yaml文件，打开后效果如下

3.运行完之后，发现字典嵌套的字典，出现了大括号：{androidProcess: 'com.tencent.mm:tools'}，这不是真正的yaml数据，不是我们想要的，解决办法看下文

ruamel.yaml写入

1.先pip安装ruamel.yam模块

pip install ruamel.yaml

2.使用方法跟yaml差不多，只是在使用dump方法多个一个参数：Dumper=yaml.RoundTripDumper

importosfrom ruamel importyaml#将字典写入到yaml

desired_caps ={'platformName': 'Android','platformVersion': '7.0','deviceName': 'A5RNW18316011440','appPackage': 'com.tencent.mm','appActivity': '.ui.LauncherUI','automationName': 'Uiautomator2','unicodeKeyboard': True,'resetKeyboard': True,'noReset': True,'chromeOptions': {'androidProcess': 'com.tencent.mm:tools'}

}

curpath= os.path.dirname(os.path.realpath(__file__))

yamlpath= os.path.join(curpath, "caps.yaml")#写入到yaml文件

with open(yamlpath, "w", encoding="utf-8") as f:

yaml.dump(desired_caps, f, Dumper=yaml.RoundTripDumper)

3.运行完之后结果：

ruamel.yaml读yaml

1.使用ruamel.yaml模块也能读yaml文件，使用方法相对于之前的yaml.load方法多加一个参数：Loader=yaml.Loader

curpath = os.path.dirname(os.path.realpath(__file__))

yamlpath= os.path.join(curpath, "caps.yaml")#写入到yaml文件

with open(yamlpath, "w", encoding="utf-8") as f:

yaml.dump(result['policy']['policy'], f, Dumper=yaml.RoundTripDumper)#读取

a = open(yamlpath,'r')print(yaml.load(a.read(),Loader=yaml.Loader))

写入中文到yaml文件中

先看yaml文件写入，代码如下：

import yaml

import os

curpath = os.path.dirname(os.path.realpath(__file__)) # 获取文件当前路径

yamlpath = os.path.join(curpath, "data.yaml") # 获取yaml文件地址

f = open(yamlpath,'r',encoding='utf-8')

d = yaml.load(f,Loader=yaml.FullLoader)

print(d) # 读取yaml文件

data = {'name': '测试'} # 需要传入的数据

with open(yamlpath, 'w', encoding='utf-8') as f:

yaml_obj = yaml.dump(data, f)

运行后，查看yaml文件

中文字符被转码

with open(yamlpath, 'w', encoding='utf-8') as f:

yaml_obj = yaml.dump(data, f, allow_unicode=True)

添加allow_unicode=True，再次运行

中文保存成功。

查看源码

发现dump_all方法在处理数据，查看dump_all源码，可以看到里面可以设置各种条件

with open(yamlpath, 'w', encoding='utf-8') as f:

yaml_obj = yaml.dump(data, f,allow_unicode=True,sort_keys=False)

比如设置sort_keys=False，写入yaml的数据则不会排序后写入，如果要追加写入，读取方式改成‘a’就可以了