Renaming modules, removing useless and changing logic of bot initialization

2025-10-29 16:57:23 +00:00 · 2022-01-06 18:25:17 -04:00
parent 9c327b8dbf
commit 74d91c15f8
22 changed files with 33 additions and 209 deletions
--- a/vulkan/commands/Control.py
+++ b/vulkan/commands/Control.py
@@ -0,0 +1,71 @@
+import discord
+from discord import Client
+from discord.ext.commands.errors import CommandNotFound, MissingRequiredArgument
+from discord.ext import commands
+from config import config
+
+
+class Control(commands.Cog):
+    """Control the flow of the Bot"""
+
+    def __init__(self, bot: Client):
+        self.__bot = bot
+        self.__comandos = {
+            'MUSIC': ['resume', 'pause', 'loop', 'stop', 'skip', 'play', 'queue', 'clear', 'np', 'shuffle', 'move', 'remove', 'reset'],
+            'WARFRAME': ['warframe'],
+            'RANDOM': ['escolha', 'cara', 'random'],
+            'HELP': ['help'],
+            'OTHERS': ['frase', 'drop']
+        }
+
+    @commands.Cog.listener()
+    async def on_ready(self):
+        print(config.STARTUP_MESSAGE)
+        await self.__bot.change_presence(status=discord.Status.online, activity=discord.Game(name=f"Vulkan | {config.BOT_PREFIX}help"))
+        print(config.STARTUP_COMPLETE_MESSAGE)
+
+    @commands.Cog.listener()
+    async def on_command_error(self, ctx, error):
+        if isinstance(error, MissingRequiredArgument):
+            await ctx.channel.send(f'Falta argumentos. Digite {config.BOT_PREFIX}help para ver todos os comandos\n\nOu tente {config.BOT_PREFIX}command help para mais informações')
+        elif isinstance(error, CommandNotFound):
+            await ctx.channel.send(f'O comando não existe')
+        else:
+            await ctx.channel.send(f'Teve um erro aí bicho')
+            raise error
+
+    @commands.command(name="help", alisases=['ajuda'], help=config.HELP_HELP)
+    async def help_msg(self, ctx):
+        helptxt = ''
+        help_music = '-- MUSIC\n'
+        help_random = '-- RANDOM\n'
+        help_warframe = '-- WARFRAME\n'
+        help_help = '-- HELP\n'
+        help_others = '-- OTHERS\n'
+
+        for command in self.__bot.commands:
+            if command.name in self.__comandos['MUSIC']:
+                help_music += f'**{command}** - {command.help}\n'
+            elif command.name in self.__comandos['HELP']:
+                help_help += f'**{command}** - {command.help}\n'
+            elif command.name in self.__comandos['OTHERS']:
+                help_others += f'**{command}** - {command.help}\n'
+            elif command.name in self.__comandos['WARFRAME']:
+                help_warframe += f'**{command}** - {command.help}\n'
+            else:
+                help_random += f'**{command}** - {command.help}\n'
+
+        helptxt = f'{help_music}\n{help_warframe}\n{help_random}\n{help_others}\n{help_help}'
+
+        embedhelp = discord.Embed(
+            colour=config.COLOURS['grey'],
+            title=f'Comandos do {self.__bot.user.name}',
+            description=helptxt
+        )
+
+        embedhelp.set_thumbnail(url=self.__bot.user.avatar_url)
+        await ctx.send(embed=embedhelp)
+
+
+def setup(bot):
+    bot.add_cog(Control(bot))