1 files changed, 59 insertions, 0 deletions
diff --git a/src/modules/data_reader.py b/src/modules/data_reader.py
new file mode 100644
index 0000000..8bd9ac9
--- /dev/null
+++ b/src/modules/data_reader.py
@@ -0,0 +1,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]
+