思路:去掉全局变量(避免每次使用时需要导包),用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值,