mirror of
https://gitlab.os-k.eu/neox/CNIRevelator.git
synced 2023-08-25 14:03:10 +02:00
Main files of the project
This commit is contained in:
NeoXgroupEU
GitHub
parent
ccf4adc566
commit
6a5a01c117
50
CNI_GLOBALVAR.py
Normal file
50
CNI_GLOBALVAR.py
Normal file
@ -0,0 +1,50 @@
|
||||
"""
|
||||
********************************************************************************
|
||||
*** Projet CNI_Revelator ***
|
||||
|
||||
GNU GPL * 07/2018
|
||||
|
||||
Adrien Bourmault
|
||||
|
||||
VARIABLES
|
||||
|
||||
********************************************************************************
|
||||
"""
|
||||
CST_REV = "8"
|
||||
CST_VERTITLE = "2.2"
|
||||
CST_TAB_VER = ["2","2","1"]
|
||||
CST_VER = "{0}.{1}.{2}".format(CST_TAB_VER[0], CST_TAB_VER[1], CST_TAB_VER[2])
|
||||
CST_TYPE = "Final Release"
|
||||
CST_NAME = "CNIRevelator"
|
||||
CST_TITLE = CST_NAME + " " + CST_VER + " - GNU/GPL Licensing 2018"
|
||||
CST_MARK = CST_NAME + " " + CST_TYPE + " " + CST_VER + " - by NeoX, GNU/GPL Licensing 2018"
|
||||
CST_SUM_VER = int(CST_TAB_VER[0])*100 + int(CST_TAB_VER[1])*10 + int(CST_TAB_VER[2])
|
||||
CST_LINK = <lien vers serveur>
|
||||
CST_COLOR = "#003380"
|
||||
|
||||
|
||||
import base64
|
||||
import hashlib
|
||||
from Crypto import Random
|
||||
from Crypto.Cipher import AES
|
||||
from tkinter import *
|
||||
from tkinter.messagebox import *
|
||||
from tkinter import filedialog
|
||||
from tkinter import ttk as ttk
|
||||
import os
|
||||
import time
|
||||
import threading
|
||||
import sys
|
||||
import urllib.request as urllib2
|
||||
import urllib.error as URLExcept
|
||||
import random
|
||||
from pypac import PACSession
|
||||
from requests.auth import HTTPProxyAuth
|
||||
import subprocess
|
||||
from datetime import datetime
|
||||
from PIL import Image, ImageFont, ImageDraw, ImageTk, ImageEnhance, ImageFilter
|
||||
import math
|
||||
import warnings
|
||||
import string
|
||||
|
||||
CST_FOLDER = os.getenv('APPDATA') + "/CNIRevelator/"
|
||||
131
CNI_Revelator.py
Normal file
131
CNI_Revelator.py
Normal file
@ -0,0 +1,131 @@
|
||||
"""
|
||||
********************************************************************************
|
||||
*** Projet CNI_Revelator ***
|
||||
|
||||
GNU GPL * 07/2018
|
||||
|
||||
Adrien Bourmault
|
||||
|
||||
main
|
||||
|
||||
********************************************************************************
|
||||
"""
|
||||
|
||||
###IMPORTS GLOBAUX
|
||||
from CNI_GLOBALVAR import *
|
||||
|
||||
|
||||
##LOGGING
|
||||
|
||||
try:
|
||||
os.remove("error.log")
|
||||
os.remove("conf.ig")
|
||||
except:
|
||||
pass
|
||||
import logging
|
||||
|
||||
CST_NIVEAU_LOG = logging.ERROR
|
||||
|
||||
from logging.handlers import RotatingFileHandler
|
||||
|
||||
# création de l'objet logger qui va nous servir à écrire dans les logs
|
||||
logger = logging.getLogger()
|
||||
# on met le niveau du logger à DEBUG, comme ça il écrit tout
|
||||
logger.setLevel(CST_NIVEAU_LOG)
|
||||
|
||||
# création d'un formateur qui va ajouter le temps, le niveau
|
||||
# de chaque message quand on écrira un message dans le log
|
||||
formatter = logging.Formatter('%(asctime)s :: %(levelname)s :: %(message)s')
|
||||
# création d'un handler qui va rediriger une écriture du log vers
|
||||
# un fichier en mode 'append', avec 1 backup et une taille max de 1Mo
|
||||
file_handler = RotatingFileHandler('error.log', 'a', 1000000, 1)
|
||||
# on lui met le niveau sur DEBUG, on lui dit qu'il doit utiliser le formateur
|
||||
# créé précédement et on ajoute ce handler au logger
|
||||
file_handler.setLevel(CST_NIVEAU_LOG)
|
||||
file_handler.setFormatter(formatter)
|
||||
logger.addHandler(file_handler)
|
||||
|
||||
##History
|
||||
|
||||
|
||||
|
||||
### IMPORTS LOCAUX
|
||||
from CNI_classes import *
|
||||
from CNI_Update import *
|
||||
|
||||
### FONCTION PRINCIPALE
|
||||
|
||||
|
||||
def main(logger):
|
||||
|
||||
logger.info("main() : " +"**** Creating App_main() ****")
|
||||
main_w = App_main(logger)
|
||||
|
||||
main_w.montext("* " + CST_NAME + " " + CST_VER + " " + CST_TYPE + " Revision " + CST_REV + " *\n")
|
||||
|
||||
import CNI_pytesseract as pytesseract
|
||||
|
||||
try:
|
||||
|
||||
os.environ["PATH"] = CST_FOLDER + "Tesseract-OCR4\\"
|
||||
os.environ["TESSDATA_PREFIX"] = CST_FOLDER + "Tesseract-OCR4\\tessdata"
|
||||
tesser_version = pytesseract.get_tesseract_version()
|
||||
|
||||
except Exception as e:
|
||||
logger.error("main() : " +"**** ERROR WITH TESSERACT MODULE " + str(e) + " ****")
|
||||
else:
|
||||
text = "Tesseract version " + str(tesser_version) +" Licensed Apache 2004 successfully initiated\n"
|
||||
main_w.montext(text)
|
||||
|
||||
|
||||
main_w.montext("\n\nEntrez la première ligne de MRZ svp \n")
|
||||
|
||||
|
||||
|
||||
logger.info("main() : " +"**** Launching App_main() ****")
|
||||
main_w.mainloop()
|
||||
logger.info("main() : " +"**** Ending App_main() ****")
|
||||
|
||||
|
||||
|
||||
##Launcher
|
||||
|
||||
logger.info("launcher : " + CST_NAME +" "+ CST_VER)
|
||||
logger.info("launcher : " +"*****Hello World*****")
|
||||
|
||||
logger.info("launcher : " +"*****Launching SoftUpdate()*****")
|
||||
try:
|
||||
Answer = SoftUpdate(logger)
|
||||
except Exception as e:
|
||||
logger.info("launcher : " +"*****FATAL ERROR*****" + str(e))
|
||||
os.abort()
|
||||
logger.info("launcher : " +"*****Ending SoftUpdate()*****")
|
||||
|
||||
|
||||
try:
|
||||
if Answer == True:
|
||||
logger.info("launcher : " +"*****Launching main()*****")
|
||||
State = main(logger)
|
||||
except Exception as e:
|
||||
logger.info("launcher : " +"*****FATAL ERROR*****" + str(e))
|
||||
os.abort()
|
||||
|
||||
logger.info("launcher : " +"*****Ending main()*****")
|
||||
logger.info("launcher : " +"*****Goodbye!*****")
|
||||
|
||||
handlers = logger.handlers[:]
|
||||
for handler in handlers:
|
||||
handler.close()
|
||||
logger.removeHandler(handler)
|
||||
|
||||
|
||||
if os.path.getsize("error.log") == 0:
|
||||
try:
|
||||
os.remove("error.log")
|
||||
except:
|
||||
raise(OSError)
|
||||
os.abort()
|
||||
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
568
CNI_Update.py
Normal file
568
CNI_Update.py
Normal file
@ -0,0 +1,568 @@
|
||||
"""
|
||||
********************************************************************************
|
||||
*** Projet CNI_Revelator ***
|
||||
|
||||
GNU GPL * 07/2018
|
||||
|
||||
Adrien Bourmault
|
||||
|
||||
UPDATE
|
||||
|
||||
********************************************************************************
|
||||
"""
|
||||
|
||||
|
||||
###IMPORTS GLOBAUX
|
||||
from CNI_GLOBALVAR import *
|
||||
|
||||
|
||||
###IMPORTS LOCAUX
|
||||
from CNI_classes import *
|
||||
|
||||
|
||||
###BEGIN
|
||||
|
||||
def SoftUpdate(logger): #Fonction de mise à jour de l'application
|
||||
|
||||
import zipfile
|
||||
# zip_ref = zipfile.ZipFile(path_to_zip_file, 'r')
|
||||
# zip_ref.extractall(directory_to_extract_to)
|
||||
# zip_ref.close()
|
||||
|
||||
logger.info("SoftUpdate() : " +"Verifying local data dir...")
|
||||
if not os.path.exists(CST_FOLDER):
|
||||
os.makedirs(CST_FOLDER)
|
||||
logger.info("SoftUpdate() : " +"Data dir created !")
|
||||
|
||||
logger.info("SoftUpdate() : " +"Looking for older version in dir...")
|
||||
list = os.listdir('.')
|
||||
|
||||
for file in list:
|
||||
|
||||
if file.startswith("CNIRevelator_"):
|
||||
|
||||
temp = ["0","0","0"]
|
||||
ver = file[13:].split(".")
|
||||
|
||||
|
||||
for i in range(len(ver)):
|
||||
|
||||
|
||||
|
||||
if ver[i] != "exe" :
|
||||
try:
|
||||
temp[i] = ver[i]
|
||||
except:
|
||||
None
|
||||
|
||||
|
||||
ver = temp.copy()
|
||||
|
||||
try :
|
||||
sum_ver = int(ver[0])*100 + int(ver[1])*10 + int(ver[2])
|
||||
|
||||
if sum_ver < CST_SUM_VER:
|
||||
|
||||
os.remove(file)
|
||||
logger.info("SoftUpdate() : " + "Removed old version : " + str(file) )
|
||||
|
||||
except:
|
||||
|
||||
logger.error("SoftUpdate() : " +"Failing to remove old version : " + str(file) )
|
||||
|
||||
None
|
||||
|
||||
|
||||
def updating(): #Fonction de lancement du thread de maj
|
||||
|
||||
def updator(): #Fonction d'exécution de la maj
|
||||
|
||||
logger.info("[updator() thread] : " + "Welcome !")
|
||||
global ret
|
||||
ret = 11
|
||||
canvas.itemconfigure(message, text="Recherche de mises-à-jour...")
|
||||
p.configure(mode = "indeterminate", value = 0, maximum = 20)
|
||||
p.start()
|
||||
upwin.update()
|
||||
logger.info("[updator() thread] : " + "Looking for updates...")
|
||||
|
||||
try:#Essai par proxy
|
||||
|
||||
def download(url, filename):
|
||||
|
||||
try: # if identifiants disponibles dans un fichier
|
||||
|
||||
logger.info("[download() thread] : " + "Trying getting credentials in the config file")
|
||||
|
||||
with open(CST_FOLDER +'conf.ig', "rb") as config:
|
||||
|
||||
AESObj = AESCipher("<clé>")
|
||||
IPN, IPN_PASS = AESObj.decrypt(config.read()).split("||")[0:2]
|
||||
logger.info("[download() thread] : " + "Got credentials !")
|
||||
|
||||
session = PACSession(proxy_auth=HTTPProxyAuth(IPN, IPN_PASS))
|
||||
logger.info("[download() thread] : " + "Authenticated to proxy successfully")
|
||||
|
||||
|
||||
except IOError as e: #Else : on vérifie les identifiants de proxy
|
||||
|
||||
logger.error("[download() thread] : " + "False or absent credentials in the config file : " + str(e))
|
||||
|
||||
NoConnect = True
|
||||
|
||||
while NoConnect:
|
||||
|
||||
class LoginDialog(Toplevel):
|
||||
global login
|
||||
global key
|
||||
|
||||
def __init__(self, parent):
|
||||
super().__init__(parent)
|
||||
self.title("Connexion")
|
||||
# ---------------------------------------
|
||||
Label(self, text="IPN : ").pack()
|
||||
self.entry_login = Entry(self)
|
||||
self.entry_login.insert(0, "")
|
||||
self.entry_login.pack()
|
||||
# ---------------------------------------
|
||||
Label(self, text="Mot de passe : ").pack()
|
||||
self.entry_pass = Entry(self, show='*')
|
||||
self.entry_pass.insert(0, "")
|
||||
self.entry_pass.pack()
|
||||
# ---------------------------------------
|
||||
Button(self, text="Connexion", command=self.connecti).pack()
|
||||
self.resizable(width=False, height=False)
|
||||
#taille souhaite de la fenetre
|
||||
w = 150
|
||||
h = 110
|
||||
#pour centrer la fenetre
|
||||
#taille de l'ecran
|
||||
self.update()
|
||||
ws = self.winfo_screenwidth()
|
||||
hs = self.winfo_screenheight()
|
||||
|
||||
if getattr( sys, 'frozen', False ) :
|
||||
self.iconbitmap(sys._MEIPASS + "\id-card.ico\id-card.ico")
|
||||
else:
|
||||
self.iconbitmap("id-card.ico")
|
||||
|
||||
upwin.update()
|
||||
#calcul la position de la fenetre
|
||||
x = (ws/2) - (w/2)
|
||||
y = (hs/2) - (h/2)
|
||||
#applique la taille et la position
|
||||
self.geometry('%dx%d+%d+%d' % (w, h, x, y))
|
||||
|
||||
def connecti(self):
|
||||
global login
|
||||
global key
|
||||
login = self.entry_login.get().strip()
|
||||
key = self.entry_pass.get().strip()
|
||||
self.destroy()
|
||||
|
||||
session = PACSession()
|
||||
|
||||
if session.get_pac() == None:
|
||||
IPN = ""
|
||||
IPN_PASS = ""
|
||||
NoConnect = False
|
||||
|
||||
break
|
||||
|
||||
canvas.itemconfigure(message, text="En attente de connexion au serveur proxy...")
|
||||
global login
|
||||
global key
|
||||
login = ""
|
||||
key = ""
|
||||
|
||||
result = LoginDialog(upwin)
|
||||
result.transient(upwin)
|
||||
result.grab_set()
|
||||
upwin.wait_window(result)
|
||||
|
||||
IPN = login
|
||||
IPN_PASS = key
|
||||
|
||||
session = PACSession(proxy_auth=HTTPProxyAuth(IPN, IPN_PASS))
|
||||
Ans = session.get("http://www.google.com")
|
||||
|
||||
if str(Ans) == "<Response [407]>":
|
||||
canvas.itemconfigure(message, text="Identifiants erronés, accès refusé")
|
||||
logger.info("[download() thread] : " + "407 Error")
|
||||
time.sleep(1)
|
||||
elif str(Ans) == "<Response [200]>":
|
||||
logger.info("[download() thread] : " + "Connection ok !")
|
||||
NoConnect = False
|
||||
else:
|
||||
raise(IOError())
|
||||
|
||||
AESObj = AESCipher("<clé>")
|
||||
|
||||
with open(CST_FOLDER +"conf.ig","wb+") as f:
|
||||
logger.info("[download() thread] : " + "Saving credentials in encrypted config file")
|
||||
f.write(AESObj.encrypt(IPN + "||" + IPN_PASS))
|
||||
|
||||
|
||||
|
||||
if IPN == "i005316":
|
||||
canvas.itemconfigure(message, text="Bienvenue Thierry !")
|
||||
elif IPN == "i020251":
|
||||
canvas.itemconfigure(message, text="Bienvenue Samia !")
|
||||
elif IPN == "i018410":
|
||||
canvas.itemconfigure(message, text="Bienvenue Adrien !")
|
||||
elif IPN == "i003067":
|
||||
canvas.itemconfigure(message, text="Bienvenue Remy !")
|
||||
elif IPN == "i018422":
|
||||
canvas.itemconfigure(message, text="Bienvenue Eloise !")
|
||||
time.sleep(1)
|
||||
|
||||
try:
|
||||
Prox_us = session.get_pac().find_proxy_for_url(CST_LINK,"neoxgroup.eu")
|
||||
PROXY_USABLE = Prox_us[6:-1].split(";")[0]
|
||||
proxy_server_url = IPN +":"+IPN_PASS+"@" + PROXY_USABLE
|
||||
ph = urllib2.ProxyHandler( { 'http' : proxy_server_url } )
|
||||
auth = urllib2.ProxyBasicAuthHandler()
|
||||
server= urllib2.build_opener( ph, auth, urllib2.HTTPHandler )
|
||||
urllib2.install_opener(server)
|
||||
logger.info("[download() thread] : " + "Proxy connection initiated successfully")
|
||||
|
||||
except:
|
||||
|
||||
logger.info("[download() thread] : " + "Proxy connection not initiated")
|
||||
|
||||
try:
|
||||
urllib2.urlretrieve(url, filename)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error("[download() thread] : " + "HTTP ERROR " )
|
||||
return e
|
||||
|
||||
##updator()
|
||||
logger.info("[updator() thread] : " + "Prepare downloading the version recap file...")
|
||||
tempfile = CST_FOLDER + 'temp' + str(random.randint(11111,99999)) + ".cniu"
|
||||
isOk = download(CST_LINK+"Version.txt", tempfile)
|
||||
|
||||
if not isOk:
|
||||
raise(isOk)
|
||||
|
||||
urllib2.urlcleanup()
|
||||
logger.info("[updator() thread] : " + "Opening version recap file...")
|
||||
file_ver = open(tempfile, "r")
|
||||
logger.info("[updator() thread] : " + "Reading version recap file...")
|
||||
version = file_ver.read()
|
||||
logger.info("[updator() thread] : " + "Closing version recap file...")
|
||||
repert = version.split("|")
|
||||
file_ver.close()
|
||||
logger.info("[updator() thread] : " + "Deleting version recap file...")
|
||||
os.remove(tempfile)
|
||||
logger.info("[updator() thread] : " + "Parsing informations about version...")
|
||||
final_f = "CNI_file"
|
||||
final_ver = ['0','0','0']
|
||||
|
||||
for file in repert:
|
||||
|
||||
if str.startswith(file,CST_NAME): #On prend les fichiers d'interêt du répertoire
|
||||
|
||||
ver = file.replace(CST_NAME+"_","").split(".") #On ne garde que le numéro de version
|
||||
|
||||
temp = ["0","0","0"]
|
||||
|
||||
for i in range(len(ver)):
|
||||
|
||||
temp[i] = ver[i]
|
||||
|
||||
ver = temp.copy()
|
||||
|
||||
sum_fver = int(final_ver[0])*100 + int(final_ver[1])*10 + int(final_ver[2])
|
||||
|
||||
sum_ver = int(ver[0])*100 + int(ver[1])*10 + int(ver[2])
|
||||
|
||||
|
||||
if sum_ver > sum_fver:
|
||||
|
||||
final_ver = ver.copy()
|
||||
final_f = file
|
||||
|
||||
sum_ver = int(final_ver[0])*100 + int(final_ver[1])*10 + int(final_ver[2])
|
||||
|
||||
if final_f != "CNI_file" and (sum_ver > CST_SUM_VER):
|
||||
#On a une maj
|
||||
logger.info("[updator() thread] : " + "New version of CNIRevelator found !")
|
||||
canvas.itemconfigure(message, text="Mise à jour disponible ! Préparation du téléchargement...")
|
||||
logger.info("[updator() thread] : " + "Preparing download")
|
||||
|
||||
with open(CST_FOLDER +'conf.ig', "rb") as config:
|
||||
|
||||
logger.info("[updator() thread] : " + "Reading credentials for proxy in config file...")
|
||||
|
||||
AESObj = AESCipher("<clé>")
|
||||
IPN, IPN_PASS = AESObj.decrypt(config.read()).split("||")[0:2]
|
||||
|
||||
session = PACSession(proxy_auth=HTTPProxyAuth(IPN, IPN_PASS))
|
||||
|
||||
try:
|
||||
|
||||
Prox_us = session.get_pac().find_proxy_for_url(CST_LINK,"neoxgroup.eu")
|
||||
PROXY_USABLE = Prox_us[6:-1].split(";")[0]
|
||||
proxy_server_url = IPN +":"+IPN_PASS+"@" + PROXY_USABLE
|
||||
ph = urllib2.ProxyHandler( { 'http' : proxy_server_url } )
|
||||
auth = urllib2.ProxyBasicAuthHandler()
|
||||
server= urllib2.build_opener( ph, auth, urllib2.HTTPHandler )
|
||||
logger.info("[updator() thread] : " + "Connection to the proxy initiated successfully !")
|
||||
|
||||
|
||||
except:
|
||||
|
||||
canvas.itemconfigure(message, text="Téléchargement en connexion directe...")
|
||||
server= urllib2.build_opener()
|
||||
logger.info("[updator() thread] : " + "Direct connection initiated successfully")
|
||||
|
||||
logger.info("[updator() thread] : " + "Launching download of " + final_f)
|
||||
Statut = Download( CST_LINK + final_f, final_f, final_f, server, p, canvas, message, logger )
|
||||
|
||||
if Statut.success:
|
||||
|
||||
try:
|
||||
os.rename(final_f, final_f + ".exe")
|
||||
except IOError:
|
||||
logger.error("[updator() thread] : " + "Unable to rename the file ! Wait 3 sec and retry")
|
||||
time.sleep(3)
|
||||
try:
|
||||
os.rename(final_f, final_f + ".exe")
|
||||
except IOError:
|
||||
logger.critical("[updator() thread] : " + "Unable to rename the file !")
|
||||
|
||||
else:
|
||||
canvas.itemconfigure(message, text="Téléchargement terminé ! Préparation du lancement...")
|
||||
logger.info("[updator() thread] : " + "Download of " + final_f + "finished successfully")
|
||||
p.configure(mode = "indeterminate", value = 0, maximum = 20)
|
||||
p.start()
|
||||
time.sleep(1)
|
||||
logger.info("[updator() thread] : " + "Launching " + final_f)
|
||||
try:
|
||||
proc = subprocess.Popen(final_f + ".exe", shell=False,stdin=None, stdout=None, stderr=None, close_fds=True)
|
||||
except:
|
||||
logger.error("[updator() thread] : " + "Unable to start the new version ! Wait 3 sec and retry")
|
||||
time.sleep(3)
|
||||
try:
|
||||
proc = subprocess.Popen(final_f + ".exe", shell=False,stdin=None, stdout=None, stderr=None, close_fds=True)
|
||||
except Exception as e:
|
||||
logger.critical("[updator() thread] : " + "Unable to start the new version ! Stopping : " + str(e))
|
||||
showerror("Erreur d'appel de procédure distante", "Le lancement du nouveau programme a échoué, vous devez le lancer manuellement une fois cette fenêtre fermée")
|
||||
ret = 12
|
||||
else:
|
||||
canvas.itemconfigure(message, text="Echec de la mise à jour : Erreur HTTP. Préparation du lancement...")
|
||||
logger.error("[updator() thread] : " + "Update has failed with HTTP error")
|
||||
time.sleep(1)
|
||||
|
||||
|
||||
|
||||
else:
|
||||
canvas.itemconfigure(message, text="Logiciel déjà à jour. Préparation du lancement...")
|
||||
logger.info("[updator() thread] : " + "CNIRevelator is up to date !")
|
||||
time.sleep(1)
|
||||
ret = 11
|
||||
|
||||
|
||||
#Tesseract
|
||||
if os.path.exists(CST_FOLDER + "Tesseract-OCR4\\tesseract.exe"):
|
||||
os.environ["PATH"] = CST_FOLDER + "Tesseract-OCR4\\"
|
||||
os.environ["TESSDATA_PREFIX"] = CST_FOLDER + "Tesseract-OCR4\\tessdata"
|
||||
|
||||
|
||||
else:
|
||||
|
||||
final_f = "tesseract_4"
|
||||
|
||||
#On a une maj
|
||||
logger.info("[updator() thread] : " + "Downloading tesseract 4 !")
|
||||
canvas.itemconfigure(message, text="Mise à jour du module OCR ! Préparation du téléchargement...")
|
||||
logger.info("[updator() thread] : " + "Preparing download")
|
||||
|
||||
with open(CST_FOLDER +'conf.ig', "rb") as config:
|
||||
|
||||
logger.info("[updator() thread] : " + "Reading credentials for proxy in config file...")
|
||||
|
||||
AESObj = AESCipher("<clé>")
|
||||
IPN, IPN_PASS = AESObj.decrypt(config.read()).split("||")[0:2]
|
||||
|
||||
session = PACSession(proxy_auth=HTTPProxyAuth(IPN, IPN_PASS))
|
||||
|
||||
try:
|
||||
|
||||
Prox_us = session.get_pac().find_proxy_for_url(CST_LINK,"neoxgroup.eu")
|
||||
PROXY_USABLE = Prox_us[6:-1].split(";")[0]
|
||||
proxy_server_url = IPN +":"+IPN_PASS+"@" + PROXY_USABLE
|
||||
ph = urllib2.ProxyHandler( { 'http' : proxy_server_url } )
|
||||
auth = urllib2.ProxyBasicAuthHandler()
|
||||
server= urllib2.build_opener( ph, auth, urllib2.HTTPHandler )
|
||||
logger.info("[updator() thread] : " + "Connection to the proxy initiated successfully !")
|
||||
|
||||
|
||||
except:
|
||||
|
||||
canvas.itemconfigure(message, text="Téléchargement en connexion directe...")
|
||||
server= urllib2.build_opener()
|
||||
logger.info("[updator() thread] : " + "Direct connection initiated successfully")
|
||||
|
||||
logger.info("[updator() thread] : " + "Launching download of " + final_f)
|
||||
|
||||
Statut = Download( CST_LINK + final_f, CST_FOLDER + final_f, final_f, server, p, canvas, message, logger )
|
||||
|
||||
if Statut.success:
|
||||
|
||||
canvas.itemconfigure(message, text="Téléchargement terminé ! Installation...")
|
||||
logger.info("[updator() thread] : " + "Download of " + final_f + "finished successfully")
|
||||
p.configure(mode = "indeterminate", value = 0, maximum = 20)
|
||||
p.start()
|
||||
|
||||
try:
|
||||
zip_ref = zipfile.ZipFile(CST_FOLDER + final_f, 'r')
|
||||
zip_ref.extractall(CST_FOLDER)
|
||||
zip_ref.close()
|
||||
os.environ["PATH"] = CST_FOLDER + "Tesseract-OCR4\\"
|
||||
os.environ["TESSDATA_PREFIX"] = CST_FOLDER + "Tesseract-OCR4\\tessdata"
|
||||
canvas.itemconfigure(message, text="Installation terminée !")
|
||||
|
||||
|
||||
|
||||
except:
|
||||
logger.error("[updator() thread] : " + "Unable to install the module. Wait and retry")
|
||||
time.sleep(3)
|
||||
try:
|
||||
zip_ref = zipfile.ZipFile(CST_FOLDER + final_f, 'r')
|
||||
zip_ref.extractall(CST_FOLDER)
|
||||
zip_ref.close()
|
||||
os.environ["PATH"] = CST_FOLDER + "Tesseract-OCR4\\"
|
||||
os.environ["TESSDATA_PREFIX"] = CST_FOLDER + "Tesseract-OCR4\\tessdata"
|
||||
canvas.itemconfigure(message, text="Installation terminée !")
|
||||
|
||||
|
||||
|
||||
except Exception as e:
|
||||
logger.critical("[updator() thread] : " + "Unable to install the module ! Stopping : " + str(e))
|
||||
showerror("Erreur d'appel de procédure distante", "L'installation du module OCR a échoué, contactez le développeur.")
|
||||
ret = 11
|
||||
|
||||
else:
|
||||
|
||||
logger.critical("[updator() thread] : " + "Unable to download the module ! ")
|
||||
showerror("Erreur de téléchargement", "L'installation du module OCR a échoué, merci d'indiquer le chemin d'accès au fichier tesseract_4 dans la fenêtre suivante")
|
||||
|
||||
path = filedialog.askopenfilename(title = "Indiquez le chemin d'accès à tesseract_4...",filetypes = (("Tesseract_4","*.cni4"), ("Tesseract_4","*.cni4")))
|
||||
|
||||
if path != "":
|
||||
|
||||
try:
|
||||
canvas.itemconfigure(message, text="Installation...")
|
||||
zip_ref = zipfile.ZipFile(path, 'r')
|
||||
zip_ref.extractall(CST_FOLDER)
|
||||
zip_ref.close()
|
||||
logger.error("[updator() thread] : " + "Manual installation successed")
|
||||
canvas.itemconfigure(message, text="Installation terminée !")
|
||||
os.environ["PATH"] = CST_FOLDER + "Tesseract-OCR4\\"
|
||||
os.environ["TESSDATA_PREFIX"] = CST_FOLDER + "Tesseract-OCR4\\tessdata"
|
||||
|
||||
|
||||
except Exception as e:
|
||||
|
||||
logger.error("[updator() thread] : " + "Manual installation has failed" + str(e) )
|
||||
showerror("Erreur de lecture", "Le module OCR n'a pas pu être installé, la saisie automatique de scans ne pourra donc fonctionner")
|
||||
|
||||
|
||||
else:
|
||||
|
||||
showerror("Opération annulée", "Le module OCR n'a été installé, la saisie automatique de scans ne pourra donc fonctionner")
|
||||
|
||||
|
||||
except URLExcept.HTTPError as e:
|
||||
|
||||
canvas.itemconfigure(message, text="Echec de la mise à jour : Erreur HTTP " + str(e.code) + " . Préparation du lancement...")
|
||||
logger.error("[updator() thread] : " + "Update has failed with HTTP error" + str(e.code) )
|
||||
|
||||
if int(e.code) == 407:
|
||||
showerror("Erreur 407", "Attention : le système de mise à jour automatique a fait face à une erreur 407, signifiant que la connexion au serveur proxy a été refusée. Vos identifiants vous seront redemandés au prochain démarrage. La mise à jour a échoué.")
|
||||
logger.info("[updator() thread] : " + "Credential error. Deleting the config file...")
|
||||
os.remove(CST_FOLDER +"conf.ig")
|
||||
p.configure(mode = "indeterminate", value = 0, maximum = 20)
|
||||
p.start()
|
||||
time.sleep(3)
|
||||
|
||||
except Exception as e:
|
||||
|
||||
canvas.itemconfigure(message, text="Echec de la mise à jour. Préparation du lancement...")
|
||||
logger.error("[updator() thread] : " + "Error from the updating system : " + str(e))
|
||||
p.configure(mode = "indeterminate", value = 0, maximum = 20)
|
||||
p.start()
|
||||
time.sleep(2)
|
||||
|
||||
|
||||
p.stop()
|
||||
upwin.destroy()
|
||||
root.destroy()
|
||||
return ret
|
||||
|
||||
##updating()
|
||||
global ret
|
||||
logger.info("updating() : " + "Launching updator() thread...")
|
||||
threading.Thread(target=updator, daemon=True).start()
|
||||
logger.info("updating() [Thread] : " + "Ending updator() thread")
|
||||
return None
|
||||
|
||||
##SoftUpdate()
|
||||
|
||||
global ret
|
||||
ret = 11
|
||||
global upwin
|
||||
root = Tk()
|
||||
root.attributes('-alpha', 0.0) #For icon
|
||||
#root.lower()
|
||||
root.iconify()
|
||||
|
||||
upwin = Toplevel(root)
|
||||
upwin.overrideredirect(1)
|
||||
upwin.configure(bg = CST_COLOR)
|
||||
|
||||
upwin.resizable(width=False, height=False)
|
||||
#taille souhaite de la fenetre
|
||||
w = 600
|
||||
h = 300
|
||||
#pour centrer la fenetre
|
||||
#taille de l'ecran
|
||||
upwin.update()
|
||||
canvas = Canvas(upwin, width=600, height=270, bg=CST_COLOR, highlightthickness=0)
|
||||
pbar = Canvas(upwin, width=600, height=30, bg=CST_COLOR)
|
||||
p = ttk.Progressbar(pbar, orient=HORIZONTAL, length=590, mode='determinate')
|
||||
upwin.update()
|
||||
ws = upwin.winfo_screenwidth()
|
||||
hs = upwin.winfo_screenheight()
|
||||
canvas.create_text(w/2, h/3, text=CST_NAME + " " + CST_VERTITLE, font="Calibri 30 bold", fill="white")
|
||||
message = canvas.create_text(w/2, h/1.150, text=" ", font="Calibri 9", fill="white")
|
||||
upwin.update()
|
||||
#calcul la position de la fenetre
|
||||
x = (ws/2) - (w/2)
|
||||
y = (hs/2) - (h/2)
|
||||
#applique la taille et la position
|
||||
upwin.geometry('%dx%d+%d+%d' % (w, h, x, y))
|
||||
canvas.grid()
|
||||
pbar.grid()
|
||||
p.grid()
|
||||
upwin.after(2000, updating)
|
||||
|
||||
if getattr( sys, 'frozen', False ) :
|
||||
root.iconbitmap(sys._MEIPASS + "\id-card.ico\id-card.ico")
|
||||
else:
|
||||
root.iconbitmap("id-card.ico")
|
||||
|
||||
# On démarre la boucle Tkinter qui s'interrompt quand on ferme la fenêtre
|
||||
logger.info("SoftUpdate() : " + "Entering upwin mainloop()")
|
||||
upwin.protocol('WM_DELETE_WINDOW', lambda: root.destroy())
|
||||
upwin.mainloop()
|
||||
logger.info("SoftUpdate() : " + "Exiting upwin mainloop()")
|
||||
|
||||
if ret == 11:
|
||||
logger.info("SoftUpdate() : " + "OK to start to main() normally !")
|
||||
return True
|
||||
else:
|
||||
logger.info("SoftUpdate() : " + "Program will stop !")
|
||||
return False
|
||||
1551
CNI_classes.py
Normal file
1551
CNI_classes.py
Normal file
File diff suppressed because one or more lines are too long
413
CNI_pytesseract.py
Normal file
413
CNI_pytesseract.py
Normal file
@ -0,0 +1,413 @@
|
||||
"""
|
||||
********************************************************************************
|
||||
*** Projet CNI_Revelator ***
|
||||
|
||||
GNU GPL * 07/2018
|
||||
|
||||
Adrien Bourmault
|
||||
|
||||
Pytesseract modification to comply with Pyinstaller
|
||||
|
||||
*****
|
||||
"""
|
||||
#!/usr/bin/env python
|
||||
|
||||
'''
|
||||
Python-tesseract. For more information: https://github.com/madmaze/pytesseract
|
||||
'''
|
||||
|
||||
try:
|
||||
import Image
|
||||
except ImportError:
|
||||
from PIL import Image
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import tempfile
|
||||
import shlex
|
||||
import string
|
||||
from glob import iglob
|
||||
from pkgutil import find_loader
|
||||
from distutils.version import LooseVersion
|
||||
from os.path import realpath, normpath, normcase
|
||||
|
||||
numpy_installed = find_loader('numpy') is not None
|
||||
if numpy_installed:
|
||||
from numpy import ndarray
|
||||
|
||||
# CHANGE THIS IF TESSERACT IS NOT IN YOUR PATH, OR IS NAMED DIFFERENTLY
|
||||
tesseract_cmd = 'tesseract'
|
||||
RGB_MODE = 'RGB'
|
||||
OSD_KEYS = {
|
||||
'Page number': ('page_num', int),
|
||||
'Orientation in degrees': ('orientation', int),
|
||||
'Rotate': ('rotate', int),
|
||||
'Orientation confidence': ('orientation_conf', float),
|
||||
'Script': ('script', str),
|
||||
'Script confidence': ('script_conf', float)
|
||||
}
|
||||
|
||||
|
||||
class Output:
|
||||
STRING = "string"
|
||||
BYTES = "bytes"
|
||||
DICT = "dict"
|
||||
|
||||
|
||||
class TesseractError(RuntimeError):
|
||||
def __init__(self, status, message):
|
||||
self.status = status
|
||||
self.message = message
|
||||
self.args = (status, message)
|
||||
|
||||
|
||||
class TesseractNotFoundError(EnvironmentError):
|
||||
def __init__(self):
|
||||
super(TesseractNotFoundError, self).__init__(
|
||||
tesseract_cmd + " is not installed or it's not in your path"
|
||||
)
|
||||
|
||||
|
||||
class TSVNotSupported(EnvironmentError):
|
||||
def __init__(self):
|
||||
super(TSVNotSupported, self).__init__(
|
||||
'TSV output not supported. Tesseract >= 3.05 required'
|
||||
)
|
||||
|
||||
|
||||
def run_once(func):
|
||||
def wrapper(*args, **kwargs):
|
||||
if wrapper._result is wrapper:
|
||||
wrapper._result = func(*args, **kwargs)
|
||||
return wrapper._result
|
||||
|
||||
wrapper._result = wrapper
|
||||
return wrapper
|
||||
|
||||
|
||||
def get_errors(error_string):
|
||||
return u' '.join(
|
||||
line for line in error_string.decode('utf-8').splitlines()
|
||||
).strip()
|
||||
|
||||
|
||||
def cleanup(temp_name):
|
||||
''' Tries to remove files by filename wildcard path. '''
|
||||
for filename in iglob(temp_name + '*' if temp_name else temp_name):
|
||||
try:
|
||||
os.remove(filename)
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
|
||||
def prepare(image):
|
||||
if isinstance(image, Image.Image):
|
||||
return image
|
||||
|
||||
if numpy_installed and isinstance(image, ndarray):
|
||||
return Image.fromarray(image)
|
||||
|
||||
raise TypeError('Unsupported image object')
|
||||
|
||||
|
||||
def save_image(image):
|
||||
temp_name = tempfile.mktemp(prefix='tess_')
|
||||
if isinstance(image, str):
|
||||
return temp_name, realpath(normpath(normcase(image)))
|
||||
|
||||
image = prepare(image)
|
||||
img_extension = image.format
|
||||
if image.format not in {'JPEG', 'PNG', 'TIFF', 'BMP', 'GIF'}:
|
||||
img_extension = 'PNG'
|
||||
|
||||
if not image.mode.startswith(RGB_MODE):
|
||||
image = image.convert(RGB_MODE)
|
||||
|
||||
if 'A' in image.getbands():
|
||||
# discard and replace the alpha channel with white background
|
||||
background = Image.new(RGB_MODE, image.size, (255, 255, 255))
|
||||
background.paste(image, (0, 0), image)
|
||||
image = background
|
||||
|
||||
input_file_name = temp_name + os.extsep + img_extension
|
||||
image.save(input_file_name, format=img_extension, **image.info)
|
||||
return temp_name, input_file_name
|
||||
|
||||
|
||||
def subprocess_args(include_stdout=True):
|
||||
# The following is true only on Windows.
|
||||
if hasattr(subprocess, 'STARTUPINFO'):
|
||||
# On Windows, subprocess calls will pop up a command window by default
|
||||
# when run from Pyinstaller with the ``--noconsole`` option. Avoid this
|
||||
# distraction.
|
||||
si = subprocess.STARTUPINFO()
|
||||
si.dwFlags |= subprocess.STARTF_USESHOWWINDOW
|
||||
# Windows doesn't search the path by default. Pass it an environment so
|
||||
# it will.
|
||||
env = os.environ
|
||||
else:
|
||||
si = None
|
||||
env = None
|
||||
|
||||
# ``subprocess.check_output`` doesn't allow specifying ``stdout``::
|
||||
#
|
||||
# Traceback (most recent call last):
|
||||
# File "test_subprocess.py", line 58, in <module>
|
||||
# **subprocess_args(stdout=None))
|
||||
# File "C:\Python27\lib\subprocess.py", line 567, in check_output
|
||||
# raise ValueError('stdout argument not allowed, it will be overridden.')
|
||||
# ValueError: stdout argument not allowed, it will be overridden.
|
||||
#
|
||||
# So, add it only if it's needed.
|
||||
if include_stdout:
|
||||
ret = {'stdout': subprocess.PIPE}
|
||||
else:
|
||||
ret = {}
|
||||
|
||||
# On Windows, running this from the binary produced by Pyinstaller
|
||||
# with the ``--noconsole`` option requires redirecting everything
|
||||
# (stdin, stdout, stderr) to avoid an OSError exception
|
||||
# "[Error 6] the handle is invalid."
|
||||
ret.update({'stdin': subprocess.PIPE,
|
||||
'stderr': subprocess.PIPE,
|
||||
'startupinfo': si,
|
||||
'env': env
|
||||
})
|
||||
return ret
|
||||
|
||||
return kwargs
|
||||
|
||||
|
||||
def run_tesseract(input_filename,
|
||||
output_filename_base,
|
||||
extension,
|
||||
lang,
|
||||
config='',
|
||||
nice=0):
|
||||
cmd_args = []
|
||||
|
||||
if not sys.platform.startswith('win32') and nice != 0:
|
||||
cmd_args += ('nice', '-n', str(nice))
|
||||
|
||||
cmd_args += (tesseract_cmd, input_filename, output_filename_base)
|
||||
|
||||
if lang is not None:
|
||||
cmd_args += ('-l', lang)
|
||||
|
||||
cmd_args += shlex.split(config)
|
||||
|
||||
if extension not in ('box', 'osd', 'tsv'):
|
||||
cmd_args.append(extension)
|
||||
|
||||
try:
|
||||
proc = subprocess.Popen(cmd_args, **subprocess_args())
|
||||
except OSError:
|
||||
raise TesseractNotFoundError()
|
||||
|
||||
status_code, error_string = proc.wait(), proc.stderr.read()
|
||||
proc.stderr.close()
|
||||
|
||||
if status_code:
|
||||
raise TesseractError(status_code, get_errors(error_string))
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def run_and_get_output(image,
|
||||
extension,
|
||||
lang=None,
|
||||
config='',
|
||||
nice=0,
|
||||
return_bytes=False):
|
||||
|
||||
temp_name, input_filename = '', ''
|
||||
try:
|
||||
temp_name, input_filename = save_image(image)
|
||||
kwargs = {
|
||||
'input_filename': input_filename,
|
||||
'output_filename_base': temp_name + '_out',
|
||||
'extension': extension,
|
||||
'lang': lang,
|
||||
'config': config,
|
||||
'nice': nice
|
||||
}
|
||||
|
||||
run_tesseract(**kwargs)
|
||||
filename = kwargs['output_filename_base'] + os.extsep + extension
|
||||
with open(filename, 'rb') as output_file:
|
||||
if return_bytes:
|
||||
return output_file.read()
|
||||
return output_file.read().decode('utf-8').strip()
|
||||
finally:
|
||||
cleanup(temp_name)
|
||||
|
||||
|
||||
def file_to_dict(tsv, cell_delimiter, str_col_idx):
|
||||
result = {}
|
||||
rows = [row.split(cell_delimiter) for row in tsv.split('\n')]
|
||||
if not rows:
|
||||
return result
|
||||
|
||||
header = rows.pop(0)
|
||||
if len(rows[-1]) < len(header):
|
||||
# Fixes bug that occurs when last text string in TSV is null, and
|
||||
# last row is missing a final cell in TSV file
|
||||
rows[-1].append('')
|
||||
|
||||
if str_col_idx < 0:
|
||||
str_col_idx += len(header)
|
||||
|
||||
for i, head in enumerate(header):
|
||||
result[head] = [
|
||||
int(row[i]) if i != str_col_idx else row[i] for row in rows
|
||||
]
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def is_valid(val, _type):
|
||||
if _type is int:
|
||||
return val.isdigit()
|
||||
|
||||
if _type is float:
|
||||
try:
|
||||
float(val)
|
||||
return True
|
||||
except ValueError:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def osd_to_dict(osd):
|
||||
return {
|
||||
OSD_KEYS[kv[0]][0]: OSD_KEYS[kv[0]][1](kv[1]) for kv in (
|
||||
line.split(': ') for line in osd.split('\n')
|
||||
) if len(kv) == 2 and is_valid(kv[1], OSD_KEYS[kv[0]][1])
|
||||
}
|
||||
|
||||
|
||||
@run_once
|
||||
def get_tesseract_version():
|
||||
'''
|
||||
Returns LooseVersion object of the Tesseract version
|
||||
'''
|
||||
try:
|
||||
return LooseVersion(
|
||||
subprocess.check_output([tesseract_cmd, '--version'],
|
||||
**subprocess_args(False)).decode('utf-8').split()[1].lstrip(string.printable[10:])
|
||||
)
|
||||
except OSError:
|
||||
raise TesseractNotFoundError()
|
||||
|
||||
|
||||
def image_to_string(image,
|
||||
lang=None,
|
||||
config='',
|
||||
nice=0,
|
||||
boxes=False,
|
||||
output_type=Output.STRING):
|
||||
'''
|
||||
Returns the result of a Tesseract OCR run on the provided image to string
|
||||
'''
|
||||
if boxes:
|
||||
# Added for backwards compatibility
|
||||
print('\nWarning: Argument \'boxes\' is deprecated and will be removed'
|
||||
' in future versions. Use function image_to_boxes instead.\n')
|
||||
return image_to_boxes(image, lang, config, nice, output_type)
|
||||
|
||||
args = [image, 'txt', lang, config, nice]
|
||||
|
||||
if output_type == Output.DICT:
|
||||
return {'text': run_and_get_output(*args)}
|
||||
elif output_type == Output.BYTES:
|
||||
args.append(True)
|
||||
|
||||
return run_and_get_output(*args)
|
||||
|
||||
|
||||
def image_to_boxes(image,
|
||||
lang=None,
|
||||
config='',
|
||||
nice=0,
|
||||
output_type=Output.STRING):
|
||||
'''
|
||||
Returns string containing recognized characters and their box boundaries
|
||||
'''
|
||||
config += ' batch.nochop makebox'
|
||||
args = [image, 'box', lang, config, nice]
|
||||
|
||||
if output_type == Output.DICT:
|
||||
box_header = 'char left bottom right top page\n'
|
||||
return file_to_dict(box_header + run_and_get_output(*args), ' ', 0)
|
||||
elif output_type == Output.BYTES:
|
||||
args.append(True)
|
||||
|
||||
return run_and_get_output(*args)
|
||||
|
||||
|
||||
def image_to_data(image,
|
||||
lang=None,
|
||||
config='',
|
||||
nice=0,
|
||||
output_type=Output.STRING):
|
||||
'''
|
||||
Returns string containing box boundaries, confidences,
|
||||
and other information. Requires Tesseract 3.05+
|
||||
'''
|
||||
|
||||
if get_tesseract_version() < '3.05':
|
||||
raise TSVNotSupported()
|
||||
|
||||
config = '{} {}'.format('-c tessedit_create_tsv=1', config.strip()).strip()
|
||||
args = [image, 'tsv', lang, config, nice]
|
||||
|
||||
if output_type == Output.DICT:
|
||||
return file_to_dict(run_and_get_output(*args), '\t', -1)
|
||||
elif output_type == Output.BYTES:
|
||||
args.append(True)
|
||||
|
||||
return run_and_get_output(*args)
|
||||
|
||||
|
||||
def image_to_osd(image,
|
||||
lang='osd',
|
||||
config='',
|
||||
nice=0,
|
||||
output_type=Output.STRING):
|
||||
'''
|
||||
Returns string containing the orientation and script detection (OSD)
|
||||
'''
|
||||
config = '{}-psm 0 {}'.format(
|
||||
'' if get_tesseract_version() < '3.05' else '-',
|
||||
config.strip()
|
||||
).strip()
|
||||
args = [image, 'osd', lang, config, nice]
|
||||
|
||||
if output_type == Output.DICT:
|
||||
return osd_to_dict(run_and_get_output(*args))
|
||||
elif output_type == Output.BYTES:
|
||||
args.append(True)
|
||||
|
||||
return run_and_get_output(*args)
|
||||
|
||||
|
||||
def main():
|
||||
if len(sys.argv) == 2:
|
||||
filename, lang = sys.argv[1], None
|
||||
elif len(sys.argv) == 4 and sys.argv[1] == '-l':
|
||||
filename, lang = sys.argv[3], sys.argv[2]
|
||||
else:
|
||||
sys.stderr.write('Usage: python pytesseract.py [-l lang] input_file\n')
|
||||
exit(2)
|
||||
|
||||
try:
|
||||
print(image_to_string(Image.open(filename), lang=lang))
|
||||
except IOError:
|
||||
sys.stderr.write('ERROR: Could not open file "%s"\n' % filename)
|
||||
exit(1)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in New Issue
Block a user