使用flask完成jwt登录

在Flask中实现JWT（JSON Web Tokens）登录，你需要几个关键步骤：

1. **安装依赖**：首先，确保已经安装了Flask-JWT-Extended库，可以使用pip安装：

   pip install flask-jwt-extended

2. **初始化应用**：在Flask应用程序中，导入并配置JWT扩展：

   from flask import Flask
   from flask_jwt_extended import JWTManager

   app = Flask(__name__)
   app.config['JWT_SECRET_KEY'] = 'your_secret_key'  # 替换为你自定义的密钥
   jwt = JWTManager(app)

3. **装饰器保护路由**：对于需要鉴权的API，使用`@jwt_required`装饰器：

   @app.route('/protected')
   @jwt_required()
   def protected():
       return {'message': 'You are authorized'}

4. **注册登录函数**：编写登录函数，生成JWT令牌并将其返回给用户：

   @app.route('/login', methods=['POST'])
   def login():
       username = request.json.get('username')  # 假设用户名和密码在请求体中
       password = request.json.get('password')

       # 验证用户名和密码，然后创建JWT token
       if validate_credentials(username, password):
           access_token = create_access_token(identity=username)  # 使用用户名作为token的主体
           return jsonify(access_token=access_token), 200
       else:
           return jsonify(error='Invalid credentials'), 401

5. **辅助函数**：创建`create_access_token`和`validate_credentials`等辅助函数来处理JWT的生成、解析和用户验证。

6. **刷新令牌**：若需要，还可以提供令牌刷新功能，让用户在令牌即将过期时获取新的令牌。

记得定期更新JWT密钥以增强安全性。