python之qrcode生成二维码

发布时间 2023-08-01 17:13:13作者: 7dao

安装下列包

  pip install qrcode
  pip install pillow

代码实现

import qrcode
from PIL import Image
import os
import sys


def gen_qrcode(string, path, logo):
    """
    生成中间带logo的二维码
    需要安装qrcode, PIL库
    @参数 string: 二维码字符串
    @参数 path: 生成的二维码保存路径
    @参数 logo: logo文件路径
    @return: None
    """
    qr = qrcode.QRCode(
        # 25*25     二维码的版本号,每一个版本号对应一个尺寸,这里尺寸不是图片的大小而的是二维码长宽被分成的份数
        version=2,
        # 纠错容量,指二维码不完整时可以正常识别出原信息的概率(ERROR_CORRECT_H的纠错率最高)
        error_correction=qrcode.constants.ERROR_CORRECT_H,
        # 生成图片的像素
        box_size=8,
        # 二维码边框宽度
        border=1,
    )
    # string为想要打开的链接
    qr.add_data(string)
    # 用make()方法生成图片
    qr.make(fit=True)
    # 得到二维码对象,并可以通过修改fill_color、back_color参数来调整小格子颜色和背景色
    img = qr.make_image(fill_color='black', back_color='white')
    # 将图片转换为RGBA格式
    img = img.convert("RGBA")
    if logo and os.path.exists(logo):
        try:
            icon = Image.open(logo)
            # img_w、img_h是二维码的尺寸
            img_w, img_h = img.size
        except Exception as e:
            print(e)
            sys.exit(1)
        factor = 4
        size_w = int(img_w / factor)
        size_h = int(img_h / factor)

        # icon_W、icon_h是logo原始的尺寸
        icon_w, icon_h = icon.size
        # size_W、size_h是二维码尺寸的1/factor
        if icon_w > size_w:
            icon_w = size_w
        if icon_h > size_h:
            icon_h = size_h
        # antialias是平滑处理
        icon = icon.resize((icon_w, icon_h), Image.ANTIALIAS)
        # 保证二维码大小不超过二维码大小的1/factor

        # 计算logo在二维码中的相对位置
        w = int((img_w - icon_w) / 2)
        h = int((img_h - icon_h) / 2)
        icon = icon.convert("RGBA")
        # 根据相对位置w、h将logo放到二维码图片上,所以说实际是logo并不是二维码的一部分,会遮挡二维码的一部分,不能太大,否则无法识别
        img.paste(icon, (w, h), icon)
    img.save(path)
    # 调用系统命令打开图片
    # xdg - open(opens a file or URL in the user's preferred application)
    # os.system('xdg-open %s' %(path)) #这是Linux系统的命令
    # windows 下打开文件
    os.startfile(path)


if __name__ == "__main__":
    # 二维码的链接
    info = "www.baidu.com"
    # 生成的图片保存文件
    pic_path = "test.png"
    # logo的文件名
    logo_path = r"E:\ChanDaoSend\777.jpg"
    # 调用函数
    gen_qrcode(info, pic_path, logo_path)