create_order中生成微信支付二维码

chatgpt-next-web/service.py

@@ -2,7 +2,6 @@ import json
 import random
 import string
 import uuid
-import time
 import jwt
 import datetime
 import requests
@@ -204,14 +203,27 @@ def create_order():
     order_key = f'order:{user_id}:{order_id}'
     redis.hmset(order_key, order_data)
     redis.expire(order_key, datetime.timedelta(hours=2))
+    
+    # 发起POST请求到http://localhost:8080/payPre获取codeUrl
+    payload = {
+        'order_id': order_id,
+        'amount': package['price'],
+        'user_id': user_id
+    }
+    response = requests.post('http://localhost:8080/payPre', json=payload)
+
+    if response.status_code == 200:
+        data = response.json()
+        code_url = data.get('data', {}).get('codeUrl')
+        if code_url:
+            order_data['codeUrl'] = code_url
+        else:
+            return jsonify({'code': 401, 'message': '支付二维码生成失败，请重试'})
 
     return jsonify({
         'code': 0,
         'message': 'Order created',
-        'data': {
-            'order_id': order_id,
-            'amount': package['price']
-        }
+        'data': order_data
     })