Flask 使用 JWT（三）flask-jwt-extended

Flask-JWT-Extended 是一个用于 Flask 应用程序的 JSON Web Token（JWT）扩展，它简化了 JWT 的生成、验证和身份验证过程。通过使用 Flask-JWT-Extended，您可以轻松地在 Flask 应用中实施 JWT 认证，提高应用程序的安全性。
在本篇教程中，我们将学习如何使用 Flask-JWT-Extended 扩展在 Flask 应用中实现 JWT 认证。我们将分步骤介绍安装、配置和使用 Flask-JWT-Extended 的详细过程，并通过示例展示如何处理用户身份验证和授权。

1. 安装 Flask-JWT-Extended
   要开始使用 Flask-JWT-Extended，首先需要将其安装到您的 Flask 项目中。您可以使用 pip 包管理器来安装它：
   pip install flask-jwt-extended
2. 配置 Flask-JWT-Extended
   安装完成后，您需要在 Flask 应用中配置 Flask-JWT-Extended。这通常在您的主应用程序模块中完成。首先，导入所需的模块：
   from flaskjwtextended import JWTManager, create_access_token, jwt_required, get_jwt_identity, unset_jwt_cookies
   然后，在应用程序初始化时实例化 JWTManager 对象：
   app = Flask(__name)
   jwt = JWTManager(app)
   接下来，您需要定义一个用于生成令牌的函数。该函数将返回一个包含用户身份信息的字典，例如：
   @app.route(‘/login’, methods=[‘POST’]) def login():
   user = authenticate_user(request.json[‘username’], request.json[‘password’]) # 假设您已经实现了 authenticate_user 函数来验证用户登录
   if user is None: return ‘Invalid username or password’, 401
   access_token = create_access_token(identity=user)
3. 处理用户身份验证和授权
   在您的应用程序中，您可以使用 jwt_required 装饰器来保护需要身份验证的端点。当访问受保护的端点时，如果用户未提供有效的 JWT，将返回 401 未授权状态。例如：
   @app.route(‘/protected’, methods=[‘GET’])
   jwt_required(get_user_identity) def protected(): # 使用自定义的 get_user_identity 函数获取用户身份信息
   return jsonify({‘message’: ‘Hello, user!’}), 200
   要检查 JWT 的有效性，您可以在视图函数中使用 get_jwt_identity 函数。如果令牌无效或过期，该函数将引发 JWTError 异常。例如：
   def get_user_identity(): # 自定义函数用于获取用户身份信息
   try:
   identity = get_jwt_identity()
   except JWTError:
   return None, 401
4. 注销和清除 JWT 令牌
   当用户注销或需要清除其 JWT 令牌时，您可以使用 unset_jwt_cookies 函数来清除客户端存储的令牌。例如：
   @app.route(‘/logout’, methods=[‘DELETE’]) def logout():
   unset_jwt_cookies(response=None)
   return jsonify({‘message’: ‘Successfully logged out’}), 200
   通过以上步骤，您应该已经成功在 Flask 应用中实现了 JWT 认证。请注意，这只是一个简单的示例，您可能需要根据您的应用程序需求进行更多的配置和定制化。另外，确保在生产环境中使用安全的加密算法和密钥来生成和验证 JWT。