mirror of
https://github.com/CHOMPStation2/CHOMPStation2.git
synced 2025-12-10 18:22:39 +00:00
128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
# Common code for the frontend interface of map tools
|
|
import sys
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
from collections import namedtuple
|
|
|
|
Settings = namedtuple('Settings', ['map_folder', 'tgm'])
|
|
MapsToRun = namedtuple('MapsToRun', ['files', 'indices'])
|
|
|
|
def string_to_num(s):
|
|
try:
|
|
return int(s)
|
|
except ValueError:
|
|
return -1
|
|
|
|
def read_settings():
|
|
# discover map folder if needed
|
|
try:
|
|
map_folder = os.environ['MAPROOT']
|
|
except KeyError:
|
|
map_folder = '_maps/'
|
|
for _ in range(8):
|
|
if os.path.exists(map_folder):
|
|
break
|
|
map_folder = os.path.join('..', map_folder)
|
|
else:
|
|
map_folder = None
|
|
|
|
# assume TGM is True by default
|
|
tgm = os.environ.get('TGM', "1") == "1"
|
|
|
|
return Settings(map_folder, tgm)
|
|
|
|
def pretty_path(settings, path_str):
|
|
if settings.map_folder:
|
|
return path_str[len(os.path.commonpath([settings.map_folder, path_str]))+1:]
|
|
else:
|
|
return path_str
|
|
|
|
def prompt_maps(settings, verb):
|
|
if not settings.map_folder:
|
|
print("Could not autodetect the _maps folder, set MAPROOT")
|
|
exit(1)
|
|
|
|
list_of_files = list()
|
|
for root, directories, filenames in os.walk(settings.map_folder):
|
|
for filename in [f for f in filenames if f.endswith(".dmm")]:
|
|
list_of_files.append(pathlib.Path(root, filename))
|
|
|
|
last_dir = ""
|
|
for i, this_file in enumerate(list_of_files):
|
|
this_dir = this_file.parent
|
|
if last_dir != this_dir:
|
|
print("--------------------------------")
|
|
last_dir = this_dir
|
|
print("[{}]: {}".format(i, pretty_path(settings, str(this_file))))
|
|
|
|
print("--------------------------------")
|
|
in_list = input("List the maps you want to " + verb + " (example: 1,3-5,12):\n")
|
|
in_list = in_list.replace(" ", "")
|
|
in_list = in_list.split(",")
|
|
|
|
valid_indices = list()
|
|
for m in in_list:
|
|
index_range = m.split("-")
|
|
if len(index_range) == 1:
|
|
index = string_to_num(index_range[0])
|
|
if index >= 0 and index < len(list_of_files):
|
|
valid_indices.append(index)
|
|
elif len(index_range) == 2:
|
|
index0 = string_to_num(index_range[0])
|
|
index1 = string_to_num(index_range[1])
|
|
if index0 >= 0 and index0 <= index1 and index1 < len(list_of_files):
|
|
valid_indices.extend(range(index0, index1 + 1))
|
|
|
|
return MapsToRun(list_of_files, valid_indices)
|
|
|
|
def process(settings, verb, *, modify=True, backup=None):
|
|
if backup is None:
|
|
backup = modify # by default, backup when we modify
|
|
assert modify or not backup # doesn't make sense to backup when not modifying
|
|
|
|
if len(sys.argv) > 1:
|
|
maps = sys.argv[1:]
|
|
else:
|
|
maps = prompt_maps(settings, verb)
|
|
maps = [str(maps.files[i]) for i in maps.indices]
|
|
print()
|
|
|
|
if not maps:
|
|
print("No maps selected.")
|
|
return
|
|
|
|
if modify:
|
|
print(f"Maps WILL{'' if settings.tgm else ' NOT'} be converted to tgm.")
|
|
if backup:
|
|
print("Backups will be created with a \".before\" extension.")
|
|
else:
|
|
print("Warning: backups are NOT being taken.")
|
|
|
|
print(f"\nWill {verb} these maps:")
|
|
for path_str in maps:
|
|
print(pretty_path(settings, path_str))
|
|
|
|
try:
|
|
confirm = input(f"\nPress Enter to {verb}...\n")
|
|
except KeyboardInterrupt:
|
|
confirm = "^C"
|
|
if confirm != "":
|
|
print(f"\nAborted.")
|
|
return
|
|
|
|
for path_str in maps:
|
|
print(f' - {pretty_path(settings, path_str)}')
|
|
|
|
if backup:
|
|
shutil.copyfile(path_str, path_str + ".before")
|
|
|
|
try:
|
|
yield path_str
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
else:
|
|
print("Succeeded.")
|
|
|
|
print("\nFinished.")
|