如何用python创建支付二维码

如何用Python创建支付二维码

使用Python创建支付二维码时，你可以利用二维码生成库如qrcode、整合支付API如微信支付或支付宝、设计并处理支付信息的格式。在本文中，我们将深入探讨如何实现这些步骤，并提供详细的代码示例。

一、二维码生成库介绍

1.1、选择合适的二维码生成库

Python中有多个库可以用来生成二维码，其中最常用的包括qrcode和Segno。这些库都能简便地生成静态二维码，但我们更推荐使用qrcode库，因为它有丰富的功能和较好的文档支持。

1.2、安装qrcode库

首先，你需要安装qrcode库，可以使用以下命令：

pip install qrcode[pil]

此外，你还需要安装Pillow库来处理图像：

pip install pillow

二、整合支付API

2.1、选择支付平台

常见的支付平台包括微信支付和支付宝。每个平台都有自己的API和文档。在本文中，我们将以微信支付为例，讲解如何整合支付API。

2.2、微信支付API的使用

首先，你需要注册微信支付商户并获取必要的API密钥和商户ID。然后，你可以通过微信支付官方提供的SDK或者直接调用API来生成支付链接。

import requests
import hashlib
import qrcode
from PIL import Image
def create_wechat_pay_qr(total_fee, out_trade_no, body, notify_url):
    # 微信支付统一下单API的URL
    url = "https://api.mch.weixin.qq.com/pay/unifiedorder"
    # 你的微信支付商户信息
    appid = 'your_appid'
    mch_id = 'your_mch_id'
    key = 'your_api_key'
    # 构造请求参数
    params = {
        'appid': appid,
        'mch_id': mch_id,
        'nonce_str': 'random_string',  # 随机字符串
        'body': body,
        'out_trade_no': out_trade_no,
        'total_fee': total_fee,
        'spbill_create_ip': 'your_ip',  # 服务器的IP地址
        'notify_url': notify_url,
        'trade_type': 'NATIVE',
    }
    # 生成签名
    def generate_sign(params, key):
        stringA = '&'.join([f"{k}={v}" for k, v in sorted(params.items())])
        stringSignTemp = f"{stringA}&key={key}"
        return hashlib.md5(stringSignTemp.encode('utf-8')).hexdigest().upper()
    params['sign'] = generate_sign(params, key)
    # 将参数转换为XML
    def dict_to_xml(params):
        xml = ["<xml>"]
        for k, v in params.items():
            xml.append(f"<{k}>{v}</{k}>")
        xml.append("</xml>")
        return ''.join(xml)
    xml_params = dict_to_xml(params)
    # 发送请求
    response = requests.post(url, data=xml_params)
    # 解析返回的XML
    import xml.etree.ElementTree as ET
    root = ET.fromstring(response.content)
    # 获取支付链接
    if root.find('return_code').text == 'SUCCESS' and root.find('result_code').text == 'SUCCESS':
        code_url = root.find('code_url').text
        return code_url
    else:
        raise Exception("WeChat Pay API request failed")
使用生成的支付链接创建二维码
code_url = create_wechat_pay_qr(100, 'order123456', 'Test Order', 'https://your_notify_url.com')
qr = qrcode.make(code_url)
qr.save("wechat_pay_qr.png")
image = Image.open("wechat_pay_qr.png")
image.show()

三、设计并处理支付信息的格式

3.1、定义支付信息格式

支付信息的格式在不同的支付平台之间可能有所不同，但基本上都包含支付金额、订单号、商品描述等信息。确保这些信息在生成二维码之前已经格式化并且符合API要求。

3.2、处理支付信息

你可以创建一个函数来处理和验证支付信息，确保所有必要的字段都已正确填写，并且数据符合API要求。

def validate_payment_info(total_fee, out_trade_no, body, notify_url):
    if not total_fee or not isinstance(total_fee, int):
        raise ValueError("Total fee must be a non-zero integer")
    if not out_trade_no or not isinstance(out_trade_no, str):
        raise ValueError("Out trade number must be a non-empty string")
    if not body or not isinstance(body, str):
        raise ValueError("Body must be a non-empty string")
    if not notify_url or not notify_url.startswith('http'):
        raise ValueError("Notify URL must be a valid URL starting with http or https")
    return True
示例调用
validate_payment_info(100, 'order123456', 'Test Order', 'https://your_notify_url.com')

四、测试和发布

4.1、单元测试

在开发过程中，编写单元测试来确保生成的二维码和支付链接是正确的，并且可以被扫描和使用。可以使用Python的unittest库来编写测试。

import unittest
class TestWeChatPayQR(unittest.TestCase):
    def test_generate_sign(self):
        params = {
            'appid': 'your_appid',
            'mch_id': 'your_mch_id',
            'nonce_str': 'random_string',
            'body': 'Test Order',
            'out_trade_no': 'order123456',
            'total_fee': 100,
            'spbill_create_ip': 'your_ip',
            'notify_url': 'https://your_notify_url.com',
            'trade_type': 'NATIVE',
        }
        key = 'your_api_key'
        sign = generate_sign(params, key)
        self.assertEqual(len(sign), 32)  # 签名长度为32位
    def test_create_wechat_pay_qr(self):
        code_url = create_wechat_pay_qr(100, 'order123456', 'Test Order', 'https://your_notify_url.com')
        self.assertTrue(code_url.startswith('weixin://'))
if __name__ == '__main__':
    unittest.main()