34、Selenium + Python 实现 UI 自动化测试-正片6

某一天，测试地址变了。或者某一天邮件发送和接收人地址变了，怎么办？我们需要去login.py文件中修改url地址；我们需要去sendmail.py文件中修改邮件发送和接收人地址。去不同文件修改不同变量值，对于不熟悉你框架的人来说是困难的，对于我们自己来说也是不方便的。

今天我们引入配置文件，将这些易发生变化的部分，放到配置文件里，做统一管理。

一、我们在框架目录下，新建config文件夹，用来存储项目配置文件。

1、新建文件config.yaml，内容如下：

# url 配置url: http://localhost:81/redmine/login# mail 相关配置host: smtp.163.comsender: [e-mail]: xxxxxxxreceiver: [e-mail]: 测试主题，按需修改contents: 邮件内容，按需修改

上面使用yaml 作为配置文件格式，后面我们再介绍其他格式的使用方法。

注意，冒号后面必须有一个空格，冒号前面可以没有空格；没有双引号或单引号。

#为注释符

2、接下来我们需要写一个函数或类，来读取yaml文件。

import yamlimport osdef read_yaml(info):    base_path = os.path.dirname(os.path.abspath(__file__))    conf_path = base_path + '\..\config\config.yaml'    with open(conf_path,'r',encoding='utf-8') as f:        return yaml.load(f).get(info)if __name__ == '__main__':    url = read_yaml('url')    print(url)    host = read_yaml('host')    print(host)

3、修改test_000_login.py、login.py 和 sendmail.py

以sendmail.py为例：

import smtplibfrom email.mime.text import MIMETextfrom email.mime.multipart import MIMEMultipartfrom first.run import reportdirfrom first.utils import readyamldef sendmail():    host = readyaml.read_yaml('host')    sender = readyaml.read_yaml('sender') #发送方邮件地址    passwd = readyaml.read_yaml('passwd')              #发送方密码    receiver = readyaml.read_yaml('receiver')  #接收报告方邮件地址    msg = MIMEMultipart()    msg['from'] = sender    msg['to'] = receiver    msg['subject'] = readyaml.read_yaml('subject')    msg.attach(MIMEText(readyaml.read_yaml('contents')))    att1 = MIMEText(open(reportdir, 'rb').read(), 'base64', 'utf-8')    att1["Content-Type"] = 'application/octet-stream'    # 这里的filename可以任意写，写什么名字，邮件中显示什么名字    att1["Content-Disposition"] = 'attachment; filename="report.html"'    msg.attach(att1)    try:        smtpobj = smtplib.SMTP(host, port=25)        smtpobj.login(sender, passwd)        smtpobj.sendmail(sender, receiver, msg.as_string())        smtpobj.quit()        print('send success')    except:        print('send err')if __name__ == '__main__':    sendmail()

然后试运行 run.py，成功。

这个时候尝试修改下配置文件中的邮件主题和邮件内容，再次运行，看收到的新邮件主题和内容是否变化。

好，简单总结一下：

1、为了使程序更易读，我们将代码和配置相分离，单独保存配置文件，本文引入 yaml 格式作为配置文件

2、你也可以使用ini、json、cvs、 properties等格式。（后面我们也会一一讲解）