在Flask中如何通过装饰器添加通用功能到视图函数

在Flask中可以通过装饰器来添加通用功能到视图函数。下面是一个示例代码：

from flask import Flask, request, jsonify

app = Flask(__name__)

# 装饰器函数，用来检查请求头中是否包含token
def check_token(func):
    def wrapper(*args, **kwargs):
        token = request.headers.get('token')
        if not token:
            return jsonify({'error': 'Missing token'}), 401
        # 可以在这里进行token的验证逻辑
        return func(*args, **kwargs)
    return wrapper

# 使用装饰器来添加通用功能到视图函数
@app.route('/')
@check_token
def index():
    return jsonify({'message': 'Hello, World!'})

if __name__ == '__main__':
    app.run()

在上面的代码中，check_token 是一个装饰器函数，用来检查请求头中是否包含名为 token 的字段。在 index 视图函数上使用 @check_token 装饰器，这样当访问 / 路由时，会先执行 check_token 装饰器中的逻辑，然后再执行 index 视图函数中的逻辑。

通过这种方式，可以很方便地将通用功能添加到多个视图函数中，避免重复的代码逻辑。