python 接口自动化之yaml文件读写数据

yaml作为测试数据写入是很方便的，之前学习最初版的自动化框架时使用excel，感觉不是很灵活，所以百度了一下Yaml文件，先介绍下yaml

一、Yaml语言介绍：

YAML 语言（ /ˈjæməl/ ）的设计目标，就是方便人类读写。它实质上是一种通用的数据串行化格式。

YAML 是一种非常灵活的格式，几乎是 JSON 的超集。除了支持注释、换行符分隔、多行字符串、裸字符串和更灵活的类型系统之外，YAML 也支持引用文件，以避免重复代码。

 

基本语法规则：

• 大小写敏感；
• 使用缩进表示层级关系；
• 缩进时不允许使用Tab键，只允许使用空格；
• 缩进的空格数目不重要，只要相同层级的元素左侧对齐即可；
• # 表示注释，从这个字符一直到行尾，都会被解析器忽略；

yaml数据结构

• 对象：键值对的集合，又称为映射（mapping）、哈希（hashes） 、字典（dictionary）；
• 数组：一组按次序排列的值，又称为序列（sequence） 、列表（list）；
• 纯量（scalars）：单个的、不可再分的值；

以下内容截取自网上文章片段：https://cloud.tencent.com/developer/article/1624577

二、新建yaml文件：

可以在yaml文件中写入数据：

三、安装pyYaml

pip install pyYaml

这里写代码的时候犯了个小失误，可能你们也会遇到，其实就是python3.X只能使用pip install pyyaml 来安装，这里可以链接一篇文章：https://www.cnblogs.com/xiaodai0/p/9266631.html

四、封装读写yaml文件的方法：

# 封装读取yaml文件方法
import yaml


class YamlHandler:
    def __init__(self,filename):
        self.filename= filename

    def read(self):
        '''读取yaml文件数据'''
        with open(self.filename,encoding='utf-8') as f:
            return yaml.load(f.read(),Loader = yaml.FullLoader)

    def write_yaml(self,data,encoding='utf-8'):
        '''向yaml文件写入数据'''
        with open(self.filename,encoding=encoding,mode='w') as f:
            return yaml.dump(data,stream=f,allow_unicode = True)

if __name__ == '__main__':
    data = {
        "user":{
            "username":"op",
            "password":"123456"
        }
    }
    '''读取config.yaml 配置文件'''
    read_data = YamlHandler('../setting/config.yaml').read()

    '''将data数据写入yaml'''
    write_data = YamlHandler('../setting/config.yaml').write_yaml(data)
    print(read_data)

 

五、运行结果：

此时config.yaml文件：