目录

快速入门 - 工厂方法

优质
小牛编辑
138浏览
2023-12-01

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

有什么办法呢?

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

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

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

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

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

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

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

现在,app.py代码如下:

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

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

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

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

  1. $ python run.py

更多阅读