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
|
import csv
from game import Difficulty, RollInfo
def safe_item_level(creature_level: int):
if creature_level <= 3:
return 0
if creature_level == 4:
return 1
return creature_level - 4
def load_csv(name, value_parser=None):
def parse_col_name(name):
try:
return Difficulty.from_str(name)
except ValueError:
return name.lower()
if value_parser is None:
value_parser = lambda x:x
res = dict() # Table by level and Difficulty
with open(f"creature_stats/{name}.csv", newline='', encoding='utf8') as file:
reader = csv.DictReader(file)
for row in reader:
level = int(row.pop("Level"))
res[level] = {parse_col_name(name):
value_parser(val) for name,val in row.items()}
return res
def parse_stat(str_data):
try:
return int(str_data)
except ValueError:
pass
separator = "-"
if separator in str_data:
# parse range
values = [int(i) for i in str_data.split(separator)[:2]]
return tuple((min(*values), max(*values)))
else:
return RollInfo.parse(str_data)
class StatsDB:
stat_names = ["ac","abilities","attack_bonus", "hp",
"perception", "saves", "resistances", "skills", "strike_damage"]
def __init__(self) -> None:
self.stats = dict()
for stat_name in StatsDB.stat_names:
self.stats[stat_name] = load_csv(stat_name, parse_stat)
self.stats["resistances"] = {k:(v["minimum"], v["maximum"])
for k,v in self.stats["resistances"].items()}
stats = StatsDB()
# if __name__ == '__main__':
# main()
|
