python接口自动化测试二十九:yaml配置文件的写和读
# 先安装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'])
封装起来方便以后调用
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'])
from common.get_config import getConfig
host = getConfig()['host_PC_TZ']
print(host)