Change submission to use database. Also use ids rather than names.

cogs/submission/submission_views.py

@@ -4,22 +4,25 @@ from typing import Optional
 
 import nextcord
 from nextcord.ext import commands
+from sqlalchemy import select
+from sqlalchemy.orm import aliased
+
+# pylint: disable=import-error
+from classes.submission import Submission, SubmissionState
+from highscores import Category, Guild, CategoryGroup, session, submission_objects
 
-from classes.submission import Submission, SubmissionState  # pylint: disable=import-error
-from highscores import (  # pylint: disable=import-error
-    highscores_data,
-    highscores_message_map,
-    submission_objects,
-)
 from .verification_view import VerificationView
 
 
-def category_select_options(guild_id: int, boss_name: str) -> "list[nextcord.SelectOption]":
+def category_select_options(group_id: int) -> "list[nextcord.SelectOption]":
     """Returns list of categories for a given boss."""
     options: "list[nextcord.SelectOption]" = []
-    category_list = highscores_data.get_guild(guild_id).get_table(boss_name).categories
-    for category in category_list:
-        options.append(nextcord.SelectOption(label=category.name))
+    subq = select(CategoryGroup).where(CategoryGroup.GroupId == group_id).subquery()
+    cat_group_alias = aliased(CategoryGroup, subq)
+    stmt = select(Category).where(Category.GroupId == cat_group_alias.GroupId)
+    categories_list = session.scalars(stmt)
+    for category in categories_list:
+        options.append(nextcord.SelectOption(label=category.CategoryName, value=str(category.CategoryId)))
 
     return options
 
@@ -35,14 +38,12 @@ class SubmissionDropdown(nextcord.ui.Select):
         bot: commands.Bot,
         options: "list[nextcord.SelectOption]",
     ):
-        super().__init__(
-            placeholder="Choose submission category.", min_values=1, max_values=1, options=options
-        )
+        super().__init__(placeholder="Choose submission category.", min_values=1, max_values=1, options=options)
         self._bot = bot
 
     async def callback(self, interaction: nextcord.Interaction):
         submission = submission_objects.get_user_submission(interaction.user.id)
-        submission.category = self.values[0]
+        submission.category_id = self.values[0]
         embed = submission.get_embed()
         await interaction.message.edit(
             embed=embed,
@@ -87,7 +88,7 @@ class NextButton(nextcord.ui.Button):
         # check if fields have been filled out for each state
         submission = submission_objects.get_user_submission(interaction.user.id)
         state: SubmissionState = submission.submission_state
-        if state == SubmissionState.CATEGORY and not submission.category:
+        if state == SubmissionState.CATEGORY and not submission.category_id:
             await interaction.send("Please choose a category.", ephemeral=True)
         elif state == SubmissionState.SCORE and not submission.score:
             await interaction.send("Please send a score.", ephemeral=True)
@@ -108,9 +109,8 @@ class SubmitButton(nextcord.ui.Button):
         self._bot = bot
 
     async def callback(self, interaction: nextcord.Interaction):
-        verification_channel_id = highscores_data.get_guild(
-            interaction.guild_id
-        ).verification_channel
+        stmt = select(Guild.VerificationChannel).where(Guild.GuildId == interaction.guild_id)
+        verification_channel_id = session.scalar(stmt)
         channel = interaction.guild.get_channel(verification_channel_id)
         submission = submission_objects.get_user_submission(interaction.user.id)
 
@@ -142,9 +142,7 @@ class SubmissionCreateButton(nextcord.ui.View):
         super().__init__(timeout=None)
         self._bot = bot
 
-    @nextcord.ui.button(
-        label="Submit", style=nextcord.ButtonStyle.primary, custom_id="submitcreate"
-    )
+    @nextcord.ui.button(label="Submit", style=nextcord.ButtonStyle.primary, custom_id="submitcreate")
     async def submitcreate_button(
         self, button: nextcord.ui.Button, interaction: nextcord.Interaction
     ):  # pylint: disable=unused-argument
@@ -154,17 +152,20 @@ class SubmissionCreateButton(nextcord.ui.View):
             await interaction.send("Please finish in progress submission first.", ephemeral=True)
             return
 
-        table = highscores_message_map[interaction.guild.id][interaction.message.id]
-        channel = interaction.guild.get_channel(
-            highscores_data.get_guild(interaction.guild_id).submission_channel
+        stmt = (
+            select(CategoryGroup.GroupId)
+            .where(CategoryGroup.GuildId == interaction.guild_id)
+            .where(CategoryGroup.MessageId == interaction.message.id)
         )
+        group_id = session.scalar(stmt)
+
+        stmt = select(Guild.SubmissionChannel).where(Guild.GuildId == interaction.guild_id)
+        channel = interaction.guild.get_channel(session.scalar(stmt))
 
         # a user should only have 1 submission active at once.
-        # split username on pipe then limit length.
-        username: str = interaction.user.display_name.split("|")[0][
-            : highscores_data.get_guild(interaction.guild_id).username_length
-        ]
-        submission: Submission = Submission(interaction.guild_id, table, username)
+        # split username on pipe. Decided to store whole username...
+        username: str = interaction.user.display_name.split("|")[0]
+        submission: Submission = Submission(interaction.guild_id, group_id, username)
 
         submission_objects.add_submission(interaction.user.id, submission)
 
@@ -197,14 +198,12 @@ class SubmissionView(nextcord.ui.View):
         self._submission_user = submission_user
         self._bot = bot
         self._state = submission_objects.get_user_submission(self._submission_user).submission_state
-        self._boss_name = submission_objects.get_user_submission(self._submission_user).table
+        self._group_id = submission_objects.get_user_submission(self._submission_user).group_id
         self._guild_id = submission_objects.get_user_submission(self._submission_user).guild_id
 
         # add items depending on given state
         if self._state is SubmissionState.CATEGORY:
-            self.add_item(
-                SubmissionDropdown(bot, category_select_options(self._guild_id, self._boss_name))
-            )
+            self.add_item(SubmissionDropdown(bot, category_select_options(self._group_id)))
             self.add_item(CancelButton())
             self.add_item(NextButton(self._bot))
         elif (