genattest.py

#!/usr/bin/env python3

from datetime import datetime
import getopt
import io
import os
import pdfrw
from reportlab.pdfgen import canvas
from reportlab.graphics.shapes import Drawing
from reportlab.graphics.barcode.qr import QrCodeWidget
from reportlab.graphics import renderPDF
import sys
import yaml

project_path = os.path.dirname(os.path.realpath(__file__))
#print(project_path)

# link with reason and reason position in form
dic_reason={
  "travail": 553,
  "achats": 482,
  "sante": 434,
  "famille": 410,
  "handicap": 373,
  "sport_animaux": 349,
  "convocation": 276,
  "missions": 252,
  "enfants": 228,
}

# get time
now = datetime.now()
date = now.strftime("%d/%m/%Y")
datefile = now.strftime("%d%m%Y")
hour = now.strftime("%H:%M")
hourh = now.strftime("%Hh%M")
hourall = now.strftime("%H%M%S")


# get id infos
settgins = {}
useDefault = True
if os.path.isfile(project_path + '/settings.yaml'):
    with open(project_path + "/settings.yaml", 'r') as stream:
        try:
            settings = yaml.safe_load(stream)
            useDefault = False
        except yaml.YAMLError as exc:
            print(exc)
if useDefault:
    settings = {
        "firstname": "Emmanuel",
        "lastname": "Macron",
        "birthday": "09/02/1972",
        "placeofbirth": "Paris",
        "address": "20 rue du Caire",
        "zipcode": 75015,
        "city": "Paris",
        "ccity": "Paris"
    }

# Transforme les clés du dictionnaire en variables
locals().update(settings)


# usage function
def usage():
    print("usage : python3 genattest.py -h -t -a -s")
    print("-h : help")
    print("-t : reason 'travail'")
    print("-a : reason 'achats'")
    print("-S : reason 'sante'")
    print("-s : reason 'sport & animaux' (to default)")
    print("-c : reason 'convocation'")
    print("-m : reason 'missions'")
    print("-e : reason 'enfants'")
    print("-f : reason 'famille'")
    print("-H : reason 'handicap'")


def run(argv):
    global reason
    reason="sport_animaux"
    try:
        opts, args = getopt.getopt(argv, "htaSscmefH", ["help", "travail",\
        "achats","sante","sport","convoc","missions","enfants","famille",\
        "handicap"])
    except getopt.GetoptError:
        usage()
        sys.exit(2)
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            usage()
            sys.exit()
        elif opt in ("-t", "--travail"):
            reason="travail"
        elif opt in ("-a", "--achats"):
            reason="achats"
        elif opt in ("-S", "--sante"):
            reason="sante"
        elif opt in ("-s", "--room"):
            reason="sport_animaux"
        elif opt in ("-c", "--convoc"):
            reason="convocation"
        elif opt in ("-m", "--missions"):
            reason="missions"
        elif opt in ("-e", "--enfants"):
            reason="enfants"
        elif opt in ("-f", "--famille"):
            reason="famille"
        elif opt in ("-H", "--handicap"):
            reason="handicap"

    canvas_data = get_overlay_canvas()
    form = merge(canvas_data, template_path=project_path + '/src/certificate.pdf')
    save(form, filename=project_path + '/merged_certificate.pdf')


def get_overlay_canvas() -> io.BytesIO:
    global zipcode
    zipcode = str(zipcode)

    data = io.BytesIO()
    pdf = canvas.Canvas(data)
    pdf.drawString(x=92, y=702, text=firstname+' '+lastname)
    pdf.drawString(x=92, y=684, text=birthday)
    pdf.drawString(x=214, y=684, text=placeofbirth)
    pdf.drawString(x=104, y=665, text=address+' '+zipcode+' '+city)
    pdf.drawString(x=47, y=dic_reason[reason], text='X')

    pdf.drawString(x=78, y=76, text=ccity)
    pdf.drawString(x=63, y=58, text=date)
    pdf.drawString(x=227, y=58, text=hour)

    #QRCode Zone
    text_QrCode="Cree le: "+date+" a "+hourh+";\n"\
    " Nom: "+lastname+";\n"\
    " Prenom: "+firstname+";\n"\
    " Naissance: "+birthday+" a "+placeofbirth+";\n"\
    " Adresse: "+address+" "+zipcode+" "+city+";\n"\
    " Sortie: "+date+" a "+hour+";\n"\
    " Motifs: "+reason

    qrw = QrCodeWidget(text_QrCode)
    b = qrw.getBounds()
    w=b[2]-b[0]
    h=b[3]-b[1]
    d = Drawing(92,92,transform=[100./w,0,0,100./h,0,0])
    d.add(qrw)
    renderPDF.draw(d, pdf, 435, 25)

    pdf.showPage()
    d = Drawing(300,300,transform=[340./w,0,0,340./h,0,0])
    d.add(qrw)
    renderPDF.draw(d, pdf, 50, 450)

    pdf.save()
    data.seek(0)
    return data



def merge(overlay_canvas: io.BytesIO, template_path: str) -> io.BytesIO:
    template_pdf = pdfrw.PdfReader(template_path)
    overlay_pdf = pdfrw.PdfReader(overlay_canvas)

    for page, data in zip(template_pdf.pages, overlay_pdf.pages):
        overlay = pdfrw.PageMerge().add(data)[0]
        pdfrw.PageMerge(page).add(overlay).render()

    form = io.BytesIO()
    pdfrw.PdfWriter().write(form, template_pdf)
    form.seek(0)
    return form


def save(form: io.BytesIO, filename: str):
    with open(filename, 'wb') as f:
        f.write(form.read())

if __name__ == '__main__':
    run(sys.argv[1:])