python接口自动化测试二十九:yaml配置文件的写和读

# 先安装ruamel.yaml模块

写入配置文件:

python接口自动化测试二十九:yaml配置文件的写和读_第1张图片

python接口自动化测试二十九:yaml配置文件的写和读_第2张图片

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)


读取配置文件

python接口自动化测试二十九:yaml配置文件的写和读_第3张图片

 

# 读取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配置文件的写和读_第4张图片

import os
from ruamel import yaml

def getConfig():
# 读取yaml文件中的PC投资端的host
curpath = os.path.dirname(os.path.realpath(__file__))
yamlpath = os.path.join(curpath, "config.yaml")
a = yaml.load(open(yamlpath, "r").read(), Loader=yaml.Loader)
return a

if __name__=="__main__":
print(getConfig()['XXX'])

 python接口自动化测试二十九:yaml配置文件的写和读_第5张图片

from common.get_config import getConfig
host = getConfig()['host_PC_TZ']

print(host)

转载于:https://www.cnblogs.com/malinalian/p/10491867.html

你可能感兴趣的:(python接口自动化测试二十九:yaml配置文件的写和读)