mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-04 11:48:22 +02:00
114 lines
5.7 KiB
Python
114 lines
5.7 KiB
Python
from tkinter import messagebox
|
|
from threading import Thread
|
|
import subprocess
|
|
import shutil
|
|
import json
|
|
import glob
|
|
import os
|
|
|
|
from .definition import *
|
|
|
|
|
|
def patch_file(self):
|
|
def func():
|
|
try:
|
|
if os.path.exists("./file/Track-WU8/"):
|
|
total_track = len(os.listdir("./file/Track-WU8/"))
|
|
else:
|
|
total_track = 0
|
|
with open("./convert_file.json") as f:
|
|
fc = json.load(f)
|
|
max_step = len(fc["img"]) + total_track + 3 + len("EGFIS")
|
|
self.Progress(show=True, indeter=False, statut=self.translate("Conversion des fichiers"), max=max_step, step=0)
|
|
|
|
self.Progress(statut=self.translate("Configuration de LE-CODE"), add=1)
|
|
self.create_lecode_config()
|
|
|
|
self.Progress(statut=self.translate("Création de ct_icon.png"), add=1)
|
|
self.patch_ct_icon()
|
|
|
|
self.Progress(statut=self.translate("Création des images descriptives"), add=1)
|
|
self.patch_img_desc()
|
|
|
|
for i, file in enumerate(fc["img"]):
|
|
self.Progress(statut=self.translate("Conversion des images")+f"\n({i + 1}/{len(fc['img'])}) {file}", add=1)
|
|
subprocess.run(["./tools/szs/wimgt", "ENCODE", "./file/" + file, "-x", fc["img"][file], "--overwrite"],
|
|
creationflags=CREATE_NO_WINDOW, check=True, stdout=subprocess.PIPE)
|
|
|
|
for file in glob.glob(self.path_mkwf+"/files/Scene/UI/MenuSingle_?.szs"):
|
|
self.patch_bmg(file)
|
|
|
|
if os.path.exists("./file/auto-add"): shutil.rmtree("./file/auto-add")
|
|
if not(os.path.exists(self.path_mkwf + "/tmp/")): os.makedirs(self.path_mkwf + "/tmp/")
|
|
subprocess.run(["./tools/szs/wszst", "AUTOADD", get_nodir(self.path_mkwf) + "/files/Race/Course/",
|
|
"--DEST", get_nodir(self.path_mkwf) + "/tmp/auto-add/"],
|
|
creationflags=CREATE_NO_WINDOW, cwd=get_dir(self.path_mkwf),
|
|
check=True, stdout=subprocess.PIPE)
|
|
shutil.move(self.path_mkwf + "/tmp/auto-add/", "./file/auto-add/")
|
|
shutil.rmtree(self.path_mkwf + "/tmp/")
|
|
|
|
max_process = 8
|
|
process_list = {}
|
|
error_count, error_max = 0, 3
|
|
|
|
for i, file in enumerate(os.listdir("./file/Track-WU8/")):
|
|
while True:
|
|
if len(process_list) < max_process:
|
|
process_list[file] = None
|
|
self.Progress(statut=self.translate("Conversion des courses")+f"\n({i + 1}/{total_track})\n" +
|
|
"\n".join(process_list.keys()), add=1)
|
|
|
|
track_szs_file = f"./file/Track/{get_filename(file)}.szs"
|
|
if os.path.exists(track_szs_file):
|
|
if os.path.getsize(track_szs_file) < 1000: # File under this size are corrupted
|
|
os.remove(track_szs_file)
|
|
|
|
if not(os.path.exists(track_szs_file)):
|
|
process_list[file] = subprocess.Popen([
|
|
"./tools/szs/wszst", "NORMALIZE", "./file/Track-WU8/" + file, "--DEST",
|
|
"./file/Track/%N.szs", "--szs", "--overwrite", "--autoadd-path",
|
|
"./file/auto-add/"], creationflags=CREATE_NO_WINDOW, stderr=subprocess.PIPE)
|
|
break
|
|
else:
|
|
for process in process_list:
|
|
if process_list[process] is not None:
|
|
returncode = process_list[process].poll()
|
|
if returncode is None: pass # if the process is still running
|
|
else: # process ended
|
|
stderr = process_list[process].stderr.read()
|
|
if b"wszst: ERROR" in stderr: # Error occured
|
|
process_list.pop(process)
|
|
os.remove(f"./file/Track/{get_filename(process)}.szs")
|
|
error_count += 1
|
|
if error_count > error_max: # Too much track wasn't correctly converted
|
|
messagebox.showerror(
|
|
self.translate("Erreur"),
|
|
self.translate("Trop de course ont eu une erreur de conversion."))
|
|
return
|
|
else: # if the error max hasn't been reach
|
|
messagebox.showwarning(
|
|
self.translate("Attention"),
|
|
self.translate("La course ") +
|
|
process +
|
|
self.translate(" n'a pas été correctement converti. (") +
|
|
str(error_count) + "/"+str(error_max)+")")
|
|
break
|
|
|
|
else:
|
|
process_list.pop(process)
|
|
break
|
|
else:
|
|
process_list.pop(process)
|
|
break
|
|
|
|
self.button_install_mod.grid(row=2, column=1, sticky="NEWS")
|
|
self.listbox_outputformat.grid(row=2, column=2, sticky="NEWS")
|
|
|
|
except: self.log_error()
|
|
finally: self.Progress(show=False)
|
|
|
|
|
|
t = Thread(target=func)
|
|
t.setDaemon(True)
|
|
t.start()
|
|
return t
|