通过Yaml关联传递多接口变量

在实际python接口自动化项目中，可能存在这样的场景，当文件a的变量需要传到项目b，或者项目c中，除了写入到全局变量中进行调用，是否还有其他的方法去完成，把变量写入到Yaml文件是一个好的方法。

1现在项目的路径下新建一个extract.yaml的文件

 

 

 2在common文件夹下创建一个Yaml_utils.py文件，见名知意这是yaml工具文件，编写读取yaml文件的方法

 

3读取yaml文件的代码

# 文件说明：yaml文件的读取与写入
import yaml
 
 
class YamlUtil:
 
    # 读取extract.yaml
    def read_extract_yaml(self):
        # 打开yaml文件，读取方式为r，编码格式utf-8， 然后重命名为 f
        with open('../extract.yaml', mode='r', encoding='utf-8') as f:
            # 读取f文件流， 读取方式FullLoader，然后赋值给value
            value = yaml.load(stream=f, Loader=yaml.FullLoader)
            return value[key]
 
    # 写入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(os.getcwd() + '/extract.yaml', mode='a', encoding='utf-8') as f:
            # 清空yaml文件
            f.truncate()

 

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

 

 

 

 

 

资源大部分来源于网络