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)