Flask 系列 - 工厂方法

工厂方法

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

有什么办法呢?

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

1
2
3
4
5
6
7
8
9
10
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代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# -*- coding: utf-8 -*-
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,如下:

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

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

1
$ python run.py

更多阅读