一、yaml配置文件介绍
配置文件,主要存放一些全局配置变量,如:
①、截图、报告、驱动存放路径;
②、发送邮件使用到的发件人邮箱及登录密码;
③、绕过登录验证码要写入的cookies信息。
二、具体操作步骤
1、先安装yaml模块
pip3 install pyyaml
2、编写一个yaml文件
# 谷歌浏览器驱动路径
driver:
driver: "D:\\Python\\Lib\\site-packages\\selenium\\webdriver\\chrome\\chromedriver.exe"
# 测试地址
url:
url: https://www.baidu.com
# 报告存放路径
path:
path: D:\\report
# 发送邮件
sendmail:
smtpserver: stmp.163.com
sender: XXX
psw: XXX
3、读取配置文件方法和一些全局参数
import yaml
class ReadYaml(object):
def __init__(self):
self.file_name = r'D:\config\config.yaml'
self.encoding = 'utf-8'
def read_conf(self):
#首先将yaml文件打开,以file对象赋值给conf
conf = open(file=self.file_name,mode='r',encoding=self.encoding)
#然后将这个file对象进行读取
str_conf = conf.read()
#这里使用yaml.Load方法将读取的结果传入进去
dict_conf = yaml.load(stream=str_conf,Loader=yaml.FullLoader)
#返回数据
return dict_conf
if __name__ == '__main__':
res = ReadYaml()
driver = res.read_conf()['driver']['driver']
print(driver)