添加查询订单接口

chatgpt-next-web/service.py

@@ -266,6 +266,41 @@ def order_center():
     })
 
 
+@app.route('/query_order', methods=['POST'])
+def query_order():
+    data = request.json
+    if 'order_id' not in data:
+        return jsonify({'code': 400, 'message': '订单无效'})
+
+    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'})
+
+    order_id = data['order_id']
+
+    # 从Redis中获取指定订单数据
+    order_key = f'order:{user_id}:{order_id}'
+    order_data = redis.hgetall(order_key)
+
+    if not order_data:
+        return jsonify({'code': 404, 'message': 'Order not found'})
+
+    # 将bytes类型的键和值转换为str类型
+    order_data_str = {
+        k.decode('utf-8'): int(v.decode('utf-8')) if k.decode('utf-8') == 'amount' else v.decode('utf-8')
+        for k, v in order_data.items()
+    }
+
+    return jsonify({'code': 0, 'message': 'Success', 'data': order_data_str})
+
+
 @app.route('/pay/notify', methods=['POST'])
 def handle_payment_notification():
     # 获取POST请求中的JSON数据，这是微信支付发送的通知数据
@@ -302,6 +337,13 @@ def get_user_orders(user_id):
     return user_orders
 
 
+def get_order_by_id(user_id, order_id):
+    # 从Redis中获取指定订单数据
+    order_key = f'order:{user_id}:{order_id}'
+    order_data = redis.hgetall(order_key)
+    return order_data
+
+
 def is_order_expired(order_data):
     print("is_order_expired", order_data)
     # 获取订单创建时间