Flask之多应用 | 小技术分享

这里所说的应用调度，平时也称为：多应用。
官方文档：https://dormousehole.readthedocs.io/en/latest/patterns/appdispatch.html

这是根据文档写的一个简单示例：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from flask import Flask
from werkzeug import run_simple
from werkzeug.middleware.dispatcher import DispatcherMiddleware

# 声明应用
app1 = Flask(__name__)
app2 = Flask(__name__)

# 组合应用
app = DispatcherMiddleware(app1, {
    '/a1': app1,
    '/a2': app2
})


@app1.route('/')
def default():
    return 'hello it is default app1'


@app1.route('/index')
def index():
    return 'hello app1'


@app2.route('/index')
def index():
    return 'hello app2'


if __name__ == "__main__":
    run_simple('localhost', 7777, app)

运行代码，发现就是在访问路径里增加前缀即可访问不同应用的程序。

什么时候用呢？

说到前缀的话，似乎用Blueprint(蓝图)也能实现？确实是，我个人的理解蓝图更多是用在程序的模块化，因为蓝图不是一个真正的应用。但若是你不需要每个应用独立配置的话，那选择蓝图确实是可以的。

看上面的代码，主要的是DispatcherMiddleware，这个实现其实很好理解：

class DispatcherMiddleware:
    def __init__(
        self,
        app: "WSGIApplication",
        mounts: t.Optional[t.Dict[str, "WSGIApplication"]] = None,
    ) -> None:
        self.app = app
        self.mounts = mounts or {}

    # 主要是这个，根据PATH_INFO分派应用程序
    def __call__(
        self, environ: "WSGIEnvironment", start_response: "StartResponse"
    ) -> t.Iterable[bytes]:
        script = environ.get("PATH_INFO", "")
        path_info = ""

        while "/" in script:
            if script in self.mounts:
                app = self.mounts[script]
                break

            script, last_item = script.rsplit("/", 1)
            path_info = f"/{last_item}{path_info}"
        else:
            app = self.mounts.get(script, self.app)

        original_script_name = environ.get("SCRIPT_NAME", "")
        environ["SCRIPT_NAME"] = original_script_name + script
        environ["PATH_INFO"] = path_info
        return app(environ, start_response)

Contents