Flask覆写wsgi_app函数实现自定义中间件

from flask import Flask
from werkzeug.wrappers import Response as BaseResponse
 
# 自定义中间件类
class CustomMiddleware:
    def __init__(self, app):
        self.app = app
 
    def __call__(self, environ, start_response):
        # 在调用Flask应用之前可以进行一些操作
        # ...
 
        # 调用Flask应用
        response = self.app(environ, start_response)
 
        # 在返回响应之前可以进行一些操作
        # ...
 
        return response
 
# 创建Flask应用
app = Flask(__name__)
 
# 覆写wsgi_app方法来应用自定义中间件
app.wsgi_app = CustomMiddleware(app.wsgi_app)
 
@app.route('/')
def index():
    return 'Hello, World!'
 
if __name__ == '__main__':
    app.run()

这段代码展示了如何在Flask应用中覆写wsgi_app方法来应用自定义的中间件。自定义中间件类CustomMiddleware实现了__init__和__call__方法，以便它可以被当作中间件来使用。在应用中，我们覆写了Flask实例的wsgi_app属性，将其设置为CustomMiddleware的实例，这样所有请求都会先经过自定义中间件的处理。