Merge pull request #202 from JackyFWong/master

Implemented .riddlemethis for the Easter season

1 files changed, 93 insertions, 0 deletions

diff --git a/bot/seasons/easter/easter_riddle.py b/bot/seasons/easter/easter_riddle.py
new file mode 100644
index 00000000..2cae299a
--- /dev/null
+++ b/bot/seasons/easter/easter_riddle.py
@@ -0,0 +1,93 @@
+import asyncio
+import logging
+import random
+from json import load
+from pathlib import Path
+
+import discord
+from discord.ext import commands
+
+from bot.constants import Colours
+
+log = logging.getLogger(__name__)
+
+with open(Path('bot', 'resources', 'easter', 'easter_riddle.json'), 'r', encoding="utf8") as f:
+    RIDDLE_QUESTIONS = load(f)
+
+TIMELIMIT = 10
+
+
+class EasterRiddle(commands.Cog):
+    """This cog contains the command for the Easter quiz!"""
+
+    def __init__(self, bot):
+        self.bot = bot
+        self.winners = []
+        self.correct = ""
+        self.current_channel = None
+
+    @commands.command(aliases=["riddlemethis", "riddleme"])
+    async def riddle(self, ctx):
+        """Gives a random riddle questions, then provides 2 hints at 10 second intervals before revealing the answer"""
+        if not self.current_channel:
+            self.current_channel = ctx.message.channel
+
+            random_question = random.choice(RIDDLE_QUESTIONS)
+            question = random_question["question"]
+            hints = random_question["riddles"]
+            self.correct = random_question["correct_answer"]
+
+            description = f"You have {TIMELIMIT} seconds before the first hint.\n\n"
+
+            q_embed = discord.Embed(title=question, description=description, colour=Colours.pink)
+
+            await ctx.send(embed=q_embed)
+            await asyncio.sleep(TIMELIMIT)
+
+            h_embed = discord.Embed(
+                title=f"Here's a hint: {hints[0]}!",
+                colour=Colours.pink
+            )
+
+            await ctx.send(embed=h_embed)
+            await asyncio.sleep(TIMELIMIT)
+
+            h_embed = discord.Embed(
+                title=f"Here's a hint: {hints[1]}!",
+                colour=Colours.pink
+            )
+
+            await ctx.send(embed=h_embed)
+            await asyncio.sleep(TIMELIMIT)
+
+            if self.winners:
+                win_list = " ".join(self.winners)
+                content = f"Well done {win_list} for getting it correct!"
+                self.winners = []
+            else:
+                content = "Nobody got it right..."
+
+            a_embed = discord.Embed(
+                title=f"The answer is: {self.correct}!",
+                colour=Colours.pink
+            )
+
+            await ctx.send(content, embed=a_embed)
+
+            self.current_channel = None
+
+    @commands.Cog.listener()
+    async def on_message(self, message):
+        """If a non-bot user enters a correct answer, their username gets added to self.winners"""
+        if self.current_channel == message.channel:
+            if self.bot.user != message.author:
+                if message.content.lower() == self.correct.lower():
+                    self.winners.append(message.author.mention)
+        else:
+            return
+
+
+def setup(bot):
+    """Cog load."""
+    bot.add_cog(EasterRiddle(bot))
+    log.info("Easter Riddle bot loaded")