blob: 3584cf0244c1b43080adfdb7e74a4078dd809026 (
plain)
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
|
import csv
from enum import Enum
class Difficulty(Enum):
Terrible = -2
Low = -1
Moderate = 0
High = 1
Extreme = 2
@classmethod
def from_str(cls, name):
name = name.strip().lower()
for member in cls:
if member.name.lower() == name:
return member
raise ValueError(f"{name!r} is not a valid {cls.__name__}")
def load_csv(name, value_parser=None):
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] = {Difficulty.from_str(name):
value_parser(val) for name,val in row.items()}
return res
def parse_range(str_data):
try:
value = int(str_data)
return (value, value)
except ValueError:
return tuple([int(i) for i in str_data.split("-")[:2]])
class StatsDB:
def __init__(self) -> None:
self.ac = load_csv("ac", int)
self.saves = load_csv("saves", int)
self.hp = load_csv("hp", parse_range)
stats = StatsDB()
# if __name__ == '__main__':
# main()
|
