Added verify and submission register subcommands. Removed bot argument from Register class.

cogs/register/__init__.py

@@ -2,63 +2,114 @@
 Bot command for registering various channels to the bot.
 """
 
-from nextcord import Embed
+from typing import Optional
+
+from nextcord import Embed, Interaction, Permissions, SlashOption
+import nextcord
 from nextcord.ext import commands
 
 from highscores import highscores_data  # pylint: disable=import-error
 
 
-class RegisterVerificationChannel(commands.Cog, name="Register"):
+class Register(commands.Cog, name="Register"):
-    """Bot command for registering various channels to the bot."""
-
-    def __init__(self, bot: commands.Bot) -> None:
-        self.bot = bot
-
-    @commands.command(name="register")
-    @commands.has_permissions(administrator=True)
-    async def register_highscores(self, ctx: commands.Context, registration_type: str) -> None:
-        """A command which registers current channel as for various uses.
-        Usage:
-        ```
-        ?register verify (Use in channel you want to verify submissions in.)
-        ?register highscore (Use in channel you want to show highscore messages in.)
-        ?register submission (Use in channel you want submission messages to be send it.)
-        ?register help (Shows help message.)
-        ```
     """
-        # respond to the message
+    Class holds various register commands.
-        channel_id = ctx.channel.id
-        guild = highscores_data.get_guild(ctx.guild.id)
 
-        if registration_type == "verify":
+    Args:
-            await ctx.send(f"Registered <#{channel_id}> as channel to post verifications in.")
+        commands.Cog: Parent class
-            guild.verification_channel = channel_id
+        name (str, optional): Name of Cog. Defaults to "Register".
-        elif registration_type == "highscore":
+    """
-            await ctx.send(f"Registered <#{channel_id}> as channel to post highscores in.")
+
+    def __init__(self) -> None:
+        """
+        Default constructor for Register class. Does nothing.
+        """
+
+    @nextcord.slash_command(default_member_permissions=Permissions(administrator=True))
+    async def register(self):
+        """
+        Parent slash command, never called.
+        """
+
+    @register.subcommand(description="Registers the channel for perminent highscores messages.")
+    async def highscore(
+        self,
+        interaction: nextcord.Interaction,
+        channel: nextcord.abc.GuildChannel = SlashOption(
+            description="Channel to register",
+            required=False,
+            channel_types=[nextcord.ChannelType.text],
+            default=None,
+        ),
+    ) -> None:
+        """
+        Command function to register the highscore printing channel.
+
+        Args:
+            interaction (nextcord.Interaction): Interaction that invoked the command
+            channel (nextcord.abc.GuildChannel, optional): Channel to register
+                Is a SlashOption. Limited to text channels, Defaults to None.
+        """
+        if not channel:
+            channel = interaction.channel
+        channel_id = channel.id
+        guild = highscores_data.get_guild(interaction.guild_id)
+        await interaction.send(f"Registered <#{channel_id}>.", ephemeral=True)
         guild.highscore_channel = channel_id
-        elif registration_type == "submission":
+        highscores_data.save_guild_data(guild)
-            await ctx.send(f"Registered <#{channel_id}> as channel to post submissions in.")
-            guild.submission_channel = channel_id
-        else:
-            embed = Embed(
-                title="Register command help", description="How to use the register command."
-            )
-            embed.add_field(
-                name="?register verify", value="Use in channel you want to verify submissions in."
-            )
-            embed.add_field(
-                name="?register highscore",
-                value="Use in channel you want to show highscore messages in.",
-            )
-            embed.add_field(
-                name="?register submission",
-                value="Use in channel you want submission messages to be send it.",
-            )
-            embed.add_field(name="?register help", value="Use to show this message.")
-            await ctx.reply(embed=embed)
-            # return, don't need to save data
-            return
 
+    @register.subcommand(description="Registers the channel for submission messages.")
+    async def submission(
+        self,
+        interaction: nextcord.Interaction,
+        channel: nextcord.abc.GuildChannel = SlashOption(
+            description="Channel to register",
+            required=False,
+            channel_types=[nextcord.ChannelType.text],
+            default=None,
+        ),
+    ) -> None:
+        """
+        Command function to register the submission message channel.
+
+        Args:
+            interaction (nextcord.Interaction): Interaction that invoked the command
+            channel (nextcord.abc.GuildChannel, optional): Channel to register
+                Is a SlashOption. Limited to text channels, Defaults to None.
+        """
+        if not channel:
+            channel = interaction.channel
+        channel_id = channel.id
+        guild = highscores_data.get_guild(interaction.guild_id)
+        await interaction.send(f"Registered <#{channel_id}>.", ephemeral=True)
+        guild.submission_channel = channel_id
+        highscores_data.save_guild_data(guild)
+
+    @register.subcommand(description="Registers the channel for verification messages.")
+    async def verify(
+        self,
+        interaction: nextcord.Interaction,
+        channel: nextcord.abc.GuildChannel = SlashOption(
+            description="Channel to register",
+            required=False,
+            channel_types=[nextcord.ChannelType.text],
+            default=None,
+        ),
+    ) -> None:
+        """
+        Command function to register the verification message channel.
+
+        Args:
+            interaction (nextcord.Interaction): Interaction that invoked the command
+            channel (nextcord.abc.GuildChannel, optional): Channel to register
+                Is a SlashOption. Limited to text channels, Defaults to None.
+        """
+        if not channel:
+            channel = interaction.channel
+        channel_id = channel.id
+        guild = highscores_data.get_guild(interaction.guild_id)
+        await interaction.send(f"Registered <#{channel_id}>.", ephemeral=True)
+        guild.verification_channel = channel_id
         highscores_data.save_guild_data(guild)
 
 
@@ -66,4 +117,4 @@ class RegisterVerificationChannel(commands.Cog, name="Register"):
 # It is necessary to add these functions to the bot.
 def setup(bot: commands.Bot) -> None:
     """Adds function to bot"""
-    bot.add_cog(RegisterVerificationChannel(bot))
+    bot.add_cog(Register())