490 lines
17 KiB
Python
490 lines
17 KiB
Python
# -*- coding: utf8 -*-
|
|
"""
|
|
********************************************************************************
|
|
* CNIRevelator *
|
|
* *
|
|
* Desc: Application launcher updating system *
|
|
* *
|
|
* Copyright © 2018-2019 Adrien Bourmault (neox95) *
|
|
* *
|
|
* This file is part of CNIRevelator. *
|
|
* *
|
|
* CNIRevelator is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation, either version 3 of the License, or *
|
|
* any later version. *
|
|
* *
|
|
* CNIRevelator is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY*without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with CNIRevelator. If not, see <https:*www.gnu.org/licenses/>. *
|
|
********************************************************************************
|
|
"""
|
|
|
|
from win32com.client import Dispatch
|
|
from tkinter.messagebox import *
|
|
from tkinter import *
|
|
import pythoncom
|
|
import sys
|
|
import time
|
|
import os
|
|
import shutil
|
|
import zipfile
|
|
import hashlib
|
|
import subprocess
|
|
import psutil
|
|
import datetime
|
|
|
|
import critical # critical.py
|
|
import github # github.py
|
|
import ihm # ihm.py
|
|
import logger # logger.py
|
|
import globs # globs.py
|
|
import downloader # downloader.py
|
|
import lang # lang.py
|
|
|
|
UPDATE_IS_MADE = False
|
|
UPATH = ' '
|
|
|
|
launcherWindow = ihm.launcherWindowCur
|
|
|
|
def createShortcut(path, target='', wDir='', icon=''):
|
|
"""
|
|
Creates a shortcut for a program or an internet link
|
|
"""
|
|
ext = path[-3:]
|
|
if ext == 'url':
|
|
shortcut = file(path, 'w')
|
|
shortcut.write('[InternetShortcut]\n')
|
|
shortcut.write('URL=%s' % target)
|
|
shortcut.close()
|
|
else:
|
|
shell = Dispatch('WScript.Shell')
|
|
shortcut = shell.CreateShortCut(shell.SpecialFolders("Desktop") + r"\{}".format(path))
|
|
shortcut.Targetpath = target
|
|
shortcut.WorkingDirectory = wDir
|
|
if icon == '':
|
|
pass
|
|
else:
|
|
shortcut.IconLocation = icon
|
|
shortcut.save()
|
|
|
|
def spawnProcess(args, cd):
|
|
"""
|
|
Creates a new independant process. Used to launch a new version after update
|
|
"""
|
|
subprocess.Popen(args, close_fds=True, cwd=cd, creationflags=subprocess.DETACHED_PROCESS)
|
|
|
|
def exitProcess(arg):
|
|
"""
|
|
Forcefully quits a process. Used to help deletion of an old version or to quit properly
|
|
"""
|
|
# Quit totally without remain in memory
|
|
for process in psutil.process_iter():
|
|
if process.pid == os.getpid():
|
|
process.terminate()
|
|
sys.exit(arg)
|
|
|
|
def setUpdateChannel(choice):
|
|
"""
|
|
Sets the new update channel and forces new update at next launch
|
|
"""
|
|
if choice == "Beta":
|
|
with open(globs.CNIRUrlConfig, 'w') as (configFile):
|
|
configFile.write("{}\n0\n0".format(globs.CNIRBetaURL))
|
|
# Force new update
|
|
try:
|
|
os.remove(globs.CNIRLastUpdate)
|
|
except:
|
|
pass
|
|
elif choice == "Stable":
|
|
with open(globs.CNIRUrlConfig, 'w') as (configFile):
|
|
configFile.write("{}\n0\n0".format(globs.CNIRDefaultURL))
|
|
# Force new update
|
|
try:
|
|
os.remove(globs.CNIRLastUpdate)
|
|
except:
|
|
pass
|
|
|
|
def getUpdateChannel():
|
|
"""
|
|
Returns the current update channel
|
|
"""
|
|
with open(globs.CNIRUrlConfig, 'r') as (configFile):
|
|
url = configFile.read()
|
|
if not "master" in url:
|
|
return "Beta"
|
|
else:
|
|
return "Stable"
|
|
|
|
def getLatestVersion(credentials):
|
|
"""
|
|
Returns the latest version of the software
|
|
"""
|
|
|
|
finalver, finalurl, finalchecksum = [None]*3
|
|
|
|
# Global Handlers
|
|
logfile = logger.logCur
|
|
|
|
|
|
# First retrieving the urls !
|
|
while True:
|
|
try:
|
|
# Open the config file
|
|
logfile.printdbg('Reading urlconf.ig')
|
|
with open(globs.CNIRUrlConfig, 'r') as (configFile):
|
|
try:
|
|
# Reading it
|
|
reading = configFile.read()
|
|
# Parsing it
|
|
urlparsed = reading.split("\n")
|
|
break
|
|
|
|
except Exception as e:
|
|
raise IOError(str(e))
|
|
|
|
except FileNotFoundError:
|
|
logfile.printdbg('Recreate urlconf.ig')
|
|
# Recreating the url file
|
|
try:
|
|
os.mkdir(globs.CNIRFolder + '\\config')
|
|
except:
|
|
pass
|
|
with open(globs.CNIRUrlConfig, 'w') as (configFile):
|
|
configFile.write("{}\n0\n0".format(globs.CNIRDefaultURL))
|
|
|
|
# Getting the list of versions of the software
|
|
logfile.printdbg('Retrieving the software versions')
|
|
try:
|
|
os.mkdir(globs.CNIRFolder + '\\downloads')
|
|
except:
|
|
pass
|
|
getTheVersions = downloader.newdownload(credentials, urlparsed[0], globs.CNIRVerStock, "the version repository").download()
|
|
|
|
logfile.printdbg('Parsing the software versions')
|
|
with open(globs.CNIRVerStock) as versionsFile:
|
|
versionsTab = versionsFile.read().split("\n")[1].split("||")
|
|
logfile.printdbg('Versions retrieved : {}'.format(versionsTab))
|
|
# Choose the newer
|
|
finalver = globs.version.copy()
|
|
for entry in versionsTab:
|
|
if not entry:
|
|
continue
|
|
verstr, url, checksum = entry.split("|")
|
|
# Calculating sum considering we can have 99 sub versions
|
|
ver = verstr.split(".")
|
|
ver = [int(i) for i in ver]
|
|
finalsum = finalver[2] + finalver[1]*100 + finalver[0]*100*100
|
|
sum = ver[2] + ver[1]*100 + ver[0]*100*100
|
|
# Make a statement
|
|
if sum >= finalsum:
|
|
finalver = ver.copy()
|
|
finalurl = url
|
|
finalchecksum = checksum
|
|
else:
|
|
finalurl = url
|
|
finalchecksum = None
|
|
|
|
return (finalver, finalurl, finalchecksum)
|
|
|
|
|
|
def tessInstall(PATH, credentials):
|
|
# Global Handlers
|
|
logfile = logger.logCur
|
|
|
|
|
|
# Verifying that Tesseract is installed
|
|
if not os.path.exists(PATH + '\\Tesseract-OCR5\\'):
|
|
finalver, finalurl, finalchecksum = getLatestVersion(credentials)
|
|
|
|
if finalurl == None:
|
|
logfile.printerr('Unable to get the Tesseract url')
|
|
return False
|
|
|
|
tesseracturl = finalurl.replace("CNIRevelator.zip", "tesseract_5.zip")
|
|
|
|
# WE ASSUME THAT THE MAIN FILE IS CNIRevelator.zip AND THAT THE TESSERACT PACKAGE IS tesseract_5.zip
|
|
logfile.printdbg('Preparing download of Tesseract OCR 4...')
|
|
getTesseract = downloader.newdownload(credentials, tesseracturl, PATH + '\\downloads\\TsrtPackage.zip', "Tesseract 5 OCR Module").download()
|
|
|
|
try:
|
|
# CHECKSUM
|
|
BUF_SIZE = 65536 # lets read stuff in 64kb chunks!
|
|
|
|
sha1 = hashlib.sha1()
|
|
|
|
with open(globs.CNIRFolder + '\\downloads\\TsrtPackage.zip', 'rb') as f:
|
|
while True:
|
|
data = f.read(BUF_SIZE)
|
|
if not data:
|
|
break
|
|
sha1.update(data)
|
|
|
|
check = sha1.hexdigest()
|
|
logfile.printdbg("SHA1: {0}".format(check))
|
|
|
|
if not check == globs.CNIRTesserHash:
|
|
logfile.printerr("Checksum error")
|
|
return False
|
|
|
|
# Unzip Tesseract
|
|
logfile.printdbg("Unzipping the package")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Installing the updates"])
|
|
zip_ref = zipfile.ZipFile(PATH + '\\downloads\\TsrtPackage.zip', 'r')
|
|
zip_ref.extractall(PATH)
|
|
zip_ref.close()
|
|
# Cleanup
|
|
try:
|
|
os.remove(UPATH + '\\downloads\\TsrtPackage.zip')
|
|
except:
|
|
pass
|
|
return True
|
|
|
|
except:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
## Main Batch Function
|
|
def batch(credentials):
|
|
# Global Handlers
|
|
logfile = logger.logCur
|
|
|
|
|
|
# Get the latest version of CNIRevelator
|
|
finalver, finalurl, finalchecksum = getLatestVersion(credentials)
|
|
|
|
if finalver == globs.version:
|
|
logfile.printdbg('The software is already the newer version')
|
|
return True
|
|
|
|
logfile.printdbg('Preparing download for the new version')
|
|
|
|
getTheUpdate = downloader.newdownload(credentials, finalurl, globs.CNIRFolder + '\\downloads\\CNIPackage.zip', "CNIRevelator {}.{}.{}".format(finalver[0], finalver[1], finalver[2])).download()
|
|
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Verifying download..."])
|
|
|
|
# CHECKSUM
|
|
BUF_SIZE = 65536 # lets read stuff in 64kb chunks!
|
|
|
|
sha1 = hashlib.sha1()
|
|
|
|
with open(globs.CNIRFolder + '\\downloads\\CNIPackage.zip', 'rb') as f:
|
|
while True:
|
|
data = f.read(BUF_SIZE)
|
|
if not data:
|
|
break
|
|
sha1.update(data)
|
|
|
|
check = sha1.hexdigest()
|
|
logfile.printdbg("SHA1: {0}".format(check))
|
|
|
|
if not check == finalchecksum:
|
|
logfile.printerr("Checksum error")
|
|
return False
|
|
|
|
# And now prepare install
|
|
global UPATH
|
|
UPATH = globs.CNIRFolder + '\\..\\CNIRevelator' + "{}.{}.{}".format(finalver[0], finalver[1], finalver[2])
|
|
logfile.printdbg("Make place")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Preparing installation..."])
|
|
# Cleanup
|
|
try:
|
|
shutil.rmtree(UPATH + 'temp')
|
|
except Exception as e:
|
|
logfile.printdbg('Unable to cleanup : ' +str(e))
|
|
try:
|
|
shutil.rmtree(UPATH)
|
|
except Exception as e:
|
|
logfile.printdbg('Unable to cleanup : ' +str(e))
|
|
# Unzip
|
|
logfile.printdbg("Unzipping the package")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Installing the updates"])
|
|
zip_ref = zipfile.ZipFile(globs.CNIRFolder + '\\downloads\\CNIPackage.zip', 'r')
|
|
zip_ref.extractall(UPATH + "temp")
|
|
zip_ref.close()
|
|
|
|
# Move to the right place
|
|
shutil.copytree(UPATH + 'temp\\CNIRevelator', UPATH)
|
|
shutil.rmtree(UPATH + 'temp')
|
|
logfile.printdbg('Extracted :' + UPATH + '\\CNIRevelator.exe')
|
|
|
|
# Make a shortcut
|
|
# hide main window
|
|
pythoncom.CoInitialize()
|
|
root = Tk()
|
|
root.withdraw()
|
|
res = askquestion(lang.all[globs.CNIRlang]["Shortcut creation"], lang.all[globs.CNIRlang]["Would you like to create/update the shortcut for CNIRevelator on your desktop ?"])
|
|
if res == "yes":
|
|
createShortcut("CNIRevelator.lnk", UPATH + '\\CNIRevelator.exe', UPATH)
|
|
root.destroy()
|
|
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Success !"])
|
|
|
|
# Cleanup
|
|
try:
|
|
os.remove(globs.CNIRFolder + '\\downloads\\CNIPackage.zip')
|
|
except:
|
|
pass
|
|
# Time to quit
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Launching the new version..."])
|
|
global UPDATE_IS_MADE
|
|
UPDATE_IS_MADE = True
|
|
return True
|
|
|
|
## Main Function
|
|
def umain():
|
|
|
|
# Global Handlers
|
|
logfile = logger.logCur
|
|
|
|
# Cleaner for the old version if detected
|
|
if len(sys.argv) > 2 and str(sys.argv[1]) == "DELETE":
|
|
globs.CNIRNewVersion = True
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Deleting old version"])
|
|
logfile.printdbg("Old install detected : {}".format(sys.argv[2]))
|
|
while os.path.exists(str(sys.argv[2])):
|
|
try:
|
|
shutil.rmtree(str(sys.argv[2]))
|
|
except Exception as e:
|
|
logfile.printerr(str(e))
|
|
logfile.printdbg('Trying stop the process !')
|
|
launcherWindow.printmsg('Fail :{}'.format(e))
|
|
try:
|
|
for process in psutil.process_iter():
|
|
if process.name() == 'CNIRevelator.exe':
|
|
logfile.printdbg('Process found. Command line: {}'.format(process.cmdline()))
|
|
if process.pid == os.getpid():
|
|
logfile.printdbg("Don't touch us ! {} = {}".format(process.pid, os.getpid()))
|
|
else:
|
|
logfile.printdbg('Terminating process !')
|
|
process.terminate()
|
|
shutil.rmtree(str(sys.argv[2]))
|
|
break
|
|
except Exception as e:
|
|
logfile.printerr(str(e))
|
|
launcherWindow.printmsg('Fail :{}'.format(e))
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]['Starting...'])
|
|
|
|
# check we want open a file
|
|
elif len(sys.argv) > 1 and str(sys.argv[1]) != "DELETE":
|
|
globs.CNIROpenFile = True
|
|
logfile.printdbg("Command line received : {}".format(sys.argv))
|
|
|
|
credentials = downloader.newcredentials()
|
|
|
|
if not credentials.valid:
|
|
if credentials.login == "nointernet":
|
|
logfile.printerr("No Internet Error. No effective update !")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["No Internet Error. No effective update !"])
|
|
else:
|
|
logfile.printerr("Credentials Error. No effective update !")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]["Credentials Error. No effective update !"])
|
|
time.sleep(2)
|
|
launcherWindow.exit()
|
|
return 0
|
|
|
|
github.credentials = credentials
|
|
|
|
# Check if update is needed
|
|
currentDate = datetime.datetime.now()
|
|
|
|
if os.path.isfile(globs.CNIRLastUpdate):
|
|
with open(globs.CNIRLastUpdate, 'r') as (configFile):
|
|
try:
|
|
# Reading it
|
|
lastUpdate = datetime.datetime.strptime(configFile.read(),"%d/%m/%Y")
|
|
except Exception as e:
|
|
critical.crashCNIR(False)
|
|
time.sleep(3)
|
|
launcherWindow.exit()
|
|
return 1
|
|
else:
|
|
lastUpdate = datetime.datetime(1970,1,1)
|
|
|
|
if not globs.CNIRNewVersion and os.path.exists(globs.CNIRFolder + '\\Tesseract-OCR5\\') and (currentDate - lastUpdate).days < 7:
|
|
logfile.printdbg("No need to update, {} days remaining".format(7 - (currentDate - lastUpdate).days))
|
|
launcherWindow.exit()
|
|
return 0
|
|
|
|
# Update batch
|
|
try:
|
|
try:
|
|
# EXECUTING THE UPDATE BATCH
|
|
success = batch(credentials)
|
|
except Exception as e:
|
|
critical.crashCNIR(False)
|
|
launcherWindow.printmsg('ERROR : ' + str(e))
|
|
time.sleep(3)
|
|
launcherWindow.exit()
|
|
return 1
|
|
|
|
if success:
|
|
logfile.printdbg("Software is up-to-date !")
|
|
launcherWindow.printmsg('Software is up-to-date !')
|
|
# Recreating the url file
|
|
lastUpdate = currentDate
|
|
try:
|
|
os.mkdir(globs.CNIRFolder + '\\config')
|
|
except:
|
|
pass
|
|
|
|
with open(globs.CNIRLastUpdate, 'w') as (configFile):
|
|
try:
|
|
# Writing it
|
|
configFile.write("{}/{}/{}".format(currentDate.day, currentDate.month, currentDate.year))
|
|
except Exception as e:
|
|
critical.crashCNIR(False)
|
|
time.sleep(3)
|
|
launcherWindow.exit()
|
|
return
|
|
if UPDATE_IS_MADE:
|
|
launcherWindow.exit()
|
|
return 0
|
|
|
|
else:
|
|
logfile.printerr("An error occured. No effective update !")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]['An error occured. No effective update !'])
|
|
time.sleep(2)
|
|
launcherWindow.exit()
|
|
return 0
|
|
|
|
except:
|
|
critical.crashCNIR(False)
|
|
launcherWindow.exit()
|
|
return 2
|
|
|
|
try:
|
|
try:
|
|
# INSTALLING TESSERACT OCR
|
|
success = tessInstall(globs.CNIRFolder, credentials)
|
|
except Exception as e:
|
|
critical.crashCNIR(False)
|
|
launcherWindow.printmsg('ERROR : ' + str(e))
|
|
time.sleep(3)
|
|
launcherWindow.exit()
|
|
return 1
|
|
|
|
if success:
|
|
logfile.printdbg("Software is up-to-date !")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]['Software is up-to-date !'])
|
|
else:
|
|
logfile.printerr("An error occured. No effective update !")
|
|
launcherWindow.printmsg(lang.all[globs.CNIRlang]['An error occured. No effective update !'])
|
|
time.sleep(2)
|
|
launcherWindow.exit()
|
|
return 0
|
|
|
|
except:
|
|
critical.crashCNIR(False)
|
|
launcherWindow.exit()
|
|
return 2
|
|
|
|
time.sleep(2)
|
|
launcherWindow.exit()
|
|
return 0
|