yaml文件关联多接口变量（接口token传参处理）

在实际python接口自动化项目中， 可能有多个py用例文件， 当你需要将用例文件1的变量， 需要传入用例文件3，或者需要传入更多的用例文件中， 使用yaml文件来保存这些变量是非常方便实用的

 

如何做？

1. 首先在项目根路径下新建一个 extract.yaml 文件。（名称可以自定义）

 

.2 然后创建一个common公共目录， 在目录下创建一个Yaml_util.py文件，名称可以自定义。编写（读取）和（写入）yaml的方法

 

3. 读取和写入yaml方法如下

# 文件说明：yaml文件的读取与写入
import yaml


class YamlUtil:

    # 读取extract.yaml
    def read_extract_yaml(self):
        # 打开yaml文件，读取方式为a追加，编码格式utf-8， 然后重命名为 f
        with open('../extract.yaml', mode='a', encoding='utf-8') as f:
            # 读取f文件流， 读取方式FullLoader，然后赋值给value
            value = yaml.load(stream=f, Loader=yaml.FullLoader)
            return value

    # 写入extract.yaml
    def write_extract_yaml(self,data):
        # w 写入yaml文件的意思
        with open('../extract.yaml', mode='a', encoding='utf-8') as f:
            # 写入的数据从data传入
            yaml.dump(data=data, stream=f)

 

如果出现中文乱码报错问题， 在写入函数中加入参数 allow_unicode=True 即可，如下

# 写入extract.yaml
def write_extract_yaml(self, data):
    # w 写入yaml文件的意思
    with open(os.getcwd()+'/extract.yaml', mode='a', encoding='utf-8') as f:
        # 写入的数据从data传入
        yaml.dump(data=data, stream=f, allow_unicode=True)

 

4. 如何实现接口变量的传递呢？请看以下代码和注释

from common.Yaml_util import YamlUtil  # 将yaml工具类导入


class Testlogin(YamlUtil):
    # 用例1
    def test001(self):
        # 假如你有个token值， 需要用用例1提取， 然后传递给用例2
        token = '123nasjnajdaksdnafdknajdabsda'
        # 将token值保存到value变量中，然后写入yaml文件中
        self.write_extract_yaml({'value': token})

    # 用例2
    def test002(self):
        # 读取yaml文件，将token值从yaml文件提取出来
        tokens = self.read_extract_yaml()['value']
        print(tokens)  # 打印即可

 

如果token值需要到不同的py测试文件中引用， 如何做？还是一样，在py文件1中写入， 在py文件2中读取即可。

需要注意几个点：

1. 保存到yaml文件中的数据，建议已字典格式传入，方便后面读取。

2. 保存到yaml文件中的value 键， 不要重名，否则会提取多个值

3. 将数据写入yaml文件时， 要使用追加写入模式，也就是mode='a'， 否则每运行一次（写入方法）会将里面的数据清空。而且需要运行用例之前，写一个清空yaml文件的方法作为前置条件。清空yaml文件方法如下

    # 清空yaml文件内容
    def clear_extract_yaml(self):
        # 打开yaml文件
        with open('/extract.yaml', mode='w', encoding='utf-8') as f:
            # 清空yaml文件
            f.truncate()

 

将这个方法放入到前置后置文件中，每次运行用例， 它就会自动清空yaml文件内容。无需调用，自动执行=True

PS：如果有人不会conftest.py文件的使用方法，请看该文章：https://www.cnblogs.com/FBGG/articles/16222744.html