Adding new class to deal with random

vulkanbot/commands/Random.py

@@ -0,0 +1,80 @@
+from random import randint, random
+import discord
+from discord.ext import commands
+from config import config
+
+
+class Random(commands.Cog):
+    """Deal with returning random things"""
+
+    def __init__(self, bot):
+        self.__bot = bot
+
+    @commands.command(name='random', help='Número aleatório de 1 a X')
+    async def random(self, ctx, arg: str):
+        try:
+            arg = int(arg)
+
+        except Exception as e:
+            embed = discord.Embed(
+                description='Manda um número aí ow animal',
+                colour=config.COLOURS['red']
+            )
+            await ctx.send(embed=embed)
+            return
+
+        if arg < 1:
+            a = arg
+            b = 1
+        else:
+            a = 1
+            b = arg
+
+        x = randint(a, b)
+        embed = discord.Embed(
+            title=f'Número Aleatório entre {a, b}',
+            description=x,
+            colour=config.COLOURS['green']
+        )
+        await ctx.send(embed=embed)
+
+    @commands.command(name='cara', help='coroa')
+    async def cara(self, ctx):
+        x = random()
+        if x < 0.5:
+            result = 'cara'
+        else:
+            result = 'coroa'
+
+        embed = discord.Embed(
+            title='Cara Cora',
+            description=f'Resultado: {result}',
+            colour=config.COLOURS['green']
+        )
+        await ctx.send(embed=embed)
+
+    @commands.command(name='escolha', help='Escolhe um dos itens, separador: Vírgula')
+    async def escolher(self, ctx, *args: str):
+        try:
+            user_input = " ".join(args)
+            itens = user_input.split(sep=',')
+
+            index = randint(0, len(itens)-1)
+
+            embed = discord.Embed(
+                title='Escolha de algo',
+                description=itens[index],
+                colour=config.COLOURS['green']
+            )
+            await ctx.send(embed=embed)
+        except Exception as e:
+            embed = discord.Embed(
+                title='Escolha de algo',
+                description='Erro: Envie várias coisas separadas por vírgula',
+                colour=config.COLOURS['green']
+            )
+            await ctx.send(embed=embed)
+
+
+def setup(bot):
+    bot.add_cog(Random(bot))