136 lines
3.4 KiB
Python
136 lines
3.4 KiB
Python
import ctypes
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import webbrowser
|
|
appPath = sys.argv[1]
|
|
appName = appPath.replace(os.path.dirname(appPath)+ "\\","")
|
|
originalAppPath = appPath + " - umlOriginal"
|
|
tmpAppPath = appPath + " - umlTemp"
|
|
modPath = os.path.join(appPath,"umlMods")
|
|
originalModPath = os.path.join(originalAppPath,"umlMods")
|
|
|
|
def loadMods(output = False):
|
|
if areModsLoaded():
|
|
if unloadMods() == False:
|
|
if output: print("Unloading mods failed!")
|
|
return False
|
|
|
|
cloneFolder(appPath,tmpAppPath,False)
|
|
for root,dirs,files in walklevel(modPath,0):
|
|
for dir in dirs:
|
|
cloneFolder(os.path.join(root,dir),tmpAppPath,True,False)
|
|
|
|
os.rename(appPath,originalAppPath)
|
|
os.rename(tmpAppPath,appPath)
|
|
ctypes.windll.kernel32.SetFileAttributesW(originalAppPath,2)
|
|
|
|
if output: print("Mods have been loaded!")
|
|
return True
|
|
|
|
def unloadMods(output = False):
|
|
if areModsLoaded() == False:
|
|
if output: print("Mods are already unloaded.")
|
|
return True
|
|
|
|
shutil.rmtree(appPath)
|
|
os.rename(originalAppPath,appPath)
|
|
ctypes.windll.kernel32.SetFileAttributesW(appPath,128)
|
|
|
|
if output: print("Unloading mods successful.")
|
|
return True
|
|
|
|
def openModsFolder():
|
|
if areModsLoaded():
|
|
webbrowser.open("file:///" +originalModPath)
|
|
else:
|
|
webbrowser.open("file:///" +modPath)
|
|
|
|
def areModsLoaded():
|
|
if os.path.isdir(originalAppPath):
|
|
return True
|
|
|
|
return False
|
|
|
|
def cloneFolder(src,dst,rpl,ignoreMods = True):
|
|
for root,dirs,files in os.walk(src):
|
|
newRoot = root.replace(src,dst)
|
|
if ignoreMods == True:
|
|
if root.replace(modPath,"") != root: continue
|
|
|
|
if os.path.isdir(newRoot) == False: os.makedirs(newRoot)
|
|
|
|
for file in files:
|
|
fullFile = os.path.join(root,file)
|
|
newFile = os.path.join(newRoot,file)
|
|
|
|
if os.path.isfile(newFile):
|
|
if rpl == True:
|
|
os.remove(newFile)
|
|
os.link(fullFile,newFile)
|
|
else:
|
|
os.link(fullFile,newFile)
|
|
|
|
def walklevel(some_dir, level=1):
|
|
some_dir = some_dir.rstrip(os.path.sep)
|
|
assert os.path.isdir(some_dir)
|
|
num_sep = some_dir.count(os.path.sep)
|
|
for root, dirs, files in os.walk(some_dir):
|
|
yield root, dirs, files
|
|
num_sep_this = root.count(os.path.sep)
|
|
if num_sep + level <= num_sep_this:
|
|
del dirs[:]
|
|
|
|
def title(string):
|
|
if os.name == "nt":
|
|
os.system("title " +string)
|
|
else:
|
|
sys.stdout.write("\x1b]2;" +string+ "\x07")
|
|
|
|
def clear():
|
|
os.system('cls' if os.name=='nt' else 'clear')
|
|
|
|
def cleanUp():
|
|
if os.path.isdir(tmpAppPath):
|
|
shutil.rmtree(tmpAppPath)
|
|
|
|
def checkUp():
|
|
if areModsLoaded() == False:
|
|
if os.path.isdir(modPath) == False:
|
|
os.makedirs(modPath)
|
|
|
|
def mainMenu():
|
|
clear()
|
|
|
|
if areModsLoaded() == False:
|
|
print("Welcome to Fier's Universal Modloader.")
|
|
print("Mods are not loaded.")
|
|
print("")
|
|
print("Please choose an action:")
|
|
print("1) Load Mods")
|
|
print("2) Open Mods-Folder")
|
|
choice = input("Choice: ")
|
|
|
|
clear()
|
|
if choice == "1": loadMods(True); input("Press ENTER to continue.")
|
|
if choice == "2": openModsFolder()
|
|
else:
|
|
print("Welcome to Fier's Universal Modloader.")
|
|
print("Mods are loaded.")
|
|
print("")
|
|
print("Please choose an action:")
|
|
print("1) Reload Mods")
|
|
print("2) Unload Mods")
|
|
print("3) Open Mods-Folder")
|
|
choice = input("Choice: ")
|
|
|
|
clear()
|
|
if choice == "1": loadMods(True); input("Press ENTER to continue.")
|
|
if choice == "2": unloadMods(True); input("Press ENTER to continue.")
|
|
if choice == "3": openModsFolder()
|
|
|
|
title("Fier's Universal Modloader: " +appName)
|
|
cleanUp()
|
|
checkUp()
|
|
while True:
|
|
mainMenu() |