工厂方法

在前面，我们都是直接通过app=Flask(__name__)来创建一个app实例的。这样做没什么问题，但如果我们想为每个实例分配不同的配置，比如有测试环境的配置，开发环境的配置和生产环境的配置等，这时就比较麻烦了。

有什么办法呢？

其实我们可以通过调用一个函数来返回一个应用实例，比如下面的方法：

  1. def create_app(config_filename):
  2.     app = Flask(__name__)
  3.     app.config.from_pyfile(config_filename)
  5.     from yourapplication.views.admin import admin
  6.     from yourapplication.views.user import user
  7.     app.register_blueprint(admin)
  8.     app.register_blueprint(user)
  10.     return app

上面的create_app函数就是一个工厂方法，我们将创建应用程序实例的工作交给了它来完成，我们以后就可以通过传入不同的配置名，以此批量生产app。

说到这，你也应该明白工厂方法的优势所在了：

  • 将创建应用实例的过程交给工厂函数，通过传入不同的配置，我们可以创建不同环境下的应用
  • 在做测试时，为每个实例分配分配不同的配置，从而测试每一种不同的情况

现在，我们对上一篇的例子进行重构，引入工厂函数。

核心代码在下面，完成代码请参考这里。

现在，app.py代码如下：

  1. # -*- coding: utf-8 -*-
  3. from flask import Flask, render_template
  4. from configs import config
  5. from book import book_bp
  6. from movie import movie_bp
  8. def create_app(config_name):
  9.     app = Flask(__name__)
  11.     # basic config
  12.     app.config.from_object(config[config_name])
  14.     # blueprints
  15.     app.register_blueprint(book_bp)
  16.     app.register_blueprint(movie_bp)
  18.     # error handler
  19.     handle_errors(app)
  21.     return app
  23. def handle_errors(app):
  24.     @app.errorhandler(404)
  25.     def page_not_found(error):
  26.         return render_template('404.html'), 404

我们还需要一个启动程序，比如run.py，如下：

  1. # -*- coding: utf-8 -*-
  3. from app import create_app
  5. if __name__ == '__main__':
  6.     app = create_app('default')
  7.     app.run(host='127.0.0.1', port=5200, debug=True)

打开终端，运行下面命令，就可以在浏览器访问我们的应用了。

  1. $ python run.py

更多阅读