Filling out skeleton dict for highscores_data rather than let it be empty and filled at data creation.

utils/data_storage.py

@@ -14,13 +14,28 @@ def open_highscores_data():
     Opens a highscores_data.json file if one exists, otherwise creates a skeleton of the data.
     @return: dict of highscores data.
     """
-    if (os.path.exists(HIGHSCORES_DATA_FILE)):
-        with open(HIGHSCORES_DATA_FILE, encoding='utf-8', mode="r") as file:
+    if os.path.exists(HIGHSCORES_DATA_FILE):
+        with open(HIGHSCORES_DATA_FILE, encoding="utf-8", mode="r") as file:
             data = json.load(file)
     else:
         # create the json object
-        data = {}
-
+        data = {"channel_id": -1}
+        config = open_highscores_config()
+        size = config["highscore_size"]
+        for boss in config["bosses"]:
+            data[boss["boss"]] = {"message_id": -1}
+            for category in config["categories"]:
+                rank_list = []
+                for i in range(size):  # pylint: disable=unused-variable
+                    rank = {"player": "", "score": -1}
+                    rank_list.append(rank)
+                data[boss["boss"]][category["category"]] = rank_list
+                if boss["hardmode"] and category["hardmode"]:
+                    rank_list = []
+                    for i in range(size):  # pylint: disable=unused-variable
+                        rank = {"player": "", "score": -1}
+                        rank_list.append(rank)
+                    data[boss["boss"]][category["category"] + "_hardmode"] = rank_list
     return data
 
 
@@ -30,7 +45,7 @@ def open_highscores_config():
     @return: dict of highscores config.
     """
     # Load bot config
-    with open(HIGHSCORES_CONFIG_FILE, encoding='utf-8', mode='r') as file:
+    with open(HIGHSCORES_CONFIG_FILE, encoding="utf-8", mode="r") as file:
         return json.load(file)
 
 
@@ -39,5 +54,5 @@ def save_highscores_data(data):
     Saves the highscores data dict into a json file.
     @param data: dict of highscores data.
     """
-    with open(HIGHSCORES_DATA_FILE, encoding='utf-8', mode='w') as file:
+    with open(HIGHSCORES_DATA_FILE, encoding="utf-8", mode="w") as file:
         file.write(json.dumps(data, indent=4))

utils/highscore_message.py

@@ -3,7 +3,6 @@ Contains various functions for sending/editing the highscore messages.
 """
 
 import nextcord
-from nextcord.ext import commands
 import highscores  # pylint: disable=import-error
 
 
@@ -55,28 +54,12 @@ async def send_highscore_message(channel, boss):
     """
     highscore_string = format_highscore_message(boss)
 
-    if (boss["boss"] in highscores.highscores_data) and (
-        "message_id" in highscores.highscores_data[boss["boss"]]
-    ):
-        message_id = highscores.highscores_data[boss["boss"]]["message_id"]
-        try:
-            message = await channel.fetch_message(message_id)
-            await message.edit(highscore_string)
-        except nextcord.NotFound:
-            # message doesn't exist.
-            message = await channel.send(highscore_string)
-            message_id = message.id
-            # check if there is a dict for the boss in the highscores data
-            if not boss["boss"] in highscores.highscores_data:
-                highscores.highscores_data[boss["boss"]] = {}
-
-            highscores.highscores_data[boss["boss"]]["message_id"] = message_id
-    else:
-        # make a new message
+    message_id = highscores.highscores_data[boss["boss"]]["message_id"]
+    try:
+        message = await channel.fetch_message(message_id)
+        await message.edit(highscore_string)
+    except nextcord.NotFound:
+        # message doesn't exist.
         message = await channel.send(highscore_string)
         message_id = message.id
-        # check if there is a dict for the boss in the highscores data
-        if not boss["boss"] in highscores.highscores_data:
-            highscores.highscores_data[boss["boss"]] = {}
-
         highscores.highscores_data[boss["boss"]]["message_id"] = message_id