#!/usr/bin/python3 # Written by Jordan ERNST Q1 2018. # Contact : pro.ernst@gmail.com import configparser import sys import os import re from datetime import date, timedelta, datetime import csv import code128 import cv2 from pywinauto.findwindows import find_window from pywinauto.win32functions import SetForegroundWindow from PIL import Image, ImageDraw, ImageFont from PyPDF2 import PdfFileReader, PdfFileWriter import subprocess from shutil import copyfile from pyfiglet import Figlet from colorama import init from termcolor import colored version = '2.0' config = 'IFPass.conf' def initialisation(): conf = configparser.ConfigParser() conf.optionxform = str # For case sensitive config file if not os.path.exists(config): # Check if config file exists print('Fichier de configuration introuvable.') IFPassdir = input(r'Quel est le répertoire IFPass ? (Ex : \\192.168.1.1\IFPass) : ') printername = input("Quel est le nom de l'imprimante à cartes ? : " ) AcrobatReader = input(r"Quel est le chemain vers Acrobat Reader ? ( Ex : C:\Program Files (x86)\Adobe\Acrobat Reader DC\Reader\AcroRd32.exe) : ") conf['DEFAULT'] = {'IFPassdir': IFPassdir, 'printername': printername, 'AcrobatReader': AcrobatReader} with open(config, 'w') as configfile: conf.write(configfile) else: conf.read(config) IFPassdir = conf['DEFAULT']['IFPassdir'] printername = conf['DEFAULT']['printername'] AcrobatReader = conf['DEFAULT']['AcrobatReader'] clientsfile = os.path.join(IFPassdir, 'Clients_IFPass.csv') clientsbkpfile = os.path.join(IFPassdir, 'Clients_IFPass_backup.csv') imgdir = os.path.join(IFPassdir, 'Cartes') pdftemplate = os.path.join(IFPassdir, 'Templates', 'IFPass_PDF_Template.pdf') pngtemplate = os.path.join(IFPassdir, 'Templates', 'IFPass_PNG_Template.png') fonttemplate = os.path.join(IFPassdir, 'Templates', 'Roboto-Bold.ttf') if not os.path.exists(imgdir): # Cartes dir creation if it doesn't exist os.makedirs(imgdir) if not os.path.exists(clientsfile): # Creation Clients_File if it doesn't exist with open(clientsfile, 'w', newline='', encoding='utf-8') as csvfile: writer = csv.writer(csvfile, delimiter=';') writer.writerow(['Titre', 'Prénom', 'Nom', 'Numéro de client', "Date d'inscription", "Date d'expiration"]) return IFPassdir, printername, AcrobatReader, clientsfile, clientsbkpfile, imgdir, pdftemplate, pngtemplate, fonttemplate def get_fullname(): firstname = input("Prénom : ").strip() firstname = firstname[0].upper() + firstname[1:].lower() if '-' in firstname: pos = firstname.find("-") firstname = firstname[:pos + 1] + firstname[pos + 1].upper() + \ firstname[pos + 2:] # Check if compound name if ' ' in firstname: pos = firstname.find(" ") firstname = firstname[:pos + 1] + firstname[pos + 1].upper() + \ firstname[pos + 2:] # Check if compound name surname = input("Nom : ").upper().strip() docteur = yes_or_no('Est-ce un Docteur ?') if docteur: titre = 'Dr.' fullname = titre + ' ' + surname + ' ' + firstname else: titre = '' fullname = surname + ' ' + firstname return titre, firstname, surname, fullname def yes_or_no(question): while "the answer is invalid": reply = input(question + " O/N : ").lower().strip() if reply[:1] in ['oui', 'ou', 'o']: return True if reply[:1] in ['non', 'no', 'n']: return False def getclientID(): with open(clientsfile, 'r', newline='', encoding='utf-8') as csvfile: lastline = csvfile.readlines()[-1] lastID = lastline.split(';')[3] if lastID == "Numéro de client": clientID = "0000000001" else: clientID = str(int(lastID) + 1).zfill(10) writeindb(clientID) return clientID def barcode_gen(clientID): print("Génération du code barre... ", end='') barcode = code128.image(clientID, thickness=4) # .save(imgdir + clientID + '.png') print(colored('[OK]', 'green')) return barcode def getpic(): while True: print("Prendre la photo... ", end='') sys.stdout.flush() cap = cv2.VideoCapture(0) while True: # Loop stream webcam try: ret, frame = cap.read() cv2.rectangle(frame, (170, 73), (470, 407), (0, 255, 0), 2) cv2.imshow('IFCamera - Touche Espace pour prendre la photo, Echap pour une carte sans photo, Q pour quitter.', frame) except cv2.error: print(colored('\nLa webcam est débranchée. Branchez-la, puis relancez le programme.', 'red')) os.system("pause") sys.exit() SetForegroundWindow(find_window(title='IFCamera - Touche Espace ' 'pour prendre la photo, Echap pour une carte sans photo, Q pour quitter.')) k = cv2.waitKey(1) if k == 27: # Echap print(colored('[OK]', 'green')) cv2.destroyAllWindows() defaultpicture = os.path.join(IFPassdir, 'Templates', 'default_avatar.jpg') picture = Image.open(defaultpicture) return picture elif k & 0xFF == ord(' '): # Space break elif k & 0xFF == ord('q'): # Q or q to quit cap.release() cv2.destroyAllWindows() sys.exit() cap.release() cv2.destroyAllWindows() cropped = frame[75:405, 172:468] cv2.imshow('IFCamera - Espace pour valider, Echap pour modifier, Q pour quitter', cropped) while True: k = cv2.waitKey(1) if k == 27: # Echap print(colored('[KO]', 'red')) cv2.destroyAllWindows() break elif k & 0xFF == ord(' '): # Space print(colored('[OK]', 'green')) cv2.destroyAllWindows() # Color conversion and cv2 img to Pillow img cropped = cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB) picture = Image.fromarray(cropped) return picture elif k & 0xFF == ord('q'): # Q or q to quit cv2.destroyAllWindows() sys.exit() def writeindb(clientID): print("Ajout dans la base de données... ", end="") with open(clientsfile, 'a', newline='', encoding='utf-8') as csvfile: writer = csv.writer(csvfile, delimiter=';') writer.writerow([titre, firstname, surname, clientID, dateinsc, dateexp]) print(colored('[OK]', 'green')) def bkpdb(): copyfile(clientsfile, clientsbkpfile) def fillcard(barcode): print("Création de la carte avec les informations...", end='') im = Image.open(pngtemplate) draw = ImageDraw.Draw(im) # Name embedding : font = ImageFont.truetype(fonttemplate, 45) draw.text((401, 310), fullname, fill=(0,0,0), font=font) # Date embedding : font = ImageFont.truetype(fonttemplate, 30) draw.text((401, 390), dateexp, fill=(0,0,0), font=font) # ID embedding : font = ImageFont.truetype('arial.ttf', 30) draw.text((693, 560), clientID, fill=(0,0,0), font=font) # Barcode + picture embedding : im.paste(barcode,(556, 460)) if version != 'devnocam': im.paste(picture,(47, 49)) # Create PDF : im = im.convert("RGB") im.save(imgdir + clientID + '_Front.pdf', 'PDF', resolution=299.0, quality=98) print(colored('[OK]', 'green')) def mergepdf(): cartefilename = os.path.join(imgdir, clientID + '.pdf') output = PdfFileWriter() pdf1 = PdfFileReader(imgdir + clientID + '_Front.pdf', 'rb') recto = pdf1.getPage(0) output.addPage(recto) pdf2 = PdfFileReader(pdftemplate, 'rb') verso = pdf2.getPage(1) output.addPage(verso) with open(cartefilename, 'wb') as f: output.write(f) os.remove(imgdir + clientID + '_Front.pdf') return cartefilename def printcard(cartefilename): # Working : subprocess.Popen('"C:\Program Files (x86)\Adobe\Acrobat Reader DC\Reader\AcroRd32.exe" /h /n /t ' + cartefilename + ' '+ printername, shell=False) subprocess.Popen('"' + AcrobatReader + '"' + ' /h /n /t ' + cartefilename + ' '+ printername, shell=False) def membersearch(): with open(clientsfile, 'r', newline='', encoding='utf-8') as csvfile: reader = csv.reader(csvfile, delimiter=';') csvlist = list(map(tuple, reader)) del csvlist[0] # We dele the first line (Prénom, Nom...) research = input('Entrez une partie du nom, prénom, ou numéro de carte (ou flashez) : ').lower() os.system('cls') results = [] for member in csvlist: resfirstname = member[1].lower() ressurname = member[2].lower() resnumber = member[3] if any(research in data for data in [resfirstname, ressurname, resnumber]): results.append(member) if results: if len(results) == 1: member = results[0] else: print('-' * 56) print(f'{"Choix":8} {"Prénom":15} {"Nom":15} {"Numéro de carte":15}') print('-' * 56) for index, result in enumerate(results, start=1): print(f'{index:^8} {result[1]:15} {result[2]:15} {result[3]:^15}') del member while 'The choice is not valid': try: memberchoice = input("De quel membre s'agit il ? (Colonne Choix) : ") member = results[int(memberchoice) - 1] break except (IndexError, ValueError): print(colored('Choix invalide ! Veillez bien à sélectionner le numéro de la colonne "Choix"', 'red', attrs=['bold'])) os.system('cls') print("Titre : ", colored(member[0], 'green')) print("Prénom : ", colored(member[1], 'green')) print("Nom : ", colored(member[2], 'green')) print("Numéro de carte : ", colored(member[3], 'green')) print("Date d'inscription :", member[4]) print("Date d'expiration : ", member[5]) dateexp = datetime.strptime(member[5], '%d/%m/%Y').date() diff = (dateexp - date.today()).days if diff > 0: print(colored(f"L'abonnement est encore valable {diff} jours.", 'green', attrs=['bold'])) elif diff < 0: print(colored("L'abonnement est expiré depuis {abs(diff)} jours.", 'red', attrs=['bold'])) # abs() to remove minus sign elif diff == 0: print(colored("Il s'agit du dernier jour de l'abonnement, il expirera demain.", 'yellow', attrs=['bold'])) else: print(colored("Aucun membre n'a été trouvé.", 'red', attrs=['bold'])) while "The program is running": init() # Initialisation of colorama IFPassdir, printername, AcrobatReader, clientsfile, clientsbkpfile, imgdir, pdftemplate, pngtemplate, fonttemplate = initialisation() os.system('cls') print('1 - Nouveau membre', '2 - Rechercher un membre', '3 - Quitter', sep='\n') choix = input('Choix : ') if choix == '1': while "the informations are incorrect": # Loop Filling informations os.system('cls') f = Figlet(font='big') print(colored(f.renderText('IFPass'), 'cyan', attrs=['bold'])) print('Version : ', version) if version in ('dev', 'devnocam'): print(colored("\nATTENTION : Il s'agit d'une version en cours de développement, potentiellement instable !", 'red')) print('\nLe programme IFPass à été écrit par Jordan ERNST Q1 2018.') print('Pour toute question ou problème contactez-moi à pro.ernst@gmail.com.\n') titre, firstname, surname, fullname = get_fullname() dateinsc = date.today() dateexp = dateinsc + timedelta(days=365) dateinsc = dateinsc.strftime('%d/%m/%Y') dateexp = dateexp.strftime('%d/%m/%Y') changeexp = yes_or_no("Voulez-vous choisir la date d'expiration ?") if changeexp: while True: dateexp = input("Quelle date d'expiration voulez-vous mettre (Format : JJ/MM/AAAA)? : ") match = re.fullmatch(r'^(0[1-9]|1[0-9]|2[0-9]|3[0-1])/(0[1-9]|1[0-2])/([0-9]){4}$', dateexp) if match: break else: print('Mauvais format ! JJ/MM/AAAA, exemple : 01/08/2042') os.system('cls') print("Titre : ", colored(titre, 'green')) print("Prénom : ", colored(firstname, 'green')) print("Nom : ", colored(surname, 'green')) print("Date d'inscription :", colored(dateinsc, 'green')) print("Date d'expiration : ", colored(dateexp, 'green')) correct = yes_or_no("Ces informations sont elles correctes ?") if correct: os.system('cls') if version != 'devnocam': picture = getpic() clientID = getclientID() barcode = barcode_gen(clientID) fillcard(barcode) cartefilename = mergepdf() if version not in ('dev', 'devnocam'): bkpdb() printcard(cartefilename) elif choix == '2': os.system('cls') membersearch() elif choix == '3': sys.exit() else: os.system('cls') print(colored('Choix incorrect !', 'red', attrs=['bold'])) os.system("pause")