1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
import ast
import dataclasses
import json
import re
import sys
import typing
from sbdata.repo import find_item_by_name, Item
from sbdata.task import register_task, Arguments
from sbdata.wiki import get_wiki_sources_by_title
@dataclasses.dataclass
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):
drop_identifier = "S" + ('+' if has_s_plus else '') + 'ABCD'[talisman_level] + str(len([i for i in [0, 1, 3, 5, 10] if i >= boss_luck]))
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 = []
for floor in get_wiki_sources_by_title(*[f'Template:Catacombs Floor {f} Loot Master' for f in ['I', 'II', 'III', 'IV', 'V', 'VI', 'VII']]).values():
for template in floor.filter_templates():
if template.name.strip() == 'Dungeon Chest Table/Row':
item = None
ifloor = None
chest = None
cost = None
drop_chances = {}
for param in template.params:
attr_name = param.name.nodes[0].strip()
attr_value = param.value.nodes[0].strip()
if attr_name == 'item':
if item is None:
item = find_item_by_name(attr_value)
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 or cost is None:
print('WARNING: Missing data for item: ' + str(template))
else:
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
|
