在 Shell 中使用 Flask

Changelog

New in version 0.3.

喜欢 Python 的原因之一是交互式的 shell ,它可以让你实时运行 Python 命令,并且立即得到结果。 Flask 本身不带交互 shell ,因为它不需要特定的前期设置,只要在 shell 中导入你的应用就可以开始使用了。

有些辅助工具可以让你在 shell 中更舒服。在交互终端中最大的问题是你不会像浏览器一样触发一个请求,这就意味着无法使用 g 和request 等对象。那么如何在 shell 中测试依赖这些对象的代码呢?

这里有一些有用的辅助函数。请记住,这些辅助函数不仅仅只能用于 shell ,还可以用于单元测试和其他需要假冒请求情境的情况下。

在读下去之前最好你已经读过 请求情境 一节。

命令行接口

自 Flask 0.11 版开始,推荐在 shell 中使用 flask shell 命令,它可以为你做许多自动化工作。比如在 shell 中自动初始化应用情境。

更多信息参见 命令行接口 。

创建一个请求情境

在 shell 中创建一个正确的请求情境的最简便的方法是使用test_request_context 方法。这个方法会创建一个RequestContext :

1. >>> ctx = app.test_request_context()

通常你会使用 with 语句来激活请求对象,但是在 shell 中,可以简便地手动使用push() 和pop() 方法:

1. >>> ctx.push()

从这里开始,直到调用 pop 之前,你可以使用请求对象:

1. >>> ctx.pop()

发送请求前/后动作

仅仅创建一个请求情境还是不够的,需要在请求前运行的代码还是没有运行。比如,在请求前可以会需要转接数据库,或者把用户信息储存在 g 对象中。

使用 preprocess_request() 可以方便地模拟请求前/后动作:

1. >>> ctx = app.test_request_context()
2. >>> ctx.push()
3. >>> app.preprocess_request()

请记住, preprocess_request() 函数可以会返回一个响应对象。如果返回的话请忽略它。

如果要关闭一个请求,那么你需要在请求后函数(由process_response() 触发)作用于响应对象前关闭:

1. >>> app.process_response(app.response_class())
2. <Response 0 bytes [200 OK]>
3. >>> ctx.pop()

teardown_request() 函数会在环境弹出后自动执行。我们可以使用这些函数来销毁请求情境所需要使用的资源(如数据库连接)。

在 Shell 中玩得更爽

如果你喜欢在 shell 中的感觉,那么你可以创建一个导入有关东西的模块,在模块中还可以定义一些辅助方法,如初始化数据库或者删除表等等。假设这个模块名为shelltools ,那么在开始时你可以:

1. >>> from shelltools import *