164 lines
4.7 KiB
Python
164 lines
4.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
支付宝支付工作脚本
|
|
用于在 subprocess 中执行，绕过 eventlet DNS 问题
|
|
|
|
用法:
|
|
python alipay_pay_worker.py check # 检查配置
|
|
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> # 查询订单
|
|
"""
|
|
|
|
import sys
|
|
import json
|
|
|
|
|
|
def check_config():
|
|
"""检查支付宝配置"""
|
|
try:
|
|
from alipay_pay import check_alipay_ready
|
|
is_ready, message = check_alipay_ready()
|
|
return {
|
|
'success': is_ready,
|
|
'message': message if is_ready else None,
|
|
'error': None if is_ready else message
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
'success': False,
|
|
'error': str(e)
|
|
}
|
|
|
|
|
|
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()
|
|
|
|
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:
|
|
return {
|
|
'success': False,
|
|
'error': str(e)
|
|
}
|
|
|
|
|
|
def query_order(order_no):
|
|
"""查询支付宝订单"""
|
|
try:
|
|
from alipay_pay import create_alipay_instance
|
|
alipay = create_alipay_instance()
|
|
|
|
result = alipay.query_order(out_trade_no=order_no)
|
|
|
|
# 转换状态为统一格式
|
|
if result.get('success'):
|
|
trade_status = result.get('trade_status')
|
|
# 映射支付宝状态到统一状态
|
|
if trade_status in ['TRADE_SUCCESS', 'TRADE_FINISHED']:
|
|
result['trade_state'] = 'SUCCESS'
|
|
elif trade_status == 'WAIT_BUYER_PAY':
|
|
result['trade_state'] = 'NOTPAY'
|
|
elif trade_status == 'TRADE_CLOSED':
|
|
result['trade_state'] = 'CLOSED'
|
|
else:
|
|
result['trade_state'] = trade_status
|
|
|
|
return result
|
|
except Exception as e:
|
|
return {
|
|
'success': False,
|
|
'error': str(e)
|
|
}
|
|
|
|
|
|
def main():
|
|
"""主函数"""
|
|
if len(sys.argv) < 2:
|
|
print(json.dumps({
|
|
'success': False,
|
|
'error': '缺少命令参数。用法: check | create <order_no> <amount> <subject> | query <order_no>'
|
|
}))
|
|
sys.exit(1)
|
|
|
|
command = sys.argv[1].lower()
|
|
|
|
try:
|
|
if command == 'check':
|
|
result = check_config()
|
|
|
|
elif command == 'create':
|
|
if len(sys.argv) < 5:
|
|
result = {
|
|
'success': False,
|
|
'error': '创建订单需要参数: order_no, amount, subject'
|
|
}
|
|
else:
|
|
order_no = sys.argv[2]
|
|
amount = sys.argv[3]
|
|
subject = sys.argv[4]
|
|
body = sys.argv[5] if len(sys.argv) > 5 else None
|
|
# 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:
|
|
result = {
|
|
'success': False,
|
|
'error': '查询订单需要参数: order_no'
|
|
}
|
|
else:
|
|
order_no = sys.argv[2]
|
|
result = query_order(order_no)
|
|
|
|
else:
|
|
result = {
|
|
'success': False,
|
|
'error': f'未知命令: {command}'
|
|
}
|
|
|
|
print(json.dumps(result, ensure_ascii=False))
|
|
|
|
except Exception as e:
|
|
print(json.dumps({
|
|
'success': False,
|
|
'error': str(e)
|
|
}, ensure_ascii=False))
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|