389 lines
16 KiB
Python
389 lines
16 KiB
Python
#!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, move
|
|
from pyfiglet import Figlet
|
|
from colorama import init
|
|
from termcolor import colored
|
|
|
|
|
|
version = '2.5' # dev/devnocam
|
|
|
|
configdir = os.path.join(os.getenv('PROGRAMDATA'), 'IFPass')
|
|
config = os.path.join(configdir, '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.')
|
|
IFPassDBdir = input(r'Quel est le répertoire de la base de données IFPass ? (Ex : \\192.168.1.1\IFPass) : ')
|
|
printername = input("Quel est le nom de l'imprimante à cartes ? : ")
|
|
adobex86 = os.path.join(os.getenv('PROGRAMFILES(x86)'), 'Adobe', "Acrobat Reader DC", "Reader", "AcroRd32.exe")
|
|
adobex64 = os.path.join(os.getenv('PROGRAMFILES'), 'Adobe', "Acrobat Reader DC", "Reader", "AcroRd32.exe")
|
|
if os.path.exists(adobex86):
|
|
AcrobatReader = adobex86
|
|
elif os.path.exists(adobex64):
|
|
AcrobatReader = adobex64
|
|
else:
|
|
acrinstalled = yes_or_no("Acrobat Reader est nécessaire pour imprimer les cartes. Est il installé ?")
|
|
if acrinstalled:
|
|
AcrobatReader = input(r"Quel est le chemin vers Acrobat Reader ? ( Ex : C:\Program Files (x86)\Adobe\Acrobat Reader DC\Reader\AcroRd32.exe) : ")
|
|
else:
|
|
print(colored('\nInstallez Acrobat Reader, puis relancez IFPass.', 'red'))
|
|
os.system("pause")
|
|
sys.exit()
|
|
conf['DEFAULT'] = {'IFPassDBdir': IFPassDBdir, 'printername': printername, 'AcrobatReader': AcrobatReader}
|
|
os.makedirs(configdir, 0o777)
|
|
with open(config, 'w') as configfile:
|
|
conf.write(configfile)
|
|
else:
|
|
conf.read(config)
|
|
IFPassDBdir = conf['DEFAULT']['IFPassDBdir']
|
|
printername = conf['DEFAULT']['printername']
|
|
AcrobatReader = conf['DEFAULT']['AcrobatReader']
|
|
|
|
clientsfile = os.path.join(IFPassDBdir, 'Clients_IFPass.csv')
|
|
clientsbkpfile = os.path.join(IFPassDBdir, 'Clients_IFPass_backup.csv')
|
|
imgdir = os.path.join(IFPassDBdir, 'Cartes')
|
|
templatesdir = os.path.join(IFPassDBdir, 'Templates')
|
|
pdftemplate = os.path.join(templatesdir, 'Templates', 'IFPass_PDF_Template.pdf')
|
|
pngtemplate = os.path.join(templatesdir, 'Templates', 'IFPass_PNG_Template.png')
|
|
fonttemplate = os.path.join(templatesdir, 'Templates', 'Roboto-Bold.ttf')
|
|
|
|
if not os.path.exists(imgdir): # Cartes dir creation if it doesn't exist
|
|
os.makedirs(imgdir, 0o777)
|
|
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"])
|
|
if not os.path.exists(templatesdir):
|
|
move('Templates', IFPassDBdir)
|
|
|
|
return IFPassDBdir, printername, AcrobatReader, clientsfile, clientsbkpfile, imgdir, pdftemplate, pngtemplate, fonttemplate
|
|
|
|
|
|
def get_fullname():
|
|
while "Empty firstname":
|
|
firstname = input("Prénom : ").strip()
|
|
if len(firstname) == 0:
|
|
os.system('cls')
|
|
print(colored("\nLe Prénom ne peut pas être vide.", 'red'))
|
|
else:
|
|
break
|
|
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
|
|
while "Empty surname":
|
|
surname = input("Nom : ").upper().strip()
|
|
if len(surname) == 0:
|
|
os.system('cls')
|
|
print(colored("\nLe Nom ne peut pas être vide.", 'red'))
|
|
else:
|
|
break
|
|
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
|
|
cap.release()
|
|
print(colored('[OK]', 'green'))
|
|
cv2.destroyAllWindows()
|
|
defaultpicture = os.path.join(IFPassDBdir, 'Templates', 'default_avatar.jpg')
|
|
picture = Image.open(defaultpicture)
|
|
return picture
|
|
elif k & 0xFF == ord(' '): # Space
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
break
|
|
elif k & 0xFF == ord('q'): # Q or q to quit
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
sys.exit()
|
|
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']))
|
|
|
|
|
|
def main():
|
|
global titre, firstname, surname, fullname, dateinsc, dateexp, clientID, clientsfile, IFPassDBdir, clientsfile, imgdir, clientsbkpfile, pngtemplate, fonttemplate, picture, pdftemplate, printername, AcrobatReader
|
|
while "The program is running":
|
|
init() # Initialisation of colorama
|
|
IFPassDBdir, printername, AcrobatReader, clientsfile, clientsbkpfile, imgdir, pdftemplate, pngtemplate, fonttemplate = initialisation()
|
|
|
|
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 pour l'Institut Français en Hongrie.")
|
|
print('Pour toute question, problème ou requête contactez-moi à pro.ernst@gmail.com.\n')
|
|
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')
|
|
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':
|
|
global picture
|
|
picture = getpic()
|
|
clientID = getclientID()
|
|
barcode = barcode_gen(clientID)
|
|
fillcard(barcode)
|
|
cartefilename = mergepdf()
|
|
if version not in ('dev', 'devnocam'):
|
|
bkpdb()
|
|
printcard(cartefilename)
|
|
break
|
|
|
|
elif choix == '2':
|
|
os.system('cls')
|
|
membersearch()
|
|
os.system("pause")
|
|
|
|
elif choix == '3':
|
|
sys.exit()
|
|
|
|
else:
|
|
os.system('cls')
|
|
print(colored('Choix incorrect !', 'red', attrs=['bold']))
|