专栏名称: 编程派
Python程序员都在看的公众号,跟着编程派一起学习Python,看最新国外教程和资源!
目录
相关文章推荐
Python开发者  ·  吴恩达出手,开源最新Python包,一个接口 ... ·  1 周前  
Python爱好者社区  ·  字节实习生被索赔800万并不冤。 ·  5 天前  
Python爱好者社区  ·  男生看见雷军路过,直接冲过去递简历,结果... ·  6 天前  
Python爱好者社区  ·  16岁中学生获正高职称,后续来了 ·  1 周前  
Python爱好者社区  ·  开源!Transformers 快速入门书 ·  1 周前  
51好读  ›  专栏  ›  编程派

Flask扩展系列(二):Mail

编程派  · 公众号  · Python  · 2017-07-10 11:30

正文

文 | 思诚之道

来源:http://www.bjhee.com/flask-ext2.html

继续介绍 Flask 常用的扩展,很多站点都需要发送邮件功能,比如用户注册成功邮件,用户重置密码邮件。你可以使用 Python 的 smtplib来发邮件,不过 Flask 有个第三方扩展 Flask-Mail,可以更方便的实现此功能。这里我们就来介绍下这个 Flask-Mail。

系列文章

  • Flask 扩展系列(一)–Restful

安装和启用

建议通过 pip 安装,简单方便:

  1. $ pip install Flask-Mail

我们可以采用下面的方法初始化一个 Mail 的实例:

  1. from flask import Flask

  2. from flask_mail import Mail

  3.  

  4. app = Flask(__name__)

  5. mail = Mail(app)

同其他扩展一样,最后一行实例化代码也可以写成下面的方式:

  1. ...

  2. mail = Mail()

  3. mail.init_app(app)

这样的写法在应用工厂模式下经常会用到。

发送邮件

我们来看一段发送 Hello World 邮件的代码:

  1. from flask import Flask

  2. from flask_mail import Mail, Message

  3.  

  4. app = Flask(__name__)

  5. app.config.update(

  6.     MAIL_SERVER='smtp.example.com',

  7.     MAIL_USERNAME='bjhee',

  8.     MAIL_PASSWORD='example'

  9. )

  10.  

  11. mail = Mail(app)

  12.  

  13. @app.route('/mail')

  14. def send_mail():

  15.     msg = Message('Hello',

  16.                   sender=('Billy.J.Hee', '[email protected]'),

  17.                   recipients=['[email protected]'])

  18.     msg.html = '

    Hello World

    '

  19.     mail.send(msg)

  20.     return 'Successful'

  21.  

  22. if __name__ == '__main__':

  23.     app.run(host='0.0.0.0', debug=True)

访问"http://localhost:5000/mail",邮件就会被发送出去。不用过多解释相信大家都看懂了,因为确实很简单。我们在 Flask 的配置项中配上邮件 SMTP 服务器,用户名和密码;在请求中创建一个 Message 消息实例,传入邮件标题,发件人(元组,包括显示名和地址),收件人(列表)和邮件体;然后调用"Mail.send()"方法发送该消息即可。

Message 对象的属性,可以在初始化时当作参数(key=value) 传入,也可以在初始化后赋值。比如我们可以这样定义收件人:

  1.     msg.recipients=['[email protected]']

由于收件人可以多个,我们也可以通过"add_recipient()"方法添加收件人:

  1.     msg.add_recipient('[email protected]')

发件人可以没有显示名,只要一个地址字符串即可:

  1.     msg.sender='[email protected]'

另外,上例中的邮件体是 HTML 格式,如果是 Plain Text 格式的话,你可以这样写:

  1.     msg.body = 'Hello World'

配置参数

Flask-Mail 扩展可以在 Flask 应用配置项中配置其参数,上例中我们已经看到了 SMTP 服务器,用户名和密码的配置项,这里列举一些常用的:

配置项功能
MAIL_SERVERSMTP邮件服务器地址,默认为localhost
MAIL_PORTSMTP邮件服务器端口,默认为25
MAIL_USERNAME邮件服务器用户名
MAIL_PASSWORD邮件服务器密码
MAIL_DEFAULT_SENDER默认发件人,如果Message对象里没指定发件人,就采用默认发件人
MAIL_USE_TLS是否启用TLS,默认为False
MAIL_USE_SSL是否启用SSL,默认为False
MAIL_MAX_EMAILS邮件批量发送个数上限,默认为没有上限
MAIL_ASCII_ATTACHMENTS将附件的文件名强制转换为ASCII字符,避免在某些情况下出现乱码,默认为False
MAIL_SUPPRESS_SEND调用”Mail.send()”方法后,邮件不会真的被发送,在测试环境中使用,默认为False

批量发送

如果一次需要发送大量邮件,建议采用下面的方式:

  1. @app.route('/batch')

  2. def send_batch():

  3.     with mail.connect() as conn:

  4.         for user in users:

  5.             msg = Message(subject='Hello, %s' % user['name'],

  6.                           body='Welcome, %s' % user['name'],

  7.                           recipients=[user['email']])

  8.             conn.send(msg)

  9.  

  10.     return 'Successful'

这样应用同邮件服务器的连接"mail.connect()"会一直保持到所有邮件发送完毕,也就是退出 with 语句后再关闭,避免多次创建关闭连接的开销。批量发送邮件个数上限由配置项"MAILMAXEMAILS"决定。

邮件带附件

回到第一个发送邮件的例子,让我们在 Hello World 邮件中加上附件:

  1. @app.route('/mail')

  2. def send_mail():

  3.     msg = Message('Hello',

  4.                   sender=('Billy.J.Hee', '[email protected]'),

  5.                   recipients=['[email protected]'])

  6.     msg.html = '

    Hello World

    '

  7.  

  8.     # Add Attachment

  9.     with app.open_resource('blank.docx') as fp:

  10.         msg.attach('blank.docx', 'application/msword', fp.read())

  11.  

  12.     mail.send(msg)

  13.     return 'Successful'

上面的代码中,我们通过"app.open_resource()"方法打开了本地当前目录下的"blank.docx"文件,然后通过"Message.attach()"方法将其附到消息对象中即可。"Message.attach()"方法的第一个参数指定了附件上的文件名,第二个参数指定了文件内容的 MIME 类型,第三个参数就是文件内容。

如果大家不知道要附上的文件 MIME 类型是什么,可以查下 MIME 参考手册。

email_dispatched 信号

Flask-Mail 扩展还提供了一个信号"email_dispatched",当邮件被调度时,该信号就会被发出。如果大家忘了什么是信号,可以参考 进阶系列的第二篇。开发者可以通过这个信号来确定邮件是否被发送成功。当然,发送成功不代表被接收成功。

  1. from flask_mail import email_dispatched

  2.  

  3. def log_mail_sent(message, app):

  4.     print 'Message "%s" is sent successfully' % (message.subject)

  5.  

  6. email_dispatched.connect(log_mail_sent)

另外,在跑测试时我们不希望邮件真的被发出去,此时可以将 Flask 应用的配置项"TESTING"设成 True,或者将 Flask-Mail 扩展的配置项"MAILSUPPRESSSEND"设成 True。这样,调用"Mail.send()"方法后,邮件不会被发出,但是你依然可以接收到"email_dispatched"信号。

本篇中的示例参考了 Flask-Mail 的官方文档和 Flask-Mail 的源码。本篇的示例代码可以在这里下载。


题图:pexels,CC0 授权。

点击阅读原文,查看更多 Python 教程和资源。