python JWTManager token_in_blocklist_loader 的使用方法 写个例子给我

当使用 JWTManager 时，可以通过 token_in_blocklist_loader 参数来指定一个函数，该函数用于检查 JWT token 是否在 blocklist 中。下面是一个使用 token_in_blocklist_loader 的例子：

from flask_jwt_extended import JWTManager, jwt_required, create_access_token, get_jwt_identity
from datetime import timedelta

app = Flask(__name__)
app.config['JWT_SECRET_KEY'] = 'super-secret'
app.config['JWT_ACCESS_TOKEN_EXPIRES'] = timedelta(hours=1)

jwt = JWTManager(app)

# Define a function to check if a token is in the blocklist
def check_if_token_in_blocklist(jwt_header, jwt_payload):
    # Check if the token is in the blocklist
    # Return True if the token is in the blocklist, False otherwise
    return False

# Set the token_in_blocklist_loader to the check_if_token_in_blocklist function
jwt.token_in_blocklist_loader(check_if_token_in_blocklist)

# Define a route to generate a JWT token
@app.route('/login', methods=['POST'])
def login():
    # Authenticate the user
    user_id = authenticate_user(request.form['username'], request.form['password'])
    if user_id is None:
        return jsonify({'message': 'Invalid credentials'}), 401

    # Create a JWT token
    access_token = create_access_token(identity=user_id)

    # Return the JWT token
    return jsonify({'access_token': access_token}), 200

# Define a route that requires a JWT token
@app.route('/protected', methods=['GET'])
@jwt_required()
def protected():
    # Get the user ID from the JWT token
    user_id = get_jwt_identity()

    # Return a message indicating that the user is authenticated
    return jsonify({'message': f'Authenticated as user {user_id}'}), 200

在上面的例子中，我们定义了一个名为 check_if_token_in_blocklist 的函数，该函数用于检查 JWT token 是否在 blocklist 中。然后，我们将该函数设置为 token_in_blocklist_loader 的值。在 login 路由中，我们使用 create_access_token 函数生成一个 JWT token，并将其返回给客户端。在 protected 路由中，我们使用 jwt_required 装饰器来保护该路由，只有携带有效的 JWT token 的请求才能访问该路由。在 protected 路由中，我们还使用 get_jwt_identity 函数获取 JWT token 中的用户 ID。