03_API 自动化测试-公共方法提取

读取yaml文件、写入yaml文件的公共方法

# 读取yaml文件的工具包 以及写入yaml
import os
import yaml

from common.getRootDir import GetRoot

class Yaml_util:

# 读取yaml文件
def read_extral_yaml(self, key=None):
root_dir = GetRoot().getRoot()
with open(root_dir + "\extract.yml", mode='r', encoding='utf-8') as f:
# value = yaml.load(stream=f,Loader=yaml.FullLoader)
value=yaml.safe_load(f)
return value[key]
# #调用的用例所在的目录，这个目录下的extract.yml文件，然后# 读Full,返回全部数据

# 写入yaml w是覆盖写 a是追加写
def write_extral_yaml(self, data):
root_dir = GetRoot().getRoot()
with open(root_dir + "\extract.yml", mode='a', encoding='utf-8') as f:
yaml.dump(data=data, stream=f, allow_unicode=True)

#清除yaml 在什么时候用呢，每次session前
def clear_extract_yaml(self):
root_dir = GetRoot().getRoot()
with open(root_dir + "\extract.yml", mode='w', encoding='utf-8') as f:
f.truncate()