diff options
Diffstat (limited to 'sbdata')
| -rw-r--r-- | sbdata/__main__.py | 25 | ||||
| -rw-r--r-- | sbdata/repo.py | 2 | ||||
| -rw-r--r-- | sbdata/tasks.py | 23 |
3 files changed, 45 insertions, 5 deletions
diff --git a/sbdata/__main__.py b/sbdata/__main__.py index 596a51f..5453611 100644 --- a/sbdata/__main__.py +++ b/sbdata/__main__.py @@ -4,7 +4,8 @@ import sys from typing import Any import questionary - +import rich +import rich.table from sbdata.repo import Item from sbdata.task import Arguments, tasks @@ -19,6 +20,12 @@ class ObjectEncoder(json.JSONEncoder): return super().default(o) +def render_thing(i): + if isinstance(i, Item): + return i.internalname + return str(i) + + def main(): args = Arguments(sys.argv) task = args.get_value( @@ -30,6 +37,22 @@ def main(): data = task.run(args) if args.has_flag('json'): print(json.dumps(data, cls=ObjectEncoder)) + if args.has_flag('explore'): + if not (isinstance(data, list) and len(data) > 0 and dataclasses.is_dataclass(data[0])): + print('Cannot explore this') + return + console = rich.get_console() + keys = list(data[0].__dict__.keys()) + query = '' + while True: + table = rich.table.Table() + for k in keys: + table.add_column(k) + for item in data: + if any(query in render_thing(val).casefold() for val in item.__dict__.values()): + table.add_row(*[render_thing(getattr(item, k)) for k in keys]) + console.print(table) + query = console.input("Search: ") if __name__ == '__main__': diff --git a/sbdata/repo.py b/sbdata/repo.py index 475f2a6..7fdfca3 100644 --- a/sbdata/repo.py +++ b/sbdata/repo.py @@ -40,7 +40,7 @@ def find_item_by_name(name: str) -> typing.Optional[Item]: if item.internalname.casefold() == name or bare_name(item.displayname) in name or (item.itemid == 'minecraft:enchanted_book' - and name in bare_name(item.lore[0]))] + and bare_name(item.lore[0]).endswith(name))] if pot: return pot[0] return None diff --git a/sbdata/tasks.py b/sbdata/tasks.py index 8ac1f8c..ede5bac 100644 --- a/sbdata/tasks.py +++ b/sbdata/tasks.py @@ -15,6 +15,7 @@ class DungeonDrop: item: Item floor: int chest: str + cost: int drop_chances: dict[str, str] def get_drop_chance(self, has_s_plus: bool, talisman_level: int, boss_luck: int): @@ -22,6 +23,16 @@ class DungeonDrop: return self.drop_chances.get(drop_identifier) +default_chest_costs: dict[str, dict[int, int]] = dict( + Wood={7: 0}, + Gold={1: 25_000, 2: 50_000, 7: 100_000}, + Diamond={1: 50_000, 2: 100_000, 7: 250_000}, + Emerald={1: 100_000, 2: 250_000, 7: 500_000}, + Obsidian={1: 250_000, 2: 500_000, 7: 1_000_000}, + Bedrock={4: 4, 7: 2_000_000} +) + + @register_task("Fetch Dungeon Loot") def fetch_dungeon_loot(args: Arguments): items = [] @@ -31,6 +42,7 @@ def fetch_dungeon_loot(args: Arguments): item = None ifloor = None chest = None + cost = None drop_chances = {} for param in template.params: @@ -42,14 +54,19 @@ def fetch_dungeon_loot(args: Arguments): elif attr_name == 'customlink': if item is None: item = find_item_by_name(attr_value.split('#')[-1]) + elif attr_name == 'cost': + cost = int(attr_value.replace(',', '')) elif attr_name == 'chest': chest = attr_value elif attr_name == 'floor': ifloor = int(attr_value) elif attr_name.startswith("S"): drop_chances[attr_name] = attr_value - if item is None or ifloor is None or chest is None: + if item is None or ifloor is None or chest is None or cost is None: print('WARNING: Missing data for item: ' + str(template)) else: - items.append(DungeonDrop(item, ifloor, chest, drop_chances)) - return items + if cost == 0: + defaults = default_chest_costs[chest] + cost = defaults[min(f for f in defaults.keys() if f >= ifloor)] + items.append(DungeonDrop(item, ifloor, chest, cost, drop_chances)) + return items |