修改message为中文

chatgpt-next-web/OnSalesData.py

@@ -1,5 +0,0 @@
-class OnSalesData:
-    def __init__(self, currentPackage, packages):
-        self.currentPackage = currentPackage
-        self.packages = packages
-

chatgpt-next-web/service.py

@@ -13,7 +13,6 @@ from flask import Flask, request, jsonify, Request, Response
 from redis import Redis
 from utils import nowtime
 import pay_package
-from OnSalesData import OnSalesData
 from ApiResponse import ApiResponse
 from flask_cors import CORS
 from mail import MyEmail
@@ -60,7 +59,7 @@ def send_verification_code_endpoint():
     # 保存验证码到Redis，并设置过期时间（例如，5分钟）
     redis.setex(email, 300, verification_code)
 
-    return jsonify({'code': 0, 'message': 'Verification code sent'})
+    return jsonify({'code': 0, 'message': '验证码已发送'})
 
 
 # 用户注册
@@ -74,15 +73,15 @@ def register():
 
     if is_email_registered(email):
         return jsonify({'code': 400, 'message': '邮箱已被注册'})
+    
+    # 检查用户名是否已被注册
+    if redis.hexists('users', username):
+        return jsonify({'code': 400, 'message': '用户名已存在'})
 
     # 检查验证码是否匹配
     stored_code = redis.get(email)
     if stored_code is None or verification_code != stored_code.decode('utf-8'):
-        return jsonify({'code': 400, 'message': 'Invalid verification code'})
-
-    # 检查用户名是否已被注册
-    if redis.hexists('users', username):
-        return jsonify({'code': 400, 'message': 'Username already exists'})
+        return jsonify({'code': 400, 'message': '验证码无效'})
 
     # 生成唯一的用户ID
     user_id = str(uuid.uuid4())
@@ -104,7 +103,7 @@ def register():
 
     return jsonify({
         'code': 0,
-        'message': 'Registration successful'
+        'message': '注册成功'
     })
 
 
@@ -118,11 +117,11 @@ def login():
     # 检查用户名和密码是否匹配
     user_data = redis.hget('users', username)
     if not user_data:
-        return jsonify({'code': 400, 'message': 'Invalid username'})
+        return jsonify({'code': 400, 'message': '用户名无效'})
 
     user_data = user_data.decode('utf-8')
     if password != eval(user_data)['password']:
-        return jsonify({'code': 400, 'message': 'Invalid password'})
+        return jsonify({'code': 400, 'message': '密码错误'})
 
     # 生成令牌
     token = generate_token(eval(user_data)['user_id'], username)
@@ -136,18 +135,6 @@ def login():
     })
 
 
-# 需要验证登录状态的接口
-@app.route('/protected', methods=['GET'])
-def protected():
-    token = parse_token(request)
-    # 验证令牌
-    if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'}), 200
-
-    # 如果用户未登录，则返回未授权的响应
-    return jsonify({'code': 401, 'message': 'Unauthorized'})
-
-
 @app.route('/resetPassword', methods=['POST'])
 def reset_password():
     email = request.json.get('email')
@@ -157,7 +144,7 @@ def reset_password():
     # 检查验证码是否匹配
     stored_code = redis.get(email)
     if stored_code is None or verification_code != stored_code.decode('utf-8'):
-        return jsonify({'code': 400, 'message': 'Invalid verification code'})
+        return jsonify({'code': 400, 'message': '验证码无效'})
 
     if not is_email_registered(email):
         return jsonify({'code': 400, 'message': '邮箱未注册'})
@@ -175,7 +162,7 @@ def logout():
     if not validate_token(token):
         # 将令牌添加到 Redis 黑名单
         redis.set(token, 'revoked')
-    return jsonify({'code': 0, 'message': 'Logout successful'})
+    return jsonify({'code': 0, 'message': '已登出'})
 
 
 @app.route('/create_order', methods=['POST'])
@@ -186,17 +173,17 @@ def create_order():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     # 根据套餐ID获取套餐信息
     package = pay_package.get_package_by_id(package_id)
     if not package:
-        return jsonify({'code': 400, 'message': 'Invalid package ID'})
+        return jsonify({'code': 400, 'message': '套餐不存在'})
 
     # 获取用户ID
     user_id = get_user_id_from_token(token)
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
     # 创建订单
     order_id = generate_order_id()
     order_data = {
@@ -243,12 +230,12 @@ def order_center():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     # 获取用户ID
     user_id = get_user_id_from_token(token)
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
 
     # 获取用户的所有订单
     user_orders = get_user_orders(user_id)
@@ -281,12 +268,12 @@ def query_order():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重���登录'})
 
     # 获取用户ID
     user_id = get_user_id_from_token(token)
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
 
     order_id = data['order_id']
 
@@ -295,7 +282,7 @@ def query_order():
     order_data = redis.hgetall(order_key)
 
     if not order_data:
-        return jsonify({'code': 404, 'message': 'Order not found'})
+        return jsonify({'code': 404, 'message': '订单无效'})
 
     # 将bytes类型的键和值转换为str类型
     order_data_str = {
@@ -334,12 +321,12 @@ def get_user_profile():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     # 获取用户ID
     user_id = get_user_id_from_token(token)
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
 
     free_count = get_free_count(user_id)
     package = get_user_package(user_id)
@@ -348,7 +335,6 @@ def get_user_profile():
 
     user_package_key = f'user:{user_id}:package'
     expiration_seconds = redis.ttl(user_package_key)
-    print("expiration_seconds = ", expiration_seconds, "is_package_expired = ", is_package_expired(user_id))
     formatted_expiration = ""
     if expiration_seconds > 0:
         expiration_datetime = (
@@ -419,66 +405,6 @@ def is_order_expired(order_data):
     return (current_time - created_at) >= expiration_time
 
 
-# 购买支付套餐
-@app.route('/purchase', methods=['POST'])
-def purchase():
-    package_id = request.json.get('package_id')
-    token = parse_token(request)
-
-    # 验证令牌
-    if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
-
-    # 根据套餐ID获取套餐信息
-    package = pay_package.get_package_by_id(package_id)
-    if not package:
-        return jsonify({'code': 400, 'message': 'Invalid package ID'})
-
-    user_id = get_user_id_from_token(token)
-    if not user_id:
-        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({'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({'code': 400, 'message': 'Cannot purchase lower level package'})
-
-    # 存储用户套餐信息到Redis
-    store_user_package(user_id, package)
-
-    return jsonify({'code': 0, 'message': 'Purchase successful'})
-
-
-# 验证用户聊天次数
-@app.route('/validate', methods=['POST'])
-def validate():
-    token = parse_token(request)
-    model = request.json.get('model')
-
-    # 验证令牌
-    if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
-
-    user_id = get_user_id_from_token(token)
-
-    if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
-
-    # 获取用户套餐信息
-    package = get_user_package(user_id)
-    if not package and get_free_count(user_id) <= 0:
-        return jsonify({'code': 400, 'message': '聊天次数已用完，请购买套餐'})
-
-    # 检查用户聊天次数是否超过限制
-    if exceeded_chat_limit(user_id, package, model):
-        return jsonify({'code': 400, 'message': 'Chat limit exceeded'})
-
-    return jsonify({'code': 0, 'message': 'Chat limit not exceeded'})
-
-
 @app.route('/v1/chat/completions', methods=['POST'])
 def proxy_chat_completions():
     token = parse_token(request)
@@ -486,12 +412,12 @@ def proxy_chat_completions():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     user_id = get_user_id_from_token(token)
 
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
 
     # 获取用户套餐信息
     package = get_user_package(user_id)
@@ -557,12 +483,12 @@ def handle_pick_up_free_chat_count():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     user_id = get_user_id_from_token(token)
 
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
+        return jsonify({'code': 400, 'message': '用户不存在'})
 
     # 领取赠送的聊天次数
     success = pick_up_free_chat_count(user_id)
@@ -579,13 +505,13 @@ def packageOnSales():
 
     # 验证令牌
     if not validate_token(token):
-        return jsonify({'code': 401, 'message': 'Invalid token'})
+        return jsonify({'code': 401, 'message': '请重新登录'})
 
     user_id = get_user_id_from_token(token)
 
     if not user_id:
-        return jsonify({'code': 400, 'message': 'User not found'})
-    
+        return jsonify({'code': 400, 'message': '用户不存在'})
+
     current_package = get_user_package(user_id)
     onSalesData = {
         'currentPackage': int(current_package.get(b'id', 0)),
@@ -787,7 +713,7 @@ def initialize_user_free_data(user_id):
 
 
 # 每天领取免费次数
-def pick_up_free_chat_count(user_id, basic_free_count = 5):
+def pick_up_free_chat_count(user_id, basic_free_count=5):
     free_data = get_user_free_data(user_id)
     if not free_data:
         free_data = initialize_user_free_data(user_id)