monorail
/
Archipelago
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow specififying returned default values. Disallow ALL zero weights.

This commit is contained in:
caitsith2 2020-06-18 08:51:38 -07:00
parent be48f96f65
commit 75e185e437
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Mystery.py
View File

@ -223,15 +223,17 @@ def convert_to_on_off(value):
return {True: "on", False: "off"}.get(value, value) return {True: "on", False: "off"}.get(value, value)
def get_choice(option, root) -> typing.Any: def get_choice(option, root, value=None) -> typing.Any:
if option not in root: if option not in root:
return None return value
if type(root[option]) is not dict: if type(root[option]) is not dict:
return interpret_on_off(root[option]) return interpret_on_off(root[option])
if not root[option]: if not root[option]:
return None return value
if any(list(map(int, root[option].values()))):
return interpret_on_off( return interpret_on_off(
random.choices(list(root[option].keys()), weights=list(map(int, root[option].values())))[0]) random.choices(list(root[option].keys()), weights=list(map(int, root[option].values())))[0])
raise RuntimeError(f"All options specified in {option} are weighted as zero.")
def handle_name(name: str): def handle_name(name: str):