mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-04 03:38:26 +02:00
80 lines
3.6 KiB
Python
80 lines
3.6 KiB
Python
import json
|
|
from .definition import *
|
|
|
|
|
|
def create_lecode_config(self):
|
|
try:
|
|
def get_star_text(track):
|
|
|
|
if "score" in track:
|
|
if 0 < track["score"] <= 3:
|
|
star_text = "★" * track["score"] + "☆" * (3 - track["score"])
|
|
return trackname_color[star_text] + " "
|
|
return ""
|
|
|
|
def get_ctfile_text(track, race=False):
|
|
if race:
|
|
return (f' T {track["music"]}; '
|
|
f'{track["special"]}; '
|
|
f'{"0x01" if track["new"] else "0x00"}; '
|
|
f'"-"; '
|
|
f'"{get_star_text(track)}{self.get_trackctname(track=track, color=True)}\\n{track["author"]}"; '
|
|
f'"-"\n')
|
|
else:
|
|
return (f' T {track["music"]}; '
|
|
f'{track["special"]}; '
|
|
f'{"0x01" if track["new"] else "0x00"}; '
|
|
f'"{self.get_trackctname(track=track)}"; '
|
|
f'"{get_star_text(track)}{self.get_trackctname(track=track, color=True)}"; '
|
|
f'"-"\n')
|
|
|
|
with open("./ct_config.json", encoding="utf-8") as f:
|
|
ctconfig = json.load(f)
|
|
|
|
with open("./file/CTFILE.txt", "w", encoding="utf-8") as ctfile, \
|
|
open("./file/RCTFILE.txt", "w", encoding="utf-8") as rctfile:
|
|
|
|
header = ("#CT-CODE\n"
|
|
"[RACING-TRACK-LIST]\n"
|
|
"%LE-FLAGS=1\n"
|
|
"%WIIMM-CUP=1\n"
|
|
"N N$SWAP | N$F_WII\n\n")
|
|
ctfile.write(header)
|
|
rctfile.write(header)
|
|
|
|
for cup in ctconfig["cup"]: # defined cup section
|
|
_cup_config = ctconfig["cup"][cup]
|
|
if int(cup) >= 9: # Track that are not original and not random selection
|
|
cup = f'\nC "{_cup_config["name"]}"\n'
|
|
ctfile.write(cup)
|
|
rctfile.write(cup)
|
|
|
|
for course in _cup_config["courses"]:
|
|
_course_config = _cup_config["courses"][course]
|
|
ctfile.write(get_ctfile_text(_course_config, race=False))
|
|
rctfile.write(get_ctfile_text(_course_config, race=True))
|
|
|
|
tracks_list = ctconfig["tracks_list"]
|
|
if not self.boolvar_use_1star_track.get(): # if 1 star track are disabled, remove them
|
|
tracks_list = list(filter(lambda track: track.get("score") != 1, tracks_list))
|
|
if not self.boolvar_use_2star_track.get(): # if 2 stars track are disabled, remove them
|
|
tracks_list = list(filter(lambda track: track.get("score") != 2, tracks_list))
|
|
if not self.boolvar_use_3star_track.get(): # if 3 stars track are disabled, remove them
|
|
tracks_list = list(filter(lambda track: track.get("score") != 3, tracks_list))
|
|
# using dict.get allow track that with no "score" attribute to not raise an exception by returning None
|
|
|
|
for i, _course_config in enumerate(tracks_list): # undefined cup section
|
|
if i % 4 == 0:
|
|
cup = f'\nC "TL{i//4}"\n'
|
|
ctfile.write(cup)
|
|
rctfile.write(cup)
|
|
|
|
ctfile.write(get_ctfile_text(_course_config, race=False))
|
|
rctfile.write(get_ctfile_text(_course_config, race=True))
|
|
|
|
for _ in range(1, 4-(i%4)): # Complete cup if track are missing
|
|
ctfile.write(EMPTY_TRACK)
|
|
rctfile.write(EMPTY_TRACK)
|
|
|
|
except:
|
|
self.log_error()
|