mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-03 19:28:25 +02:00
130 lines
No EOL
6.4 KiB
Python
130 lines
No EOL
6.4 KiB
Python
from tkinter import messagebox
|
|
from threading import Thread
|
|
import subprocess
|
|
import shutil
|
|
import json
|
|
import glob
|
|
import os
|
|
|
|
from .definition import *
|
|
|
|
|
|
def install_mod(self):
|
|
def func():
|
|
try:
|
|
with open("./fs.json") as f:
|
|
fs = json.load(f)
|
|
|
|
# This part is used to estimate the max_step
|
|
extracted_file = []
|
|
max_step, step = 1, 0
|
|
|
|
def count_rf(path):
|
|
nonlocal max_step
|
|
max_step += 1
|
|
extension = get_extension(path)
|
|
if extension == "szs":
|
|
if not (os.path.realpath(path) in extracted_file):
|
|
extracted_file.append(os.path.realpath(path))
|
|
max_step += 1
|
|
|
|
for fp in fs:
|
|
for f in glob.glob(self.path_mkwf + "/files/" + fp, recursive=True):
|
|
if type(fs[fp]) == str:
|
|
count_rf(path=f)
|
|
elif type(fs[fp]) == dict:
|
|
for nf in fs[fp]:
|
|
if type(fs[fp][nf]) == str:
|
|
count_rf(path=f)
|
|
elif type(fs[fp][nf]) == list:
|
|
for ffp in fs[fp][nf]: count_rf(path=f)
|
|
###
|
|
extracted_file = []
|
|
max_step += 4 # PATCH main.dol et PATCH lecode.bin, conversion, changement d'ID
|
|
self.Progress(show=True, indeter=False, statut=self.translate("Installation du mod"), max=max_step, step=0)
|
|
|
|
def replace_file(path, file, subpath="/"):
|
|
self.Progress(statut=self.translate("Modification de")+f"\n{get_nodir(path)}", add=1)
|
|
extension = get_extension(path)
|
|
|
|
if extension == "szs":
|
|
if not (os.path.realpath(path) in extracted_file):
|
|
subprocess.check_output(["./tools/szs/wszst", "EXTRACT", get_nodir(path), "-d",
|
|
get_nodir(path) + ".d", "--overwrite"],
|
|
creationflags=CREATE_NO_WINDOW, cwd=get_dir(path))
|
|
extracted_file.append(os.path.realpath(path))
|
|
|
|
szs_extract_path = path + ".d"
|
|
if os.path.exists(szs_extract_path + subpath):
|
|
if subpath[-1] == "/":
|
|
filecopy(f"./file/{file}", szs_extract_path + subpath + file)
|
|
else:
|
|
filecopy(f"./file/{file}", szs_extract_path + subpath)
|
|
|
|
elif path[-1] == "/":
|
|
filecopy(f"./file/{file}", path + file)
|
|
else:
|
|
filecopy(f"./file/{file}", path)
|
|
|
|
for fp in fs:
|
|
for f in glob.glob(self.path_mkwf + "/files/" + fp, recursive=True):
|
|
if type(fs[fp]) == str:
|
|
replace_file(path=f, file=fs[fp])
|
|
elif type(fs[fp]) == dict:
|
|
for nf in fs[fp]:
|
|
if type(fs[fp][nf]) == str:
|
|
replace_file(path=f, subpath=nf, file=fs[fp][nf])
|
|
elif type(fs[fp][nf]) == list:
|
|
for ffp in fs[fp][nf]: replace_file(path=f, subpath=nf, file=ffp)
|
|
|
|
for file in extracted_file:
|
|
self.Progress(statut=self.translate("Recompilation de")+f"\n{get_nodir(file)}", add=1)
|
|
subprocess.check_output(["./tools/szs/wszst", "CREATE", get_nodir(file) + ".d", "-d", get_nodir(file),
|
|
"--overwrite"], creationflags=CREATE_NO_WINDOW, cwd=get_dir(file))
|
|
if os.path.exists(file + ".d"): shutil.rmtree(file + ".d")
|
|
|
|
self.Progress(statut=self.translate("Patch main.dol"), add=1)
|
|
subprocess.check_output(["./tools/szs/wstrt", "patch", get_nodir(self.path_mkwf) + "/sys/main.dol",
|
|
"--clean-dol", "--add-lecode"], creationflags=CREATE_NO_WINDOW,
|
|
cwd=get_dir(self.path_mkwf))
|
|
|
|
self.Progress(statut=self.translate("Patch lecode-PAL.bin"), add=1)
|
|
|
|
shutil.copytree("./file/Track/", self.path_mkwf+"/files/Race/Course/", dirs_exist_ok=True)
|
|
if not(os.path.exists(self.path_mkwf+"/tmp/")): os.makedirs(self.path_mkwf+"/tmp/")
|
|
filecopy("./file/CTFILE.txt", self.path_mkwf+"/tmp/CTFILE.txt")
|
|
filecopy("./file/lpar-default.txt", self.path_mkwf + "/tmp/lpar-default.txt")
|
|
filecopy("./file/lecode-PAL.bin", self.path_mkwf + "/tmp/lecode-PAL.bin")
|
|
|
|
subprocess.check_output(
|
|
["./tools/szs/wlect", "patch", "./tmp/lecode-PAL.bin", "-od", "./files/rel/lecode-PAL.bin",
|
|
"--track-dir", "./files/Race/Course/", "--move-tracks", "./files/Race/Course/", "--le-define",
|
|
"./tmp/CTFILE.txt", "--lpar", "./tmp/lpar-default.txt", "--overwrite"],
|
|
creationflags=CREATE_NO_WINDOW, cwd=self.path_mkwf)
|
|
|
|
shutil.rmtree(self.path_mkwf + "/files/Race/Course/")
|
|
shutil.rmtree(self.path_mkwf + "/tmp/")
|
|
|
|
outputformat = self.listbox_outputformat.get()
|
|
self.Progress(statut=self.translate("Conversion en")+f" {outputformat}", add=1)
|
|
|
|
if outputformat in ["ISO", "WBFS", "CISO"]:
|
|
self.path_mkwf_format = os.path.realpath(self.path_mkwf + "/../MKWFaraphel." + outputformat.lower())
|
|
subprocess.check_output(["./tools/wit/wit", "COPY", get_nodir(self.path_mkwf), "--DEST",
|
|
get_nodir(self.path_mkwf_format), f"--{outputformat.lower()}", "--overwrite"],
|
|
creationflags=CREATE_NO_WINDOW, cwd=get_dir(self.path_mkwf))
|
|
shutil.rmtree(self.path_mkwf)
|
|
|
|
self.Progress(statut=self.translate("Changement de l'ID du jeu"), add=1)
|
|
subprocess.check_output(["./tools/wit/wit", "EDIT", get_dir(self.path_mkwf_format), "--id", "RMCP60"],
|
|
creationflags=CREATE_NO_WINDOW, cwd=get_dir(self.path_mkwf_format))
|
|
|
|
messagebox.showinfo(self.translate("Fin"), self.translate("L'installation est terminé !"))
|
|
|
|
except: self.log_error()
|
|
finally: self.Progress(show=False)
|
|
|
|
t = Thread(target=func)
|
|
t.setDaemon(True)
|
|
t.start()
|
|
return t |