python token flask_flask 实现token机制

要在 Flask 中实现 Token 机制，可以使用 JWT（JSON Web Tokens）库。下面是一个简单的示例代码：

from flask import Flask, request
from flask_jwt import JWT, jwt_required, current_identity
from werkzeug.security import safe_str_cmp

app = Flask(__name__)
app.config['SECRET_KEY'] = 'super-secret'
app.config['JWT_EXPIRATION_DELTA'] = timedelta(days=1)

class User:
    def __init__(self, id, username, password):
        self.id = id
        self.username = username
        self.password = password

    def __str__(self):
        return f"User(id='{self.id}')"

users = [
    User(1, 'user1', 'password1'),
    User(2, 'user2', 'password2'),
]

def authenticate(username, password):
    user = next((user for user in users if user.username == username), None)
    if user and safe_str_cmp(user.password.encode('utf-8'), password.encode('utf-8')):
        return user

def identity(payload):
    user_id = payload['identity']
    return next((user for user in users if user.id == user_id), None)

jwt = JWT(app, authenticate, identity)

@app.route('/protected')
@jwt_required()
def protected():
    return f'Hello, {current_identity}!'

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

在上面的代码中，我们首先定义了一个 `User` 类来存储用户信息。然后定义了一个 `authenticate` 函数来用于用户认证，它会查找用户列表中是否存在指定的用户名和密码。如果存在，则返回该用户对象。

接下来，我们定义了一个 `identity` 函数，它会根据 JWT 中存储的用户 ID 来查找用户对象。最后，我们创建了一个 `jwt` 对象，并将其绑定到 Flask 中。

最后，我们定义了一个受保护的路由 `/protected`，并使用 `@jwt_required()` 装饰器来保护它。这意味着在访问该路由时，用户必须先提供有效的 JWT 才能继续访问。

当用户成功提供有效的 JWT 且通过身份验证时，`current_identity` 会返回该用户对象，我们可以在响应中使用它。