Flask工厂函数

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

有什么办法呢？

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

def create_app(config_filename):
    app = Flask(__name__)
    app.config.from_pyfile(config_filename)
    from yourapplication.views.admin import admin
    from yourapplication.views.user import user
    app.register_blueprint(admin)
    app.register_blueprint(user)
    return app

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

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

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

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

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

现在，app.py代码如下：


from flask import Flask, render_template
from configs import config
from book import book_bp
from movie import movie_bp
def create_app(config_name):
    app = Flask(__name__)
    # basic config
    app.config.from_object(config[config_name])
    # blueprints
    app.register_blueprint(book_bp)
    app.register_blueprint(movie_bp)
    # error handler
    handle_errors(app)
    return app
def handle_errors(app):
    @app.errorhandler(404)
    def page_not_found(error):
        return render_template('404.html'), 404

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

from app import create_app
if __name__ == '__main__':
    app = create_app('default')
    app.run(host='127.0.0.1', port=5200, debug=True)

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

	
$ python run.py

猜你喜欢