### 实现注册、登录、查询和删除接口 以下是基于 Flask 和 PyJWT 的示例代码，展示如何实现注册、登录、查询以及删除接口的功能。其中，查询接口需要通过登录生成的 Token 进行身份验证。 #### 安装依赖库 首先安装所需的依赖库： ```bash pip install flask pyjwt cryptography ``` --- #### 示例代码 ```python from flask import Flask, request, jsonify import jwt import datetime app = Flask(__name__) # 密钥用于签名 JWT Token SECRET_KEY = "your_secret_key" # 模拟数据库存储用户数据 users_db = {} @app.route('/register', methods=['POST']) def register(): """ 注册新用户。 输入：用户名 (username)，密码 (password) 输出：成功消息或错误提示 """ data = request.json username = data.get('username') password = data.get('password') if not username or not password: return jsonify({"error": "Username and Password are required."}), 400 if username in users_db: return jsonify({"error": "User already exists."}), 409 users_db[username] = {"password": password} return jsonify({"message": "User registered successfully."}), 201 @app.route('/login', methods=['POST']) def login(): """ 用户登录并生成 Token。 输入：用户名 (username)，密码 (password) 输出：Token 或错误提示 """ data = request.json username = data.get('username') password = data.get('password') user = users_db.get(username) if not user or user["password"] != password: return jsonify({"error": "Invalid credentials."}), 401 # 创建 Token token_payload = { 'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1), 'iat': datetime.datetime.utcnow(), 'sub': username } token = jwt.encode(token_payload, SECRET_KEY, algorithm='HS256') # 使用 HS256 算法加密 return jsonify({'token': token.decode('utf-8')}), 200 @app.route('/query', methods=['GET']) def query(): """ 查询接口，需携带有效的 Token。 输入：无 输出：欢迎信息或错误提示 """ auth_header = request.headers.get('Authorization') if not auth_header: return jsonify({"error": "Missing Authorization header."}), 401 try: token = auth_header.split(" ")[1] payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256']) # 验证 Token username = payload['sub'] return jsonify({"message": f"Welcome, {username}!"}) except jwt.ExpiredSignatureError: return jsonify({"error": "Token has expired."}), 401 except jwt.InvalidTokenError: return jsonify({"error": "Invalid Token."}), 401 @app.route('/delete/<string:username>', methods=['DELETE']) def delete_user(username): """ 删除指定用户。 输入：用户名 (路径参数) 输出：删除成功的消息或错误提示 """ if username not in users_db: return jsonify({"error": "User does not exist."}), 404 del users_db[username] return jsonify({"message": "User deleted successfully."}), 200 if __name__ == '__main__': app.run(debug=True) ``` --- ### 功能说明 1. **注册接口 (`/register`)** 接收 `POST` 请求，输入字段为 `username` 和 `password`。如果用户提供合法的数据，则将其保存到模拟数据库中[^1]。 2. **登录接口 (`/login`)** 接收 `POST` 请求，验证用户的 `username` 和 `password` 是否匹配。如果验证成功，则生成一个带有过期时间的 JWT Token 并返回给客户端[^5]。 3. **查询接口 (`/query`)** 接收 `GET` 请求，要求在请求头部提供有效的 JWT Token。只有当 Token 合法且未过期时，才会返回欢迎信息[^2]。 4. **删除接口 (`/delete/{username}`)** 接收 `DELETE` 请求，根据路径中的 `username` 参数删除对应的用户记录。如果没有找到该用户，则返回错误信息[^1]。 --- ### 注意事项 - **安全性**：实际应用中不应明文存储密码，建议使用哈希算法（如 bcrypt）对密码进行加密后再存储。 - **Token 生命周期管理**：可以通过设置更短的有效期来增强系统的安全性，并考虑引入刷新机制以延长访问权限[^3]。 - **异常处理**：对于可能发生的各种异常情况（如网络中断），应在生产环境中增加更多的健壮性逻辑[^4]。 ---