python接口自动化测试二十九：yaml配置文件的写和读（ruamel.yaml）

yaml 通常用来存储数据，类似于json

 

安装：pip install ruamel.yaml

 

建一个空的yaml文件

 

写入数据

import os
from ruamel import yaml
# 将字典写入到yaml
data = {
        'host1': '123',
        'host2': 456,
        'host3': {'asd': '123'},
        'host4': [12, '234', ['wer', 234]],
        'host5': {'asd': '123', 'eee': [12, '234', ['wer', 234]]}
        }

curpath = os.path.dirname(os.path.realpath(__file__))  # 当前脚本路径
yamlpath = os.path.join(curpath, "config.yaml")  # 在当前脚本路径中，找到config.yaml文件

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

 

读取yaml文件

import os
from ruamel import yaml

curpath = os.path.dirname(os.path.realpath(__file__))  # 当前脚本路径
yamlpath = os.path.join(curpath, "config.yaml")  # 在当前脚本路径中，找到config.yaml文件
# 读取yaml文件
data = yaml.load(open(yamlpath, "r").read(), Loader=yaml.Loader)
print(data)
print(data['host4'][1])

 

封装起来以后好调用

import os
from ruamel import yaml


class WRYaml:
    """ yaml文件的读和写 """

    def __init__(self):
        """ 指定yaml文件的路径 """
        self.configpath = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'config')

    def read_yaml(self, yaml_file='conf.yaml'):
        """ 读取yaml里面里面的数据"""
        try:
            with open(os.path.join(self.configpath, yaml_file), "r", encoding='utf8') as f:
                return yaml.load(f, Loader=yaml.Loader)
        except Exception as error:
            print(f'读取yaml失败，错误如下：{error}')
            return False

    def write_yaml(self, data, yaml_file='conf.yaml', mode='w'):
        """ 往yaml里面写入数据
            yamlFile：yaml文件名
            data：要写入的数据
            mode：写入方式： w，覆盖写入， a，追加写入
            将原数据读取出来，如果没有要加入的key，则创建一个，如果有，则执行key下面的数据修改
        """
        try:
            old_data = self.read_yaml(yaml_file) or {}
            for data_key, data_value in data.items():
                if not old_data.get(data_key):
                    old_data.setdefault(data_key, {})
                for value_key, value_value in data_value.items():
                    old_data[data_key][value_key] = value_value
            with open(os.path.join(self.configpath, yaml_file), mode, encoding="utf-8") as f:
                yaml.dump(old_data, f, Dumper=yaml.RoundTripDumper)
            return True
        except Exception as error:
            print(f'yaml文件写入失败，错误如下：\n{error}')
            return False


if __name__ == "__main__":
    wryaml = WRYaml()
    # 写入数据文件
    data = {
        'test': {'AAA': 134511, 'BBB': 333}
    }
    print(wryaml.write_yaml(yaml_file='conf.yaml', data=data))
    # 读取数据文件
    print(wryaml.read_yaml('conf.yaml'))