Python实现微信小程序支付功能

Python实现微信小程序支付功能完整攻略

微信小程序是一种轻量级的应用程序，被广泛应用于各种场景中，例如线上商城、酒店预订、交通出行等。在这些场景中，支付功能是微信小程序的核心功能之一。本文将详细讲解如何用Python实现微信小程序的支付功能，包括如何获取access_token、预下单、生成签名等。

获取access_token

在使用微信支付之前，首先需要获取调用微信支付接口需要的access_token。可以通过微信公众平台中的开发者中心获取access_token。

import requests
import time


APPID = 'your appid'
SECRET = 'your secret'

def get_access_token():
    url = 'https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid={0}&secret={1}'.format(APPID, SECRET)
    res = requests.get(url).json()
    access_token = res.get('access_token', '')
    if access_token:
        return access_token
    else:
        print(res)
        return ''

预下单与生成签名

在获取access_token之后，就可以进行预下单操作了。预下单是指商户向微信支付系统提交支付请求的过程。在预下单之前，需要生成签名，以保证请求的合法性。签名的生成过程如下：

import hashlib
import random
import string

def generate_md5(string):
    md5 = hashlib.md5()
    md5.update(string.encode('utf-8'))
    return md5.hexdigest()

def generate_nonce_str(length=32):
    return ''.join(random.choices(string.ascii_letters + string.digits, k=length))

def generate_sign(data):
    key = 'your key'
    stringA = '&'.join(["{0}={1}".format(k, data.get(k, '')) for k in sorted(data)])
    stringSignTemp = '{0}&key={1}'.format(stringA, key)
    sign = generate_md5(stringSignTemp).upper()
    return sign

def pre_order():
    appid = 'your appid'
    body = 'your body'
    mch_id = 'your mchid'
    nonce_str = generate_nonce_str()
    notify_url = 'your notify url'
    out_trade_no = 'your order number'
    spbill_create_ip = 'your spbill ip'
    total_fee = 'your total fee'
    trade_type = 'JSAPI'
    openid = 'your openid'

    data = {
        'appid': appid,
        'body': body,
        'mch_id': mch_id,
        'nonce_str': nonce_str,
        'notify_url': notify_url,
        'out_trade_no': out_trade_no,
        'spbill_create_ip': spbill_create_ip,
        'total_fee': total_fee,
        'trade_type': trade_type,
        'openid': openid
        }

    data['sign'] = generate_sign(data)

    url = 'https://api.mch.weixin.qq.com/pay/unifiedorder'
    res = requests.post(url, data=data.encode('utf-8')).text
    print(res)

生成支付参数

预下单成功后，需要生成支付参数以调起微信支付。生成支付参数的代码如下：

def generate_payment_params(prepay_id):
    appid = 'your appid'
    nonce_str = generate_nonce_str()
    package = 'prepay_id={0}'.format(prepay_id)
    sign_type = 'MD5'
    timestamp = str(int(time.time()))

    data = {
        'appId': appid,
        'nonceStr': nonce_str,
        'package': package,
        'signType': sign_type,
        'timeStamp': timestamp
    }
    data['sign'] = generate_sign(data)

    return data

测试用例

下面给出两个简单的测试用例，用于检验以上代码的正确性。

测试用例一

假设商品价格为1元，订单号为"test0001"，用户的openid为"oVd0K1TVf2lmSL5Kwg0JeScdaE2o"，则可以使用如下代码进行测试：

def test_pay(order_id, openid, total_fee):
    access_token = get_access_token()
    if not access_token:
        return ''
    prepay_id = pre_order(access_token, order_id, openid, total_fee)
    if not prepay_id:
        return ''
    paymentParams = generate_payment_params(prepay_id)
    return paymentParams

测试用例二

假设商品价格为2元，订单号为"test0002"，用户的openid为"oVd0K1TVf2lmSL5Kwg0JeScdaE2o"，则可以使用如下代码进行测试：

def test_pay(order_id, openid, total_fee):
    access_token = get_access_token()
    if not access_token:
        return ''
    prepay_id = pre_order(access_token, order_id, openid, total_fee)
    if not prepay_id:
        return ''
    paymentParams = generate_payment_params(prepay_id)
    return paymentParams

以上就是Python实现微信小程序支付功能的完整攻略，希望对你有所帮助。