AO2XP/install_update.py
2021-06-22 18:56:22 -04:00

47 lines
1.5 KiB
Python

import zipfile, tarfile, platform, os, time, shutil
from Tkinter import Tk, HORIZONTAL
from ttk import Label, Progressbar
ext = {
"Windows": "zip",
"Darwin": "zip",
"Linux": "gz"
}
def extractzip(): # Mac
archive = zipfile.ZipFile("update.zip")
if platform.system() == "Darwin": shutil.rmtree("AO2XP.app", ignore_errors=True) # delete the old app package.
archive.extractall() # extract the new version
if os.path.exists("appbase"): # on a mac with the base folder moved out of .app Resources folder
os.system("mv appbase AO2XP.app/Contents/Resources/") # put it back in place
def extractgz(): # Linux
archive = tarfile.open("update.tar.gz")
archive.extractall()
if os.path.exists("update." + ext[platform.system()]):
window = Tk()
window.title("AO2XP updater")
window.wm_resizable(0,0)
window.geometry("256x56")
lb = Label(window)
lb.pack()
pb = Progressbar(orient=HORIZONTAL, mode="indeterminate")
pb.pack()
pb.start()
lb["text"] = "Waiting 3 seconds for AO2XP to close..."
time.sleep(3)
lb["text"] = "Extracting update." + ext[platform.system()] + "..."
globals()["extract" + ext[platform.system()]]() # call the extract function according to OS
lb["text"] = "Done!\nYou can now start AO2XP."
os.remove("update." + ext[platform.system()])
time.sleep(4)
else:
print "This program will be automatically run by AO2XP to apply updates.\nYou do not need to run this yourself."
raw_input("Press enter to exit.\n")