morgan/Grinch-AP
1
0
Fork 0
mirror of https://github.com/MarioSpore/Grinch-AP.git synced 2025-10-21 12:11:33 -06:00
Code Issues Packages Projects Releases Wiki Activity

Show "did you mean 'item/location_name'" in invalid item/location error. (#469)

Browse Source
This commit is contained in:
CaitSith2
2022-04-26 02:28:43 -07:00
committed by GitHub
parent 4f10dbb896
commit c326566bd2
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
Options.py
View File

@@ -6,6 +6,7 @@ import typing
import random import random
from schema import Schema, And, Or from schema import Schema, And, Or
from thefuzz import process as fuzzy_process
class AssembleOptions(abc.ABCMeta): class AssembleOptions(abc.ABCMeta):
@@ -456,13 +457,17 @@ class VerifyKeys:
if self.verify_item_name: if self.verify_item_name:
for item_name in self.value: for item_name in self.value:
if item_name not in world.item_names: if item_name not in world.item_names:
picks = fuzzy_process.extract(item_name, world.item_names, limit=2)
raise Exception(f"Item {item_name} from option {self} " raise Exception(f"Item {item_name} from option {self} "
f"is not a valid item name from {world.game}") f"is not a valid item name from {world.game}. "
f"Did you mean '{picks[0][0]}' ({picks[0][1]}% sure)")
elif self.verify_location_name: elif self.verify_location_name:
for location_name in self.value: for location_name in self.value:
if location_name not in world.location_names: if location_name not in world.location_names:
picks = fuzzy_process.extract(location_name, world.location_names, limit=2)
raise Exception(f"Location {location_name} from option {self} " raise Exception(f"Location {location_name} from option {self} "
f"is not a valid location name from {world.game}") f"is not a valid location name from {world.game}. "
f"Did you mean '{picks[0][0]}' ({picks[0][1]}% sure)")
class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys): class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys):