diff options
Diffstat (limited to 'bot/seasons/evergreen/fun.py')
-rw-r--r-- | bot/seasons/evergreen/fun.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/bot/seasons/evergreen/fun.py b/bot/seasons/evergreen/fun.py new file mode 100644 index 00000000..ebc10fca --- /dev/null +++ b/bot/seasons/evergreen/fun.py @@ -0,0 +1,40 @@ +import logging +import random + +from discord.ext import commands + +log = logging.getLogger(__name__) + + +class Fun: + """ + A collection of general commands for fun. + """ + + def __init__(self, bot): + self.bot = bot + + @commands.command() + async def roll(self, ctx, str_input: str = None): + output = "" + + if not str_input: + output = "To use .roll, format it as such: .roll (number)" + else: + try: + num_rolls = int(str_input) + if num_rolls > 6: + num_rolls = 6 + elif num_rolls < 1: + return + for i in range(num_rolls): + output += ":terning%d: " % random.randint(1, 6) + except ValueError: + return + await ctx.send(output) + + +# Required in order to load the cog, use the class name in the add_cog function. +def setup(bot): + bot.add_cog(Fun(bot)) + log.debug("Fun cog loaded") |