import barcode
from barcode.writer import ImageWriter
import traceback
import qrcode

# 更多了解：https://pypi.org/project/python-barcode/0.8.1/
# pip install python-barcode

# 查看 python-barcode 支持的条形码格式
# print(f"python-barcode支持的条形码格式：\n{barcode.PROVIDED_BARCODES}")

# 创建条形码格式对象，参数为支持的格式

def create_bar(message,filename):
    try:
        EAN = barcode.get_barcode_class('code39')

        # 创建条形码对象
        ean = EAN(message, writer=ImageWriter())
        filename= filename.replace('.png','')
        ean.save(f"/data/upload/mergefile/uploads/{filename}")
        return True
    except Exception as e:
        traceback.print_exc()
        return False


def create_qr(message,filename):
    try:
        qr = qrcode.QRCode(version=1, error_correction=qrcode.ERROR_CORRECT_M, box_size=10, border=5)

        qr.add_data(message)

        qr.make(fit=True)

        img = qr.make_image(fill_color='black', back_color='white')
        # img.show()
        img.save(f"/data/upload/mergefile/uploads/{filename}")
        return True
    except Exception as e:
        traceback.print_exc()
        return False