# -*- coding:utf-8 -*-
import os
import qrcode
import time
from PIL import Image
from pyzbar import pyzbar

"""
pip install -U pip
pip install Pillow
pip install pyzbar
pip install qrcode
"""


def make_qr_code_easy(content, save_path=None):
 """
 Generate QR Code by default
 :param content: The content encoded in QR Codeparams
 :param save_path: The path where the generated QR Code image will be saved in.
 If the path is not given the image will be opened by default.
 """
 img = qrcode.make(data=content)
 if save_path:
 img.save(save_path)
 else:
 img.show()


def make_qr_code(content, save_path=None):
 """
 Generate QR Code by given params
 :param content: The content encoded in QR Code
 :param save_path: The path where the generated QR Code image will be saved in.
 If the path is not given the image will be opened by default.
 """
 qr_code_maker = qrcode.QRCode(version=2,
 error_correction=qrcode.constants.ERROR_CORRECT_M,
 box_size=8,
 border=1,
 )
 qr_code_maker.add_data(data=content)
 qr_code_maker.make(fit=True)
 img = qr_code_maker.make_image(fill_color="black", back_color="white")
 if save_path:
 img.save(save_path)
 else:
 img.show()


def make_qr_code_with_icon(content, icon_path, save_path=None):
 """
 Generate QR Code with an icon in the center
 :param content: The content encoded in QR Code
 :param icon_path: The path of icon image
 :param save_path: The path where the generated QR Code image will be saved in.
 If the path is not given the image will be opened by default.
 :exception FileExistsError: If the given icon_path is not exist.
 This error will be raised.
 :return:
 """
 if not os.path.exists(icon_path):
 raise FileExistsError(icon_path)

 # First, generate an usual QR Code image
 qr_code_maker = qrcode.QRCode(version=4,
 error_correction=qrcode.constants.ERROR_CORRECT_H,
 box_size=8,
 border=1,
 )
 qr_code_maker.add_data(data=content)
 qr_code_maker.make(fit=True)
 qr_code_img = qr_code_maker.make_image(fill_color="black", back_color="white").convert('RGBA')

 # Second, load icon image and resize it
 icon_img = Image.open(icon_path)
 code_width, code_height = qr_code_img.size
 icon_img = icon_img.resize((code_width // 4, code_height // 4), Image.ANTIALIAS)

 # Last, add the icon to original QR Code
 qr_code_img.paste(icon_img, (code_width * 3 // 8, code_width * 3 // 8))

 if save_path:
 qr_code_img.save(save_path)
 else:
 qr_code_img.show()


def decode_qr_code(code_img_path):
 """
 Decode the given QR Code image, and return the content
 :param code_img_path: The path of QR Code image.
 :exception FileExistsError: If the given code_img_path is not exist.
 This error will be raised.
 :return: The list of decoded objects
 """
 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], scan_locations=True)


if __name__ == "__main__":

 # # 简易版
 # make_qr_code_easy("make_qr_code_easy", "make_qr_code_easy.png")
 # results = decode_qr_code("make_qr_code_easy.png")
 # if len(results):
 # print(results[0].data.decode("utf-8"))
 # else:
 # print("Can not recognize.")
 #
 # # 参数版
 # make_qr_code("make_qr_code", "make_qr_code.png")
 # results = decode_qr_code("make_qr_code.png")
 # if len(results):
 # print(results[0].data.decode("utf-8"))
 # else:
 # print("Can not recognize.")
 #
 # 带中间 logo 的
 # make_qr_code_with_icon("https://blog.52itstyle.vip", "icon.jpg", "make_qr_code_with_icon.png")
 # results = decode_qr_code("make_qr_code_with_icon.png")
 # if len(results):
 # print(results[0].data.decode("utf-8"))
 # else:
 # print("Can not recognize.")

 # 识别答题卡二维码 16 识别失败
 t1 = time.time()
 count = 0
 for i in range(1, 33):
 results = decode_qr_code(os.getcwd()+"\\img\\"+str(i)+".png")
 if len(results):
 print(results[0].data.decode("utf-8"))
 else:
 print("Can not recognize.")
 count += 1
 t2 = time.time()
 print("识别失败数量:" + str(count))
 print("测试时间:" + str(int(round(t2 * 1000))-int(round(t1 * 1000))))