Flask笔记六之中间件操作

在Flask中，中间件是一种扩展或修改应用程序行为的方法。中间件是在特定的请求处理之前和之后运行的函数。Flask提供了一个注册中间件的机制。

在Flask中，可以使用before_request和after_request装饰器来注册中间件。

解决方案1：使用装饰器注册中间件

from flask import Flask
 
app = Flask(__name__)
 
@app.before_request
def before_request():
    print("这是在请求处理之前运行的中间件")
 
@app.after_request
def after_request(response):
    print("这是在请求处理之后运行的中间件")
    return response

解决方案2：使用自定义中间件函数

from flask import Flask
 
app = Flask(__name__)
 
def my_middleware():
    def middleware(wsgi_app):
        def application(environ, start_response):
            print("这是在请求处理之前运行的中间件")
            response = wsgi_app(environ, start_response)
            print("这是在请求处理之后运行的中间件")
            return response
        return application
    return middleware
 
app.wsgi_app = my_middleware()(app.wsgi_app)

解决方案3：使用Flask的wsgi_app属性直接注册中间件

from flask import Flask
 
app = Flask(__name__)
 
def my_middleware(wsgi_app):
    def application(environ, start_response):
        print("这是在请求处理之前运行的中间件")
        response = wsgi_app(environ, start_response)
        print("这是在请求处理之后运行的中间件")
        return response
    return application
 
app.wsgi_app = my_middleware(app.wsgi_app)

以上三种方法都可以注册中间件，你可以根据具体需求选择合适的方法。

注意：中间件函数必须接受两个参数，并返回一个函数，该函数又接受两个参数并返回响应。在中间件中，你可以在请求处理之前和之后执行任何操作，比如参数验证、权限检查、响应处理等。