1. 程式人生 > >python接口自動化測試二十九:yaml配置文件的寫和讀

python接口自動化測試二十九:yaml配置文件的寫和讀

load ndt odi 模塊 自動化測試 path dirname Coding ima

# 先安裝ruamel.yaml模塊

寫入配置文件:

技術分享圖片

技術分享圖片

import os
# 先安裝ruamel.yaml模塊
from ruamel import yaml

# 將字典寫入到yaml
dict = {
‘host1‘: ‘123‘,
‘host2‘: ‘456‘,
‘host3‘: ‘789‘,
‘host4‘: ‘147‘,
‘host5‘: {‘asd‘: ‘123‘}
}

curpath = os.path.dirname(os.path.realpath(__file__))
yamlpath = os.path.join(curpath, "config.yaml")

# 寫入到yaml文件
with open(yamlpath, "w", encoding="utf-8") as f:
yaml.dump(dict, f, Dumper=yaml.RoundTripDumper)


讀取配置文件

技術分享圖片

# 讀取yaml文件
# a = open(yamlpath, "r")
# print(yaml.load(a.read(), Loader=yaml.Loader))

a = yaml.load(open(yamlpath, "r").read(), Loader=yaml.Loader)
print(a)
print(a[‘host1‘])

python接口自動化測試二十九:yaml配置文件的寫和讀