添加支付订单、订单中心api

chatgpt-next-web/service.py

@@ -171,6 +171,123 @@ def logout():
     return jsonify({'code': 0, 'message': 'Logout successful'})
 
 
+@app.route('/create_order', methods=['POST'])
+def create_order():
+    # 从请求中获取套餐ID和用户令牌
+    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'})
+
+    # 获取用户ID
+    user_id = get_user_id_from_token(token)
+    if not user_id:
+        return jsonify({'code': 400, 'message': 'User not found'})
+    # 创建订单
+    order_id = str(uuid.uuid4())
+    order_data = {
+        'order_id': order_id,
+        'user_id': user_id,
+        'package_id': package_id,
+        'amount': package['price'],
+        'status': 'pending',  # 订单状态设置为待支付
+        'created_at': nowtime().strftime('%Y-%m-%d %H:%M:%S')  # 将时间转换为字符串格式保存
+    }
+    # 将订单信息存储到Redis，并设置过期时间为2小时
+    order_key = f'order:{user_id}:{order_id}'
+    redis.hmset(order_key, order_data)
+    redis.expire(order_key, datetime.timedelta(hours=2))
+
+    return jsonify({
+        'code': 0,
+        'message': 'Order created',
+        'data': {
+            'order_id': order_id,
+            'amount': package['price']
+        }
+    })
+
+
+@app.route('/order_center', methods=['POST'])
+def order_center():
+    token = parse_token(request)
+
+    # 验证令牌
+    if not validate_token(token):
+        return jsonify({'code': 401, 'message': 'Invalid token'})
+
+    # 获取用户ID
+    user_id = get_user_id_from_token(token)
+    if not user_id:
+        return jsonify({'code': 400, 'message': 'User not found'})
+
+    # 获取用户的所有订单
+    user_orders = get_user_orders(user_id)
+
+    print(user_orders)
+
+    # 获取该用户的有效订单列表
+    valid_orders = [
+        order for order in user_orders
+        if not is_order_expired(order)
+    ]
+
+    print(valid_orders)
+    return jsonify({
+        'code': 0,
+        'message': 'Order center',
+        'data': {
+            'user_orders': valid_orders
+        }
+    })
+
+
+def get_user_orders(user_id):
+    # 从Redis中获取该用户的所有订单数据
+    user_orders = []
+    for key in redis.scan_iter(f'order:{user_id}:*'):  # 修改获取键的方式
+        order_data = redis.hgetall(key)
+        # 将bytes类型的键和值转换为str类型
+        order_data_str = {
+            k.decode('utf-8'): v.decode('utf-8') for k, v in order_data.items()
+        }
+        user_orders.append(order_data_str)
+    return user_orders
+
+
+def is_order_expired(order_data):
+    print("is_order_expired", order_data)
+    # 获取订单创建时间
+    created_at_str = order_data.get('created_at')
+    if not created_at_str:
+        print("created_at_str is None")
+        return True
+
+    # 将订单创建时间转换为datetime对象
+    created_at = datetime.datetime.strptime(
+        created_at_str,
+        '%Y-%m-%d %H:%M:%S'
+    )
+
+    # 订单有效期为2小时
+    expiration_time = datetime.timedelta(hours=2)
+
+    # 获取当前时间
+    current_time = nowtime()
+
+    print(current_time, created_at, expiration_time)
+
+    # 检查时间差是否超过订单有效期
+    return (current_time - created_at) >= expiration_time
+
+
 # 购买支付套餐
 @app.route('/purchase', methods=['POST'])
 def purchase():
@@ -374,10 +491,8 @@ def generate_token(user_id, username):
 # 验证令牌
 def validate_token(token):
     try:
-        print("validate_token", "token:", token)
         # 使用密钥进行解码
         payload = jwt.decode(token, SECERT_KEY, algorithms=['HS256'])
-        print("validate_token", "payload:", payload)
 
         # 检查令牌的过期时间
         if 'exp' in payload and nowtime() > datetime.datetime.fromtimestamp(payload['exp']):
@@ -385,7 +500,7 @@ def validate_token(token):
 
         if 'user_id' not in payload or 'username' not in payload:
             return False
-        
+
         return True
     except (jwt.DecodeError, jwt.InvalidTokenError):
         return False