添加支付套餐检查

redis/test_user_redis.py

@@ -137,6 +137,14 @@ def purchase():
     if not user_id:
         return jsonify({'message': 'User not found'}), 400
 
+    # 检查用户是否已经支付过当前套餐
+    if has_purchased_package(user_id, package_id):
+        return jsonify({'message': 'Package already purchased'}), 400
+
+    # 检查如果用户已经支付了高级套餐，则不能支付比高级套餐更低级的基础套餐
+    if has_purchased_advanced_package(user_id) and package_id == '1':
+        return jsonify({'message': 'Cannot purchase lower level package'}), 400
+    
     # 存储用户套餐信息到Redis
     store_user_package(user_id, package)
 
@@ -152,6 +160,7 @@ def validate():
     
     if not user_id:
         return jsonify({'message': 'User not found'}), 400
+    
     # 获取用户套餐信息
     package = get_user_package(user_id)
     if not package:
@@ -178,12 +187,14 @@ def get_user_id_by_username(username):
 def get_package_by_id(package_id):
     packages = {
         '1': {
+            'package_id': '1',
             'name': 'Package 1',
             'basic_chat_limit': 10,
             'advanced_chat_limit': 10,
             'price': 10
         },
         '2': {
+            'package_id': '2',
             'name': 'Package 2',
             'basic_chat_limit': -1,  # -1 表示无限次
             'advanced_chat_limit': -1,
@@ -197,6 +208,7 @@ def get_package_by_id(package_id):
 # 存储用户套餐信息到Redis
 def store_user_package(user_id, package):
     user_package_key = f'user:{user_id}:package'
+    redis.hset(user_package_key, 'package_id', package['package_id'])
     redis.hset(user_package_key, 'name', package['name'])
     redis.hset(user_package_key, 'basic_chat_limit',
                package['basic_chat_limit'])
@@ -211,6 +223,20 @@ def get_user_package(user_id):
     return package
 
 
+# 检查用户是否已经支付过指定套餐
+def has_purchased_package(user_id, package_id):
+    user_package_key = f'user:{user_id}:package'
+    purchased_package_id = redis.hget(user_package_key, 'package_id')
+    return purchased_package_id.decode('utf-8') == str(package_id)
+
+
+# 检查用户是否已经支付了高级套餐
+def has_purchased_advanced_package(user_id):
+    user_package_key = f'user:{user_id}:package'
+    purchased_package_id = redis.hget(user_package_key, 'package_id')
+    return purchased_package_id.decode('utf-8') == '2'
+
+
 # 检查用户聊天次数是否超过限制
 def exceeded_chat_limit(user_id, package):
     user_basic_chat_key = f'user:{user_id}:basic_chat'