diff --git a/cogs/commands/test.py b/cogs/commands/test.py index 8d91434..82e63a2 100644 --- a/cogs/commands/test.py +++ b/cogs/commands/test.py @@ -10,6 +10,17 @@ from config import DakinyConfig from utils import get_current_cycle +def is_me(): + """ + predicate that makes sure only I can run the test commands. + """ + + def predicate(interaction: nextcord.Interaction): + return interaction.user.id == 206602713511886849 + + return application_checks.check(predicate) + + class TestCog(commands.Cog, name="Test commands."): """ Class holds commands for testing. @@ -21,37 +32,12 @@ class TestCog(commands.Cog, name="Test commands."): """ self.bot = bot - def is_me(): - def predicate(interaction: nextcord.Interaction): - return interaction.user.id == 206602713511886849 - - return application_checks.check(predicate) - @nextcord.slash_command() - @is_me() async def test(self, interaction: nextcord.Interaction): """ Parent slash command for help. Never called. """ - @test.subcommand() - @is_me() - async def msg_builder(self, interaction: nextcord.Interaction): - """ - Tests the message builder. - - Args: - interaction (nextcord.Interaction): Interaction that invoked the command. - """ - - cycle = ElectionCycle(0, self.bot) - - message = cycle.build_election_announcement() - - # doesn't tell discord that the message was responded to. Is a test command so I don't really care atm. - for m in message: - await interaction.channel.send(content=m) - @test.subcommand() @is_me() async def election_start(self, interaction: nextcord.Interaction):