update pay ui

alipay_pay_worker.py

@@ -6,7 +6,8 @@
 
 用法:
     python alipay_pay_worker.py check              # 检查配置
-    python alipay_pay_worker.py create <order_no> <amount> <subject> [body]  # 创建订单
+    python alipay_pay_worker.py create <order_no> <amount> <subject> [body] [pay_type]  # 创建订单
+        pay_type: page=电脑网站支付(默认), wap=手机网站支付
     python alipay_pay_worker.py query <order_no>   # 查询订单
 """
 
@@ -31,19 +32,39 @@ def check_config():
         }
 
 
-def create_order(order_no, amount, subject, body=None):
-    """创建支付宝订单"""
+def create_order(order_no, amount, subject, body=None, pay_type='page'):
+    """创建支付宝订单
+
+    Args:
+        order_no: 订单号
+        amount: 金额
+        subject: 标题
+        body: 描述
+        pay_type: 支付类型 'page'=电脑网站支付, 'wap'=手机网站支付
+    """
     try:
         from alipay_pay import create_alipay_instance
         alipay = create_alipay_instance()
 
-        result = alipay.create_page_pay_url(
-            out_trade_no=order_no,
-            total_amount=str(amount),
-            subject=subject,
-            body=body,
-            timeout_express='30m'
-        )
+        if pay_type == 'wap':
+            # 手机网站支付
+            result = alipay.create_wap_pay_url(
+                out_trade_no=order_no,
+                total_amount=str(amount),
+                subject=subject,
+                body=body,
+                timeout_express='30m',
+                quit_url='https://valuefrontier.cn/pricing'  # 用户取消支付时返回的页面
+            )
+        else:
+            # 电脑网站支付（默认）
+            result = alipay.create_page_pay_url(
+                out_trade_no=order_no,
+                total_amount=str(amount),
+                subject=subject,
+                body=body,
+                timeout_express='30m'
+            )
 
         return result
     except Exception as e:
@@ -108,7 +129,9 @@ def main():
                 amount = sys.argv[3]
                 subject = sys.argv[4]
                 body = sys.argv[5] if len(sys.argv) > 5 else None
-                result = create_order(order_no, amount, subject, body)
+                # pay_type: page=电脑网站支付, wap=手机网站支付
+                pay_type = sys.argv[6] if len(sys.argv) > 6 else 'page'
+                result = create_order(order_no, amount, subject, body, pay_type)
 
         elif command == 'query':
             if len(sys.argv) < 3: