Flask 学习-32.flask_jwt_extended 自定义装饰器

前言

创建自己的装饰器来扩展此扩展提供的装饰器的功能。例如，您可能想要创建自己的装饰器来验证 JWT 是否存在以及验证当前用户是否是管理员。

自定义装饰器

flask_jwt_extended.verify_jwt_in_request()可用于构建您自己的装饰器。这与 . 使用的功能相同flask_jwt_extended.jwt_required()。

官方文档示例

from functools import wraps

from flask import Flask
from flask import jsonify

from flask_jwt_extended import create_access_token
from flask_jwt_extended import get_jwt
from flask_jwt_extended import JWTManager
from flask_jwt_extended import verify_jwt_in_request

app = Flask(__name__)

app.config["JWT_SECRET_KEY"] = "super-secret"  # Change this!
jwt = JWTManager(app)


# Here is a custom decorator that verifies the JWT is present in the request,
# as well as insuring that the JWT has a claim indicating that this user is
# an administrator
def admin_required():
    def wrapper(fn):
        @wraps(fn)
        def decorator(*args, **kwargs):
            verify_jwt_in_request()
   
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24