reponse json 添加code

redis/test_user_redis.py

@@ -40,7 +40,7 @@ def send_verification_code_endpoint():
     # 保存验证码到Redis，并设置过期时间（例如，5分钟）
     redis.setex(email, 300, verification_code)
 
-    return jsonify({'message': 'Verification code sent'})
+    return jsonify({'code': 0, 'message': 'Verification code sent'})
 
 
 # 用户注册
@@ -73,7 +73,7 @@ def register():
     }
     redis.hset('users', username, json.dumps(user_data))
 
-    return jsonify({'message': 'Registration successful'})
+    return jsonify({'code': 0, 'message': 'Registration successful'})
 
 
 # 用户登录
@@ -86,16 +86,16 @@ def login():
     # 检查用户名和密码是否匹配
     user_data = redis.hget('users', username)
     if not user_data:
-        return jsonify({'message': 'Invalid username'}), 400
+        return jsonify({'code': 400, 'message': 'Invalid username'})
 
     user_data = user_data.decode('utf-8')
     if password != eval(user_data)['password']:
-        return jsonify({'message': 'Invalid password'}), 400
+        return jsonify({'code': 400, 'message': 'Invalid password'})
 
     # 登录验证通过，将用户信息存储到会话中
     session['username'] = request.json.get('username')
 
-    return jsonify({'message': 'Login successful'})
+    return jsonify({'code': 0, 'message': 'Login successful'})
 
 
 # 需要验证登录状态的接口
@@ -105,10 +105,10 @@ def protected():
     if 'username' in session:
         username = session['username']
         # 其他业务逻辑...
-        return jsonify({'message': f'Hello, {username}! This is a protected endpoint.'})
+        return jsonify({'code': 0, 'message': f'Hello, {username}! This is a protected endpoint.'})
 
     # 如果用户未登录，则返回未授权的响应
-    return jsonify({'message': 'Unauthorized'}), 401
+    return jsonify({'code': 401, 'message': 'Unauthorized'})
 
 
 # 用户注销
@@ -116,7 +116,7 @@ def protected():
 def logout():
     # 清除会话中的用户信息
     session.pop('username', None)
-    return jsonify({'message': 'Logout successful'})
+    return jsonify({'code': 0, 'message': 'Logout successful'})
 
 
 # 购买支付套餐
@@ -127,29 +127,29 @@ def purchase():
     # 根据套餐ID获取套餐信息
     package = get_package_by_id(package_id)
     if not package:
-        return jsonify({'message': 'Invalid package ID'}), 400
+        return jsonify({'code': 400, 'message': 'Invalid package ID'})
     
     # 根据用户名查询用户ID
     username = session.get('username')
     if not username:
-        return jsonify({'message': 'User not logged in'}), 400
+        return jsonify({'code': 400, 'message': 'User not logged in'})
 
     user_id = get_user_id_by_username(username)
     if not user_id:
-        return jsonify({'message': 'User not found'}), 400
+        return jsonify({'code': 400, 'message': 'User not found'})
 
     # 检查用户是否已经支付过当前套餐
     if not is_package_expired(user_id) and has_purchased_package(user_id, package_id):
-        return jsonify({'message': 'Package already purchased'}), 400
+        return jsonify({'code': 400, 'message': 'Package already purchased'})
 
     # 检查如果用户已经支付了高级套餐，则不能支付比高级套餐更低级的基础套餐
     if not is_package_expired(user_id) and has_purchased_advanced_package(user_id) and package_id == '1':
-        return jsonify({'message': 'Cannot purchase lower level package'}), 400
+        return jsonify({'code': 400, 'message': 'Cannot purchase lower level package'})
     
     # 存储用户套餐信息到Redis
     store_user_package(user_id, package)
 
-    return jsonify({'message': 'Purchase successful'})
+    return jsonify({'code': 0, 'message': 'Purchase successful'})
 
 
 # 验证用户聊天次数
@@ -160,18 +160,18 @@ def validate():
     user_id = get_user_id_by_username(username)
     
     if not user_id:
-        return jsonify({'message': 'User not found'}), 400
+        return jsonify({'code': 400, 'message': 'User not found'})
     
     # 获取用户套餐信息
     package = get_user_package(user_id)
     if not package:
-        return jsonify({'message': 'User has not purchased any package'}), 400
+        return jsonify({'code': 400, 'message': 'User has not purchased any package'})
 
     # 检查用户聊天次数是否超过限制
     if exceeded_chat_limit(user_id, package):
-        return jsonify({'message': 'Chat limit exceeded'}), 400
+        return jsonify({'code': 400, 'message': 'Chat limit exceeded'})
 
-    return jsonify({'message': 'Chat limit not exceeded'})
+    return jsonify({'code': 0, 'message': 'Chat limit not exceeded'})
 
 
 # 获取用户ID通过用户名