You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
141 lines
4.2 KiB
141 lines
4.2 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 4月
|
|
# author: NinEveN
|
|
# date: 2021/4/15
|
|
|
|
import socket
|
|
import time
|
|
import uuid
|
|
import hashlib
|
|
import xml.etree.ElementTree as ET
|
|
import requests
|
|
|
|
|
|
def get_nonce_str():
|
|
"""
|
|
获取随机字符串
|
|
:return:
|
|
"""
|
|
return str(uuid.uuid4()).replace('-', '')
|
|
|
|
|
|
def dict_to_order_xml(dict_data):
|
|
"""
|
|
dict to order xml
|
|
ASCII码从小到大排序
|
|
:param dict_data:
|
|
:return:
|
|
"""
|
|
xml = ["<xml>"]
|
|
for k in sorted(dict_data):
|
|
xml.append("<{0}>{1}</{0}>".format(k, dict_data.get(k)))
|
|
xml.append("</xml>")
|
|
return "".join(xml)
|
|
|
|
|
|
def dict_to_xml(dict_data):
|
|
xml = ["<xml>"]
|
|
for k, v in dict_data.items():
|
|
xml.append("<{0}>{1}</{0}>".format(k, v))
|
|
xml.append("</xml>")
|
|
return "".join(xml)
|
|
|
|
|
|
def xml_to_dict(xml_data):
|
|
"""
|
|
xml to dict
|
|
:param xml_data:
|
|
:return:
|
|
"""
|
|
xml_dict = {}
|
|
root = ET.fromstring(xml_data)
|
|
for child in root:
|
|
xml_dict[child.tag] = child.text
|
|
return xml_dict
|
|
|
|
|
|
class WxPay(object):
|
|
"""
|
|
https://pay.weixin.qq.com/wiki/doc/api/wxa/wxa_api.php?chapter=7_4&index=3
|
|
"""
|
|
|
|
def __init__(self, app_id, mch_id, notify_url, merchant_key):
|
|
self.url = 'https://api.mch.weixin.qq.com/v3/pay/transactions/native'
|
|
self.app_id = app_id # 微信分配的小程序ID
|
|
self.mch_id = mch_id # 商户号
|
|
self.notify_url = notify_url # 通知地址
|
|
self.spbill_create_ip = socket.gethostbyname(socket.gethostname()) # 获取本机ip
|
|
self.merchant_key = merchant_key # 商户KEY，修改为自己的
|
|
|
|
def create_sign(self, pay_data):
|
|
"""
|
|
生成签名
|
|
https://pay.weixin.qq.com/wiki/doc/api/wxa/wxa_api.php?chapter=4_3
|
|
:param pay_data:
|
|
:return:
|
|
"""
|
|
|
|
# 拼接stringA
|
|
string_a = '&'.join(["{0}={1}".format(k, pay_data.get(k)) for k in sorted(pay_data)])
|
|
# 拼接key
|
|
string_sign_temp = '{0}&key={1}'.format(string_a, self.merchant_key).encode('utf-8')
|
|
# md5签名
|
|
sign = hashlib.md5(string_sign_temp).hexdigest()
|
|
return sign.upper()
|
|
|
|
def get_pay_info(self, pay_data):
|
|
"""
|
|
支付统一下单
|
|
:return:
|
|
"""
|
|
# 调用签名函数
|
|
|
|
post_data = {
|
|
'appid': self.app_id, # 小程序ID
|
|
'mch_id': self.mch_id, # 商户号
|
|
'description': pay_data.get('description'), # 商品描述
|
|
'out_trade_no': pay_data.get('out_trade_no'), # 商户订单号
|
|
'time_expire': pay_data.get('time_expire'), # 交易结束时间 示例值：2018-06-08T10:34:56+08:00
|
|
'attach': pay_data.get('attach'), # 附加数据，在查询API和支付通知中原样返回，可作为自定义参数使用
|
|
'notify_url': self.notify_url, # 通知地址
|
|
'amount': {
|
|
'total': pay_data.get('total'), # 订单总金额，单位为分。示例值：100
|
|
'currency': 'CNY'
|
|
}
|
|
}
|
|
sign = self.create_sign(post_data)
|
|
post_data['sign'] = sign
|
|
|
|
xml = dict_to_xml(post_data)
|
|
|
|
# 统一下单接口请求
|
|
r = requests.post(self.url, data=xml.encode("utf-8"))
|
|
r.encoding = "utf-8"
|
|
res = xml_to_dict(r.text)
|
|
err_code_des = res.get('err_code_des')
|
|
# 出错信息
|
|
if err_code_des:
|
|
return {'code': 40001, 'msg': err_code_des}
|
|
prepay_id = res.get('prepay_id')
|
|
|
|
return self.re_sign(post_data, prepay_id)
|
|
|
|
def re_sign(self, post_data, prepay_id):
|
|
"""
|
|
再次对返回的数据签名
|
|
https://pay.weixin.qq.com/wiki/doc/api/wxa/wxa_api.php?chapter=7_7&index=3
|
|
:param post_data:
|
|
:param prepay_id:
|
|
:return:
|
|
"""
|
|
pay_sign_data = {
|
|
'appId': self.app_id, # 注意大小写与统一下单不一致
|
|
'timeStamp': post_data.get('out_trade_no'),
|
|
'nonceStr': post_data.get('nonce_str'),
|
|
'package': 'prepay_id={0}'.format(prepay_id),
|
|
'signType': 'MD5',
|
|
}
|
|
pay_sign = self.create_sign(pay_sign_data)
|
|
pay_sign_data['paySign'] = pay_sign
|
|
return pay_sign_data
|
|
|