mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-04 03:38:26 +02:00
raphael60650
ae5e3e5987
if track is corrupted, it will be reconverted if too much track aren't properly converted, the installation stop.
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)
|
|
|
|
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
|