批量识别发票二维码图片自动下载pdf

# -*- coding:utf-8 -*-
import os
from pyzbar import pyzbar
from PIL import Image
import requests
from io import BytesIO
import time
import sys


def get_ewm(img_adds):
    """ 读取二维码的内容： img_adds：二维码地址（可以是网址也可是本地地址 """
    if os.path.isfile(img_adds):
        # 从本地加载二维码图片
        img = Image.open(img_adds)
    else:
        # 从网络下载并加载二维码图片
        rq_img = requests.get(img_adds).content
        img = Image.open(BytesIO(rq_img))

    # img.show()  # 显示图片，测试用

    txt_list = pyzbar.decode(img)

    for txt in txt_list:
        barcodeData = txt.data.decode("utf-8")

        return barcodeData


def getDownLoad(url, pdfDir, fName):
    '''打开网页下载文件'''
    r = requests.get(url)
    with open("%s\\%s.pdf" % (pdfDir, fName), "wb") as code:
        code.write(r.content)


def mkdir(path):
    '''新建文件夹'''
    folder = os.path.exists(path)

    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs(path)  # makedirs 创建文件时如果路径不存在会创建这个路径
    else:
        print("---  There is this folder!  ---")


print("初始化完成...\r\n获取路径")

gen_path = os.path.dirname(os.path.realpath(sys.argv[0]))
# 以上这句获取路径就正确了。

newDirStr = time.strftime("%Y-%m-%d%H%M%S", time.localtime())
print('生成文件夹名:\r\n%s' % newDirStr)
pdfDir = "%s\\%s\\" % (gen_path, newDirStr)
print('生成文件夹完整路径:\r\n%s' % pdfDir,'\r\n【准备就绪】\r\n',)
# files = os.listdir(os.getcwd)
errList = []#错误列表
urlList = []#ulr列表
cfList = []#重复列表
imgFiles = []#图片列表

files = os.listdir(gen_path)#当前目录下的所有文件，不包含子文件夹内的文件

if len(files) != 0:#如果当前目录下有文件
    mkdir(pdfDir)#则新建文件夹

n = 1
x = 0
for file in files:#遍历当前文件夹内所有文件
    if file[-3:] in 'jpg.png.gif':#如果文件是图片，则执行下面的代码，否则什么也不发生
        x += 1#图片数量计数
        url = get_ewm(file)#识别二维码，返回值为二维码的文本
        url1 = ''#解析后重新生成的发票下载地址
        if url == None:#如果解析的结果为空则为失败
            errList.append(file)
            print('【失败】   -',  url, file, '\r\n')
            continue
        elif url in urlList:#如果解析的结果已经出现过则为重复
            cfList.append('%s\t-->\t%s' % (file, imgFiles[urlList.index(url)]))
            print('【重复】   -', url[-39:], file, '\r\n')

            continue
        else:#否则判断为解析成功
            print('【成功】', '%3d' % n, url[-39:], file)
            n += 1#解析成功数量计数

        if not('getDoc' in url):#要确保解析的结果是可用的
            url1 = 'https://fp.hubei.chinatax.gov.cn:8088/api?action=getDoc&code=%s&type=12' % url[-39:]

        urlList.append(url)#登记url
        imgFiles.append(file)#登记图片文件

        getDownLoad(url1, pdfDir, url[-39:])#下载发票

# print(string)
if len(cfList) > 0:
    print('\r\n【重复警告】')
    print('=' * 100)
    print('第一次出现\t-->\t再次出现')
    print('\r\n'.join(cfList))
    print('=' * 30, '\r\n')

if len(errList) != 0:
    print('\r\n【未完成警告】')
    print('=' * 100)
    print('以下【%s】个图片未完成,请手动完成：\r\n%s' %
          (len(errList), '\r\n'.join(errList)))
    print('=' * 30, '\r\n')

print('总共【%s】个\r\n成功【%s】个\r\n失败【%s】个\r\n重复【%s】个' % (x, n-1, len(errList), len(cfList)))


input('按任意键打开发票文件夹')
os.startfile(pdfDir)

input('【结束】\r\n按任意键关闭')

生成EXE执行文件

用pyinstaller

pyinstall-F --add-data="resource;." -i "c.ico" qr.py

确保生成的EXE可以执行

然后pyinstall -F --add-data="resource;." -i "c.ico" qr.py

resource是当前目录下的文件夹，里面放的是import的第三方库

-i "c.ico" 是选参

点击下载我已经生成的EXE可执行文件