接口自动化测试框架之——关于接口关联的封装

思路：去掉全局变量（避免每次使用时需要导包），用YAML文件代替保存（对YAML进行读写等操作）。

新建一个公共包common，在其下新建yaml_util.py，写入三个方法：读、写、清空

import os
import yaml


# # 读取yaml，读取yaml时，传入key来读取对应的值
def read_yaml(key):
    with open(os.getcwd() + '/extract.yaml') as f:
        value = yaml.load(stream=f, Loader=yaml.FullLoader, mode='r', encodings='utf-8')
        return value[key]


# 写入yaml
def write_yaml(data):
    with open(os.getcwd() + '/extract.yaml', mode='a', encoding='utf-8') as f:
        value = yaml.dump(data, stream=f, allow_unicode=True)


# 清空yaml
def clean_yaml():
    with open(os.getcwd() + '/extract.yaml', mode='w', encoding='utf-8') as f:
        f.truncate()

假设我们需要获取access_token值，