Ein kleiner CSV to QRcode generator

TQR.py

import csv
import multiprocessing
import argparse
import os.path
import sys
from multiprocessing import Pool
import logging
from PIL import Image, ImageDraw,ImageFont
import qrcode
import uuid
from pathlib import Path
from qrcode.image.styledpil import StyledPilImage
from qrcode.image.styles.moduledrawers import RoundedModuleDrawer
from qrcode.image.styles.colormasks import RadialGradiantColorMask
import re
from PIL import ImageColor

def makeQRImage(QRData):
    logging.info(f'start working on QRcode')
    img = Image.new('RGBA', (800, 1000), (255, 0, 0, 0))
    draw = ImageDraw.Draw(img)
    draw.rounded_rectangle([(50,50),(750,950)], outline =args.bcolor,width=15,radius=40)
    draw.rounded_rectangle([(50, 700), (750, 950)], outline=args.bcolor, fill=args.bcolor, width=15, radius=40)
    draw.rounded_rectangle([(50, 700), (750, 800)], outline=args.bcolor, fill=args.bcolor, width=15, radius=0)
    font = ImageFont.truetype("Gidole-Regular.ttf", size=args.fsize)
    w, h = draw.textsize(QRData[0],font=font)
    draw.text(((800 - w) / 2, 730), QRData[0], fill=args.fcolor,font=font)
    w, h = draw.textsize(args.scanText,font=font)
    draw.text(((800 - w) / 2, 830), args.scanText, fill=args.fcolor,font=font)
    qr = qrcode.QRCode(version=None, error_correction=qrcode.constants.ERROR_CORRECT_L, box_size=args.boxsize, border=0)
    qr.add_data(QRData[1])
    qr.make(fit=True)
    qrimg = qr.make_image(color_mask=RadialGradiantColorMask(edge_color=ImageColor.getcolor(args.bcolor, "RGB")), image_factory=StyledPilImage, module_drawer=RoundedModuleDrawer())
    qrimg = qrimg.convert("RGBA")
    datas = qrimg.getdata()
    newData = []
    for item in datas:
        if item[0] == 255 and item[1] == 255 and item[2] == 255:
            newData.append((255, 255, 255, 0))
        else:
            newData.append(item)
    qrimg.putdata(newData)
    qrimg_w, qrimg_h = qrimg.size
    offset = ((800 - qrimg_w) // 2, ((650 - qrimg_h) // 2)+50)
    img.paste(qrimg, offset)
    imguuid = uuid.uuid4()
    logging.info(f'done working on QRcode')
    logging.info(f'saving QRcode {QRData} as {imguuid}.png')
    Path("output").mkdir(parents=True, exist_ok=True)
    img.save(f"output/{imguuid}.png")

if __name__ == '__main__':
    multiprocessing.freeze_support()
    logging.basicConfig(format='%(asctime)s %(levelname)s: %(message)s', level=logging.DEBUG)
    parser = argparse.ArgumentParser(description='This tool makes QR codes from CSV')
    # add Args here
    parser.add_argument('--infile', type=argparse.FileType('r', encoding='UTF-8'), required=True)
    parser.add_argument('--workercount',type=int,default=multiprocessing.cpu_count())
    parser.add_argument('--bcolor', type=str, default="#0000ff")
    parser.add_argument('--fcolor', type=str, default="#ff0000")
    parser.add_argument('--boxsize', type=int, default=10, help="parameter controls how many pixels each box of the QR code is")
    parser.add_argument('--fsize', type=int, default=55)
    parser.add_argument('--scanText', type=str, default="Scan Me Baby!")
    args = parser.parse_args()
    match = re.search(r'^#(?:[0-9a-fA-F]{3}){1,2}$', args.bcolor)
    if match:
        logging.info("HTML color is valid")
    else:
        logging.error(f"color {args.bcolor} is not valid")
        sys.exit(1)
    match = re.search(r'^#(?:[0-9a-fA-F]{3}){1,2}$', args.fcolor)
    if match:
        logging.info("font HTML color is valid")
    else:
        logging.error(f"color {args.fcolor} is not valid")
        sys.exit(1)
    logging.info(f'start reading CSV file {args.infile.name}')
    reader = csv.reader(args.infile)
    data = list(reader)
    logging.info(f'done reading CSV file {args.infile.name} with {len(data)} lines')
    logging.info(f'init multiprocessing pool with {args.workercount} workers')
    with Pool(args.workercount) as p:
        p.map(makeQRImage, data)

currently hard coded to https://github.com/larsenwork/Gidole in the same folder