第六章 Email

很多程序在一些特定事件触发时需要通知用户，最常见的方法就是email。虽然Python标准库中的smtplib就可以用来实现这一功能，但Flask-Email扩展更好的包装并集成了smtplib。

Flask-Email实现Email支持

通过pip安装Flask-Email：

(venv) $ pip install flask-mail

该扩展连接到一个smtp(简单邮件传输协议)服务器，并把email传递给它，由其进行分发。如果没有配置，Flask-Email将默认连接到localhost的25端口，并无需认证就发送。（<small>注：在开发服务器模式下，邮件将无法发送，因为localhost只是一个web服务器。</small>）
表6-1展示了配置smtp服务器时可用的配置键名。

Table 6-1. Flask-Mail SMTP server configuration keys
*键名*               *默认值*       *说明*
MAIL_HOSTNAME        localhost     email服务器的主机名或IP地址
MAIL_PORT           25            email服务器的端口号
MAIL_USE_TLS          False      使用安全传输层协议 (TLS)
MAIL_USE_SSL          False         使用安全套接字层(SSL)
MAIL_USERNAME         None          Mail帐户的帐户名
MAIL_PASSWORD         None          Mail帐户的密码

在开发过程中，连接到外部smtp服务器可能会更方便。例子6-1展示了如何配置程序来通过google的Gmail账户来发送邮件。

Example 6-1. hello.py: Flask-Mail configuration for Gmail
import os
# ...
app.config['MAIL_SERVER'] = 'smtp.googlemail.com'
app.config['MAIL_PORT'] = 587
app.config['MAIL_USE_TLS'] = True
app.config['MAIL_USERNAME'] = os.environ.get('MAIL_USERNAME')
app.config['MAIL_PASSWORD'] = os.environ.get('MAIL_PASSWORD')

永远不要把账户信息直接写到你的脚本当中！尤其你计划把自己的项目开源，更是如此。为了保护账户信息，你应该从环境变量中导入敏感信息。

Flask-Mail可以如例子6-2这样初始化：

Example 6-2. hello.py: Flask-Mail initialization
from flask.ext.mail import Mail
mail = Mail(app)

你应该在环境变量中定义email服务器的用户名和密码。Linux或者Mac系统，可以使用bash如下操作：

(venv) $ export MAIL_USERNAME=<Gmail username>
(venv) $ export MAIL_PASSWORD=<Gmail password>

在Windows下，你可以这样做：

(venv) $ set MAIL_USERNAME=<Gmail username>
(venv) $ set MAIL_PASSWORD=<Gmail password>

在Python Shell中发送Email

为了测试配置，你可以启动shell会话并发送测试邮件：

(venv) $ python hello.py shell
>>> from flask.ext.mail import Message
>>> from hello import mail
>>> msg = Message('test subject', sender='you@example.com',
... recipients=['you@example.com'])
>>> msg.body = 'text body'
>>> msg.html = '<b>HTML</b> body'
>>> with app.app_context():
... mail.send(msg)
...

注意在Flask-email的send()函数中使用了current_app，也就是说这一函数必须在活动的应用程序上下文中执行。

在程序中集成Email

为了避免每次都手动创建email消息，最好是把程序的邮件发送功能的通用部分抽出来独立成一个函数。附加好处就是，这个函数能以jinja2模板的方式更灵活地组织邮件内容。例子6-3实现了这一功能：

Example 6-3. hello.py: Email support
from flask.ext.mail import Message
app.config['FLASKY_MAIL_SUBJECT_PREFIX'] = '[Flasky]'
app.config['FLASKY_MAIL_SENDER'] = 'Flasky Admin <flasky@example.com>'
def send_email(to, subject, template, **kwargs):
    msg = Message(app.config['FLASKY_MAIL_SUBJECT_PREFIX'] + subject,sender=app.config['FLASKY_MAIL_SENDER'], recipients=[to])
    msg.body = render_template(template + '.txt', **kwargs)
    msg.html = render_template(template + '.html', **kwargs)
    mail.send(msg)

这个函数需要两个程序指定的配置项，分别定义了邮件主题和发送者。send_email函数携带了目标地址，主题，email正文模板和一个列表参数。模板名字不带扩展名，这两个版本的模板分别用来对应纯文本和富文本的正文。调用者把关键字|键值参数(kwargs)传递给render_template()，以便模板用来生成邮件正文。
　　经过简单扩展之后，index()视图函数一旦接收到一个新名字，就能给管理员发送邮件。例子6-4显示了这一变更。

Example 6-4. hello.py: Email example
# ...
app.config['FLASKY_ADMIN'] = os.environ.get('FLASKY_ADMIN')
# ...
@app.route('/', methods=['GET', 'POST'])
def index():
    form = NameForm()
    if form.validate_on_submit():
        user = User.query.filter_by(username=form.name.data).first()
        if user is None:
            user = User(username=form.name.data)
            db.session.add(user)
            session['known'] = False
            if app.config['FLASKY_ADMIN']:
                send_email(app.config['FLASKY_ADMIN'], 'New User','mail/new_user', user=user)
        else:
            session['known'] = True
        session['name'] = form.name.data
        form.name.data = ''
        return redirect(url_for('index'))
    return render_template('index.html', form=form,name=session.get('name'),known=session.get('known', False))

在程序启动的时候，邮件接收者——FLASK-ADMIN，从环境变量中被导入到配置的同名变量中。我们还需要创建纯文本和html两个邮件模板——它们保存在templates文件夹的mail子文件夹里，便于与其它模板区分开来。邮件模板需要把用户名做为模板参数传递过来，所以调用send_mail()时需要包含着那个关键字|键值参数。
　　除了之前设置的MAIL_USERNAME和MAIL_PASSWORD环境变量之外，此处还需要再设置FLASMKY_ADMIN环境变量，Linux和MacOS用户可使用如下命令行：

(venv) $ export FLASKY_ADMIN=<your-email-address>

windows用户则可如下操作：

(venv) $ set FLASKY_ADMIN=<Gmail username>

设置好后，你可以进行测试，每当你输入一个新用户名，你都会收到一封邮件。

异步发送邮件

发送几次测试邮件之后，你可能会注意到，在发送邮件时，mail.send()函数总是会阻塞几秒，这时候浏览器看上去就没有响应了。为了避免出现处理请求期间的不必要的延迟，应该把邮件发送函数转移到后台线程。请看例子6-5：

Example 6-5. hello.py: Asynchronous email support
from threading import Thread
def send_async_email(app, msg):
    with app.app_context():
        mail.send(msg)
def send_email(to, subject, template, **kwargs):
    msg = Message(app.config['FLASKY_MAIL_SUBJECT_PREFIX'] + subject,sender=app.config['FLASKY_MAIL_SENDER'], recipients=[to])
    msg.body = render_template(template + '.txt', **kwargs)
    msg.html = render_template(template + '.html', **kwargs)
    thr = Thread(target=send_async_email, args=[app, msg])
    thr.start()
    return thr

这里的代码凸现了一个有趣的问题。很多Flask扩展的操作要求有活动的程序和请求上下文。Flask-Mail的send()函数使用current_app，它要求激活程序上下文。但由于mail.send()是在另外一个线程中，这就需要使用app.app_context()来手工指定这一激活的上下文。
　　如果你现在再进行测试，就会发现响应性得以改进了。但要记住一点，对于一个要发送大量邮件的程序来说，把发送邮件单独成一个工作比为每个邮件启动一个新线程更合理。例如：执行send_async_email()函数调用不如使用Celery任务队列来做。
　　到本章为止，我们已经概略完成了大部分web应用具备的功能。现在的问题就是hello.py脚本已经开始变大，难以维护。下一章，我们将学习如何架构大型的程序。
<<第五章 Flask与数据库 第七章 大型程序架构>>