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 os.path
import json
import sys
from random import randrange, randint
class MaidDataReader:
PATH = os.path.join(os.path.dirname(__file__), "..", "..", "resources", "data.json")
def __init__(self):
self.data = None
with open(MaidDataReader.PATH, "r") as f:
content = f.read()
self.data = json.loads(content)
if self.data is None:
sys.exit("Could not read JSON data.")
rare_color_count = len(self.data["common"]["colors"]["rare"])
self.color_random_max = rare_color_count + 2 * len(self.data["common"]["colors"]["double_chance"])
self.rare_color_start_index = self.color_random_max - rare_color_count
def get_random_color(self):
index = randrange(self.color_random_max)
if index >= self.rare_color_start_index:
return self.data["common"]["colors"]["rare"][index - self.rare_color_start_index]
return self.data["common"]["colors"]["double_chance"][int(index / 2)]
def get_butler_random_suit_color(self):
return self.__get_butler_random_color(self.data["butler"]["colors"]["suit"])
def get_butler_random_hair_color(self):
return self.__get_butler_random_color(self.data["butler"]["colors"]["hair"])
@staticmethod
def generate_master_age(type_index):
if type_index == 0:
return randint(1, 6) + randint(1, 6)
elif type_index == 1:
return randint(1, 6) + randint(1, 6) + 5
elif type_index == 2:
return randint(1, 6) + randint(1, 6) + 8
elif type_index == 3:
return randint(1, 6) + randint(1, 6) + 12
elif type_index == 4:
return int(str(randint(1, 6)) + str(randint(1, 6)))
elif type_index == 5:
return int(str(randint(1, 6)) + str(randint(1, 6))) + 10
raise ValueError("Invalid master type index: {}".format(type_index))
def __get_butler_random_color(self, color_list):
index = randrange(len(color_list) + 1)
if index >= len(color_list):
return self.get_random_color()
return color_list[index]
|
