Python3 yaml文件讀寫操作

簡介

yaml作為配置文件是非常友好的一種格式，前面一篇講了yaml的一些基礎語法和讀取方法，本篇繼續講yaml文件寫入方法
用yaml模塊寫入字典嵌套字典這種復雜的數據，會出現大括號{ }，不是真正的yaml文件數據，可以用ruamel模塊就解決。

安裝方法：

pip install yaml

yaml文件寫入

1.用原生的yaml模塊寫入這種字典嵌套字典的復雜數據

import os
import yaml


# 將字典寫入到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

import os
from ruamel import yaml


# 將字典寫入到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’就可以了