alttpr-gwaa-kiwi/alttpr-python
1
0
Fork 0
Code Issues Pull Requests Packages Releases Activity

formatting, and fix output

Browse Source
This commit is contained in:
Thomas Prescott
2021-03-02 20:12:24 -06:00
parent 7674e7dafb
commit 1bac04a24c
2 changed files with 68 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
CLI.py
View File

@@ -17,6 +17,7 @@ class ArgumentDefaultsHelpFormatter(argparse.RawTextHelpFormatter):
def _get_help_string(self, action): def _get_help_string(self, action):
return textwrap.dedent(action.help) return textwrap.dedent(action.help)
def parse_cli(argv, no_defaults=False): def parse_cli(argv, no_defaults=False):
def defval(value): def defval(value):
return value if not no_defaults else None return value if not no_defaults else None
@@ -41,36 +42,36 @@ def parse_cli(argv, no_defaults=False):
# get args # get args
args = [] args = []
with open(os.path.join("resources","app","cli","args.json")) as argsFile: with open(os.path.join("resources", "app", "cli", "args.json")) as argsFile:
args = json.load(argsFile) args = json.load(argsFile)
for arg in args: for arg in args:
argdata = args[arg] argdata = args[arg]
argname = "--" + arg argname = "--" + arg
argatts = {} argatts = {}
argatts["help"] = "(default: %(default)s)" argatts["help"] = "(default: %(default)s)"
if "action" in argdata: if "action" in argdata:
argatts["action"] = argdata["action"] argatts["action"] = argdata["action"]
if "choices" in argdata: if "choices" in argdata:
argatts["choices"] = argdata["choices"] argatts["choices"] = argdata["choices"]
argatts["const"] = argdata["choices"][0] argatts["const"] = argdata["choices"][0]
argatts["default"] = argdata["choices"][0] argatts["default"] = argdata["choices"][0]
argatts["nargs"] = "?" argatts["nargs"] = "?"
if arg in settings: if arg in settings:
default = settings[arg] default = settings[arg]
if "type" in argdata and argdata["type"] == "bool": if "type" in argdata and argdata["type"] == "bool":
default = settings[arg] != 0 default = settings[arg] != 0
argatts["default"] = defval(default) argatts["default"] = defval(default)
arghelp = fish.translate("cli","help",arg) arghelp = fish.translate("cli", "help", arg)
if "help" in argdata and argdata["help"] == "suppress": if "help" in argdata and argdata["help"] == "suppress":
argatts["help"] = argparse.SUPPRESS argatts["help"] = argparse.SUPPRESS
elif not isinstance(arghelp,str): elif not isinstance(arghelp, str):
argatts["help"] = '\n'.join(arghelp).replace("\\'","'") argatts["help"] = '\n'.join(arghelp).replace("\\'", "'")
else: else:
argatts["help"] = arghelp + " " + argatts["help"] argatts["help"] = arghelp + " " + argatts["help"]
parser.add_argument(argname,**argatts) parser.add_argument(argname, **argatts)
parser.add_argument('--seed', default=defval(int(settings["seed"]) if settings["seed"] != "" and settings["seed"] is not None else None), help="\n".join(fish.translate("cli","help","seed")), type=int) parser.add_argument('--seed', default=defval(int(settings["seed"]) if settings["seed"] != "" and settings["seed"] is not None else None), help="\n".join(fish.translate("cli", "help", "seed")), type=int)
parser.add_argument('--count', default=defval(int(settings["count"]) if settings["count"] != "" and settings["count"] is not None else 1), help="\n".join(fish.translate("cli","help","count")), type=int) parser.add_argument('--count', default=defval(int(settings["count"]) if settings["count"] != "" and settings["count"] is not None else 1), help="\n".join(fish.translate("cli", "help", "count")), type=int)
parser.add_argument('--customitemarray', default={}, help=argparse.SUPPRESS) parser.add_argument('--customitemarray', default={}, help=argparse.SUPPRESS)
# included for backwards compatibility # included for backwards compatibility
@@ -91,7 +92,7 @@ def parse_cli(argv, no_defaults=False):
if multiargs.multi: if multiargs.multi:
defaults = copy.deepcopy(ret) defaults = copy.deepcopy(ret)
for player in range(1, multiargs.multi + 1): for player in range(1, multiargs.multi + 1):
playerargs = parse_cli(shlex.split(getattr(ret,f"p{player}")), True) playerargs = parse_cli(shlex.split(getattr(ret, f"p{player}")), True)
for name in ['logic', 'mode', 'swords', 'goal', 'difficulty', 'item_functionality', for name in ['logic', 'mode', 'swords', 'goal', 'difficulty', 'item_functionality',
'shuffle', 'door_shuffle', 'intensity', 'crystals_ganon', 'crystals_gt', 'openpyramid', 'shuffle', 'door_shuffle', 'intensity', 'crystals_ganon', 'crystals_gt', 'openpyramid',
@@ -171,7 +172,7 @@ def parse_settings():
"quickswap": False, "quickswap": False,
"heartcolor": "red", "heartcolor": "red",
"heartbeep": "normal", "heartbeep": "normal",
"sprite": os.path.join(".","data","sprites","official","001.link.1.zspr"), "sprite": os.path.join(".", "data", "sprites", "official", "001.link.1.zspr"),
"fastmenu": "normal", "fastmenu": "normal",
"ow_palettes": "default", "ow_palettes": "default",
"uw_palettes": "default", "uw_palettes": "default",
@@ -282,8 +283,10 @@ def parse_settings():
# Priority fallback is: # Priority fallback is:
# 1: CLI # 1: CLI
# 2: Settings file # 2: Settings file(s)
# 3: Canned defaults # 3: Canned defaults
def get_args_priority(settings_args, gui_args, cli_args): def get_args_priority(settings_args, gui_args, cli_args):
args = {} args = {}
args["settings"] = parse_settings() if settings_args is None else settings_args args["settings"] = parse_settings() if settings_args is None else settings_args
@@ -310,7 +313,7 @@ def get_args_priority(settings_args, gui_args, cli_args):
for k in vars(args["cli"]): for k in vars(args["cli"]):
load_doesnt_have_key = k not in args["load"] load_doesnt_have_key = k not in args["load"]
cli_val = cli[k] cli_val = cli[k]
if isinstance(cli_val,dict) and 1 in cli_val: if isinstance(cli_val, dict) and 1 in cli_val:
cli_val = cli_val[1] cli_val = cli_val[1]
different_val = (k in args["load"] and k in cli) and (str(args["load"][k]) != str(cli_val)) different_val = (k in args["load"] and k in cli) and (str(args["load"][k]) != str(cli_val))
cli_has_empty_dict = k in cli and isinstance(cli_val, dict) and len(cli_val) == 0 cli_has_empty_dict = k in cli and isinstance(cli_val, dict) and len(cli_val) == 0
@@ -319,9 +322,9 @@ def get_args_priority(settings_args, gui_args, cli_args):
args["load"][k] = cli_val args["load"][k] = cli_val
newArgs = {} newArgs = {}
for key in [ "settings", "gui", "cli", "load" ]: for key in ["settings", "gui", "cli", "load"]:
if args[key]: if args[key]:
if isinstance(args[key],dict): if isinstance(args[key], dict):
newArgs[key] = argparse.Namespace(**args[key]) newArgs[key] = argparse.Namespace(**args[key])
else: else:
newArgs[key] = args[key] newArgs[key] = args[key]

74
Utils.py
View File

@@ -6,19 +6,24 @@ import sys
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
from collections import defaultdict from collections import defaultdict
def int16_as_bytes(value): def int16_as_bytes(value):
value = value & 0xFFFF value = value & 0xFFFF
return [value & 0xFF, (value >> 8) & 0xFF] return [value & 0xFF, (value >> 8) & 0xFF]
def int32_as_bytes(value): def int32_as_bytes(value):
value = value & 0xFFFFFFFF value = value & 0xFFFFFFFF
return [value & 0xFF, (value >> 8) & 0xFF, (value >> 16) & 0xFF, (value >> 24) & 0xFF] return [value & 0xFF, (value >> 8) & 0xFF, (value >> 16) & 0xFF, (value >> 24) & 0xFF]
def pc_to_snes(value): def pc_to_snes(value):
return ((value<<1) & 0x7F0000)|(value & 0x7FFF)|0x8000 return ((value << 1) & 0x7F0000) | (value & 0x7FFF) | 0x8000
def snes_to_pc(value): def snes_to_pc(value):
return ((value & 0x7F0000)>>1)|(value & 0x7FFF) return ((value & 0x7F0000) >> 1) | (value & 0x7FFF)
def parse_player_names(names, players, teams): def parse_player_names(names, players, teams):
names = [n for n in re.split(r'[, ]', names) if n] names = [n for n in re.split(r'[, ]', names) if n]
@@ -32,9 +37,11 @@ def parse_player_names(names, players, teams):
names = names[players:] names = names[players:]
return ret return ret
def is_bundled(): def is_bundled():
return getattr(sys, 'frozen', False) return getattr(sys, 'frozen', False)
def local_path(path): def local_path(path):
# just do stuff here and bail # just do stuff here and bail
return os.path.join(".", path) return os.path.join(".", path)
@@ -44,51 +51,26 @@ def local_path(path):
if is_bundled(): if is_bundled():
# we are running in a bundle # we are running in a bundle
local_path.cached_path = sys._MEIPASS # pylint: disable=protected-access,no-member local_path.cached_path = sys._MEIPASS # pylint: disable=protected-access,no-member
else: else:
# we are running in a normal Python environment # we are running in a normal Python environment
local_path.cached_path = os.path.dirname(os.path.abspath(__file__)) local_path.cached_path = os.path.dirname(os.path.abspath(__file__))
return os.path.join(local_path.cached_path, path) return os.path.join(local_path.cached_path, path)
local_path.cached_path = None local_path.cached_path = None
def output_path(path): def output_path(path):
# just do stuff here and bail if output_path.cached_path is None:
return os.path.join(".", path)
if output_path.cached_path is not None:
return os.path.join(output_path.cached_path, path)
if not is_bundled():
output_path.cached_path = '.' output_path.cached_path = '.'
return os.path.join(output_path.cached_path, path) return os.path.join(output_path.cached_path, path)
else:
# has been packaged, so cannot use CWD for output.
if sys.platform == 'win32':
#windows
documents = os.path.join(os.path.expanduser("~"),"Documents")
elif sys.platform == 'darwin':
from AppKit import NSSearchPathForDirectoriesInDomains # pylint: disable=import-error
# http://developer.apple.com/DOCUMENTATION/Cocoa/Reference/Foundation/Miscellaneous/Foundation_Functions/Reference/reference.html#//apple_ref/c/func/NSSearchPathForDirectoriesInDomains
NSDocumentDirectory = 9
NSUserDomainMask = 1
# True for expanding the tilde into a fully qualified path
documents = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, True)[0]
elif sys.platform.find("linux") or sys.platform.find("ubuntu") or sys.platform.find("unix"):
documents = os.path.join(os.path.expanduser("~"),"Documents")
else:
raise NotImplementedError('Not supported yet')
output_path.cached_path = os.path.join(documents, 'ALttPDoorRandomizer')
if not os.path.exists(output_path.cached_path):
os.makedirs(output_path.cached_path)
if not os.path.join(output_path.cached_path, path):
os.makedirs(os.path.join(output_path.cached_path, path))
return os.path.join(output_path.cached_path, path)
output_path.cached_path = None output_path.cached_path = None
def open_file(filename): def open_file(filename):
if sys.platform == 'win32': if sys.platform == 'win32':
os.startfile(filename) os.startfile(filename)
@@ -96,15 +78,17 @@ def open_file(filename):
open_command = 'open' if sys.platform == 'darwin' else 'xdg-open' open_command = 'open' if sys.platform == 'darwin' else 'xdg-open'
subprocess.call([open_command, filename]) subprocess.call([open_command, filename])
def close_console(): def close_console():
if sys.platform == 'win32': if sys.platform == 'win32':
#windows # windows
import ctypes.wintypes import ctypes.wintypes
try: try:
ctypes.windll.kernel32.FreeConsole() ctypes.windll.kernel32.FreeConsole()
except Exception: except Exception:
pass pass
def make_new_base2current(old_rom='Zelda no Densetsu - Kamigami no Triforce (Japan).sfc', new_rom='working.sfc'): def make_new_base2current(old_rom='Zelda no Densetsu - Kamigami no Triforce (Japan).sfc', new_rom='working.sfc'):
from collections import OrderedDict from collections import OrderedDict
import json import json
@@ -125,7 +109,7 @@ def make_new_base2current(old_rom='Zelda no Densetsu - Kamigami no Triforce (Jap
if offset - 1 in out_data: if offset - 1 in out_data:
out_data[offset-1].extend(out_data.pop(offset)) out_data[offset-1].extend(out_data.pop(offset))
with open('data/base2current.json', 'wt') as outfile: with open('data/base2current.json', 'wt') as outfile:
json.dump([{key:value} for key, value in out_data.items()], outfile, separators=(",", ":")) json.dump([{key: value} for key, value in out_data.items()], outfile, separators=(",", ":"))
basemd5 = hashlib.md5() basemd5 = hashlib.md5()
basemd5.update(new_rom_data) basemd5.update(new_rom_data)
@@ -241,7 +225,6 @@ def room_palette_data(old_rom):
print(f'{hex(header_offset)}: {[hex(x) for x in rooms]}') print(f'{hex(header_offset)}: {[hex(x) for x in rooms]}')
# Palette notes: # Palette notes:
# HC: 0 # HC: 0
# Sewer/Dungeon: 1 # Sewer/Dungeon: 1
@@ -303,21 +286,22 @@ def print_wiki_doors_by_region(d_regions, world, player):
toprint += ('| '+'<br />'.join(strs_to_print)) toprint += ('| '+'<br />'.join(strs_to_print))
toprint += "\n" toprint += "\n"
toprint += ('|}') + "\n" toprint += ('|}') + "\n"
with open(os.path.join(".","resources", "user", "regions-" + d + ".txt"),"w+") as f: with open(os.path.join(".", "resources", "user", "regions-" + d + ".txt"), "w+") as f:
f.write(toprint) f.write(toprint)
def update_deprecated_args(args): def update_deprecated_args(args):
if args: if args:
argVars = vars(args) argVars = vars(args)
truthy = [ 1, True, "True", "true" ] truthy = [1, True, "True", "true"]
# Hints default to TRUE # Hints default to TRUE
# Don't do: Yes # Don't do: Yes
# Do: No # Do: No
if "no_hints" in argVars: if "no_hints" in argVars:
src = "no_hints" src = "no_hints"
if isinstance(argVars["hints"],dict): if isinstance(argVars["hints"], dict):
tmp = {} tmp = {}
for idx in range(1,len(argVars["hints"]) + 1): for idx in range(1, len(argVars["hints"]) + 1):
tmp[idx] = argVars[src] not in truthy # tmp = !src tmp[idx] = argVars[src] not in truthy # tmp = !src
args.hints = tmp # dest = tmp args.hints = tmp # dest = tmp
else: else:
@@ -326,9 +310,9 @@ def update_deprecated_args(args):
# Do: Yes # Do: Yes
if "hints" in argVars: if "hints" in argVars:
src = "hints" src = "hints"
if isinstance(argVars["hints"],dict): if isinstance(argVars["hints"], dict):
tmp = {} tmp = {}
for idx in range(1,len(argVars["hints"]) + 1): for idx in range(1, len(argVars["hints"]) + 1):
tmp[idx] = argVars[src] not in truthy # tmp = !src tmp[idx] = argVars[src] not in truthy # tmp = !src
args.no_hints = tmp # dest = tmp args.no_hints = tmp # dest = tmp
else: else:
@@ -376,6 +360,7 @@ def update_deprecated_args(args):
return args return args
def print_wiki_doors_by_room(d_regions, world, player): def print_wiki_doors_by_room(d_regions, world, player):
for d, region_list in d_regions.items(): for d, region_list in d_regions.items():
tile_map = {} tile_map = {}
@@ -407,14 +392,15 @@ def print_wiki_doors_by_room(d_regions, world, player):
toprint += ('|-') + "\n" toprint += ('|-') + "\n"
toprint += ('! Door !! Room Side !! Requirement') + "\n" toprint += ('! Door !! Room Side !! Requirement') + "\n"
for ext in region.exits: for ext in region.exits:
ext_part = ext.name.replace(region.name,'') ext_part = ext.name.replace(region.name, '')
ext_part = ext_part.strip() ext_part = ext_part.strip()
toprint += ('{{DungeonRoomDoorList/Row|{{ROOTPAGENAME}}|{{SUBPAGENAME}}|' + ext_part + '|Side|}}') + "\n" toprint += ('{{DungeonRoomDoorList/Row|{{ROOTPAGENAME}}|{{SUBPAGENAME}}|' + ext_part + '|Side|}}') + "\n"
toprint += ('|}') + "\n" toprint += ('|}') + "\n"
toprint += ('') + "\n" toprint += ('') + "\n"
with open(os.path.join(".","resources", "user", "rooms-" + d + ".txt"),"w+") as f: with open(os.path.join(".", "resources", "user", "rooms-" + d + ".txt"), "w+") as f:
f.write(toprint) f.write(toprint)
def print_xml_doors(d_regions, world, player): def print_xml_doors(d_regions, world, player):
root = ET.Element('root') root = ET.Element('root')
for d, region_list in d_regions.items(): for d, region_list in d_regions.items():