Python-二维码处理（qrcodepyzbar）

Posted 2022-12-26 梦之海岛

文章目录

• 前言
• 一、第三方库
• 二、算法
• • [1]. 生成
  • [2]. 单个识别
  • [3]. 批量识别
• 三、示例
• • [1]. 生成
  • [2]. 单个识别
  • [3]. 批量识别
• 四、代码

前言

在生成二维码或文本文件的时候需要结束程序
脚本的缺点还是挺多的，有空的大佬可以自己再改改

qrcode · PyPI

Python二维码应用（一）QRcode二维码生成&识别

DealQRcode 小狐狸FM

一、第三方库

Pillow
pyzbar
qrcode

二、算法

[1]. 生成

• 生成二维码时，直接使用qrcode.make()，传入一个文本
  返回一个对象后，使用show()方法显示二维码，save()保存二维码
• qrcode · PyPI

import os
import qrcode
from PIL import Image
from pyzbar import pyzbar

data_path = "data/qr.jpg"  #生成单个二维码时保存的路径
txt_path = "qr.txt" #批量扫描二维码时，保存的文本路径

def Make_QR(text):
    '''生成二维码'''
    img = qrcode.make(text)  # 生成
    # img.show() #显示
    img.save(data_path)  # 保存

[2]. 单个识别

• 先看一下大佬代码中有关二维码识别的部分，其他的内容不需要去了解
  在主函数中调用了decode_qr_code函数，这个函数中先是判断了二维码的路径是否存在
  然后直接调用pyzbar和Image第三方库的函数，返回了二维码的数据后，再解码获取内容
• Python二维码应用（一）QRcode二维码生成&识别

# 一堆代码
def decode_qr_code(code_img_path):
    if not os.path.exists(code_img_path):
        raise FileExistsError(code_img_path)

    # Here, set only recognize QR Code and ignore other type of code
    return pyzbar.decode(Image.open(code_img_path), symbols=[pyzbar.ZBarSymbol.QRCODE])
    
if __name__ == "__main__": 
   	# 一堆代码 
    results = decode_qr_code("qrcode.png")
    print("2、正在解码：")
    if len(results):
        print("解码结果是：")
        print(results[0].data.decode("utf-8"))
    else:
        print("Can not recognize.")

函数中比较关键的语句就两处，函数的返回值和主函数的输出

return pyzbar.decode(Image.open(code_img_path), symbols=[pyzbar.ZBarSymbol.QRCODE])
print(results[0].data.decode("utf-8"))

我们可以自己测试一下，把重要的内容拼接起来
删除部分代码并测试结果，有些代码比较长时，是可以进行缩减的

import os
import qrcode
from PIL import Image
from pyzbar import pyzbar
if __name__ == '__main__': #主函数
    print(pyzbar.decode(Image.open("data/qr.jpg"), symbols=[pyzbar.ZBarSymbol.QRCODE])[0].data.decode("utf-8")) #输出结果
    print(pyzbar.decode(Image.open("data/qr.jpg"))[0].data.decode("utf-8")) #输出结果
    print(pyzbar.decode(Image.open("data/qr.jpg"))[0].data)
    print(pyzbar.decode(Image.open("data/qr.jpg"))[0])
    print(pyzbar.decode(Image.open("data/qr.jpg")))

得到的结果如下，??yes是二维码保存的内容

??yes
??yes
b'??yes'
Decoded(data=b'??yes', type='QRCODE', rect=Rect(left=40, top=40, width=210, height=210), polygon=[Point(x=40, y=40), Point(x=40, y=250), Point(x=250, y=250), Point(x=250, y=40)])
[Decoded(data=b'??yes', type='QRCODE', rect=Rect(left=40, top=40, width=210, height=210), polygon=[Point(x=40, y=40), Point(x=40, y=250), Point(x=250, y=250), Point(x=250, y=40)])]

经过测试，如果要获取二维码识别的结果，用下方的一行代码就行，"data/qr.jpg"为二维码路径

pyzbar.decode(Image.open("data/qr.jpg"))[0].data.decode("utf-8")

弄成函数方便之后的批量识别

def Read_QR(qr_path):
    '''识别单个二维码'''
    return pyzbar.decode(Image.open(qr_path))[0].data.decode("utf-8")

[3]. 批量识别

• 批量识别时，先通过os库获取某个文件夹下的所有文件名，然后再调用Read_QR对所有文件进行读取。
• 读取的结果是字符串类型，对结果添加保存的路径和换行符后，保存到一个列表中
• 最后使用join()方法把列表的内容一次性存入文件（其实也可以删除上一行的\\n添加到"\\n".join()中）

def Read_All(folder_path):
    '''批量识别二维码'''
    list_path = os.listdir(folder_path) #保存该文件夹下所有图片的路径
    list_result = [] #所有图片的识别结果
    for path in list_path: #遍历所有图片
        print(f"图片path的识别结果:",Read_QR(folder_path + "/" + path))
        list_result.append(f"图片path的识别结果:" + Read_QR(folder_path + "/" + path) + "\\n")
    with open(txt_path,"w") as fp: #文件写入
        fp.write("".join(list_result))

三、示例

[1]. 生成

[2]. 单个识别

[3]. 批量识别

四、代码

# -*- coding: utf-8 -*-
# 作者：小狐狸FM
# 功能：二维码处理
# 缺点：二维码内容为中文时会出现乱码，批量扫描时文件夹内文件都应该是二维码文件
import os
import qrcode
from PIL import Image
from pyzbar import pyzbar

data_path = "data/qr.jpg"  #生成单个二维码时保存的路径
txt_path = "qr.txt" #批量扫描二维码时，保存的文本路径

def Make_QR(text):
    '''生成二维码'''
    img = qrcode.make(text)  # 生成
    # img.show() #显示
    img.save(data_path)  # 保存

def Read_QR(qr_path):
    '''识别单个二维码'''
    return pyzbar.decode(Image.open(qr_path))[0].data.decode("utf-8")

def Read_All(folder_path):
    '''批量识别二维码'''
    list_path = os.listdir(folder_path) #保存该文件夹下所有图片的路径
    list_result = [] #所有图片的识别结果
    for path in list_path: #遍历所有图片
        print(f"图片path的识别结果:",Read_QR(folder_path + "/" + path))
        list_result.append(f"图片path的识别结果:" + Read_QR(folder_path + "/" + path) + "\\n")
    with open(txt_path,"w") as fp: #文件写入
        fp.write("".join(list_result))

if __name__ == '__main__': #主函数

    while True:
        print("-----------------------------")
        print("|     0. 退出                |")
        print("|     1. 生成二维码           |")
        print("|     2. 扫描二维码           |")
        print("|     3. 批量扫描二维码        |")
        print("-----------------------------")
        flag = int(input("请选择: "))
        if flag==1:
            text = input("二维码内容：")
            Make_QR(text) #生成二维码
        elif flag==2:
            qr_path = input("二维码路径: ")
            if os.path.exists(qr_path)==False: #文件不存在时
                print("文件不存在!")
                continue
            print("识别结果：",Read_QR(qr_path)) #识别
        elif flag==3:
            folder_path = input("二维码所在文件夹: ")
            if os.path.exists(folder_path)==False: #文件夹不存在时
                print("文件夹不存在!")
                continue
            Read_All(folder_path)
        else:
            break