Merge branch 'master' into master

1 files changed, 234 insertions, 0 deletions

diff --git a/bot/seasons/evergreen/trivia_quiz.py b/bot/seasons/evergreen/trivia_quiz.py
new file mode 100644
index 00000000..798523e6
--- /dev/null
+++ b/bot/seasons/evergreen/trivia_quiz.py
@@ -0,0 +1,234 @@
+import asyncio
+import json
+import logging
+import random
+from pathlib import Path
+
+import discord
+from discord.ext import commands
+from fuzzywuzzy import fuzz
+
+from bot.constants import Roles
+
+
+logger = logging.getLogger(__name__)
+
+
+ANNOYED_EXPRESSIONS = ["-_-", "-.-"]
+
+WRONG_ANS_RESPONSE = [
+    "No one gave the correct answer",
+    "Better luck next time"
+]
+
+
+class TriviaQuiz(commands.Cog):
+    """A cog for all quiz commands."""
+
+    def __init__(self, bot: commands.Bot) -> None:
+        self.bot = bot
+        self.questions = self.load_questions()
+        self.game_status = {}
+        self.game_owners = {}
+        self.question_limit = 4
+        self.player_dict = {}
+        self.categories = {
+            "general": "Test your general knowledge"
+            # "retro": "Questions related to retro gaming."
+        }
+
+    @staticmethod
+    def load_questions() -> dict:
+        """Load the questions from  json file."""
+        p = Path("bot", "resources", "evergreen", "trivia_quiz.json")
+        with p.open() as json_data:
+            questions = json.load(json_data)
+            return questions
+
+    @commands.command(name="quiz", aliases=["trivia"])
+    async def quiz_game(self, ctx: commands.Context, category: str = "general") -> None:
+        """
+        Start/Stop a quiz!
+
+        arguments:
+        option:
+        - start : to start a quiz in a channel
+        - stop  : stop the quiz running in that channel.
+
+        Questions for the quiz can be selected from the following categories:
+        - general : Test your general knowledge. (default)
+        (we wil be adding more later)
+        """
+        category = category.lower()
+
+        if ctx.channel.id not in self.game_status:
+            self.game_status[ctx.channel.id] = False
+            self.player_dict[ctx.channel.id] = {}
+
+        if not self.game_status[ctx.channel.id]:
+            self.game_owners[ctx.channel.id] = ctx.author
+            self.game_status[ctx.channel.id] = True
+            start_embed = discord.Embed(colour=discord.Colour.red())
+            start_embed.title = "Quiz game Starting!!"
+            start_embed.description = "Each game consists of 5 questions.\n"
+            start_embed.description += "**Rules :**\nNo cheating and have fun!"
+            start_embed.set_footer(
+                text="Points for a question reduces by 25 after 10s or after a hint. Total time is 30s per question"
+            )
+            await ctx.send(embed=start_embed)  # send an embed with the rules
+            await asyncio.sleep(1)
+
+        else:
+            if (
+                    ctx.author == self.game_owners[ctx.channel.id]
+                    or Roles.moderator in [role.id for role in ctx.author.roles]
+            ):
+                await ctx.send("Quiz is no longer running.")
+                await self.declare_winner(ctx.channel, self.player_dict[ctx.channel.id])
+                self.game_status[ctx.channel.id] = False
+                del self.game_owners[ctx.channel.id]
+            else:
+                await ctx.send(f"{ctx.author.mention}, you are not authorised to stop this game :ghost: !")
+
+        if category not in self.categories:
+            embed = self.category_embed
+            await ctx.send(embed=embed)
+            return
+        topic = self.questions[category]
+
+        unanswered = 0
+        done_question = []
+        hint_no = 0
+        answer = None
+        while self.game_status[ctx.channel.id]:
+            if len(done_question) > self.question_limit and hint_no == 0:
+                await ctx.send("The round ends here.")
+                await self.declare_winner(ctx.channel, self.player_dict[ctx.channel.id])
+                break
+            if unanswered > 3:
+                await ctx.send("Game stopped due to inactivity.")
+                await self.declare_winner(ctx.channel, self.player_dict[ctx.channel.id])
+                break
+            if hint_no == 0:
+                while True:
+                    question_dict = random.choice(topic)
+                    if question_dict["id"] not in done_question:
+                        done_question.append(question_dict["id"])
+                        break
+                q = question_dict["question"]
+                answer = question_dict["answer"]
+
+                embed = discord.Embed(colour=discord.Colour.gold())
+                embed.title = f"Question #{len(done_question)}"
+                embed.description = q
+                await ctx.send(embed=embed)
+
+            def check(m: discord.Message) -> bool:
+                ratio = fuzz.ratio(answer.lower(), m.content.lower())
+                return ratio > 85 and m.channel == ctx.channel
+            try:
+                msg = await self.bot.wait_for('message', check=check, timeout=10)
+            except asyncio.TimeoutError:
+                if self.game_status[ctx.channel.id] is False:
+                    break
+                if hint_no < 2:
+                    hint_no += 1
+                    if "hints" in question_dict:
+                        hints = question_dict["hints"]
+                        await ctx.send(f"**Hint #{hint_no+1}\n**{hints[hint_no]}")
+                    else:
+                        await ctx.send(f"Cmon guys, {30-hint_no*10}s left!")
+
+                else:
+                    response = random.choice(WRONG_ANS_RESPONSE)
+                    expression = random.choice(ANNOYED_EXPRESSIONS)
+                    await ctx.send(f"{response} {expression}")
+                    await self.send_answer(ctx.channel, question_dict)
+                    await asyncio.sleep(1)
+                    hint_no = 0
+                    unanswered += 1
+                    await self.send_score(ctx.channel, self.player_dict[ctx.channel.id])
+                    await asyncio.sleep(2)
+
+            else:
+                points = 100 - 25*hint_no
+                if msg.author in self.player_dict[ctx.channel.id]:
+                    self.player_dict[ctx.channel.id][msg.author] += points
+                else:
+                    self.player_dict[ctx.channel.id][msg.author] = points
+                hint_no = 0
+                unanswered = 0
+                await ctx.send(f"{msg.author.mention} got the correct answer :tada: {points} points for ya.")
+                await self.send_answer(ctx.channel, question_dict)
+                await self.send_score(ctx.channel, self.player_dict[ctx.channel.id])
+                await asyncio.sleep(2)
+
+    @staticmethod
+    async def send_score(channel: discord.TextChannel, player_data: dict) -> None:
+        """A function which sends the score."""
+        embed = discord.Embed(colour=discord.Colour.blue())
+        embed.title = "Score Board"
+        embed.description = ""
+        for k, v in player_data.items():
+            embed.description += f"{k} : {v}\n"
+        await channel.send(embed=embed)
+
+    @staticmethod
+    async def declare_winner(channel: discord.TextChannel, player_data: dict) -> None:
+        """Announce the winner of the quiz in the game channel."""
+        if player_data:
+            highest_points = max(list(player_data.values()))
+            no_of_winners = list(player_data.values()).count(highest_points)
+
+            # Check if more than 1 player has highest points.
+            if no_of_winners > 1:
+                word = "You guys"
+                winners = []
+                points_copy = list(player_data.values()).copy()
+                for _ in range(no_of_winners):
+                    index = points_copy.index(highest_points)
+                    winners.append(list(player_data.keys())[index])
+                    points_copy[index] = 0
+                winners_mention = None
+                for winner in winners:
+                    winners_mention += f"{winner.mention} "
+
+            else:
+                word = "You"
+                author_index = list(player_data.values()).index(highest_points)
+                winner = list(player_data.keys())[author_index]
+                winners_mention = winner.mention
+            await channel.send(
+                f"Congratz {winners_mention} :tada: "
+                f"{word} have won this quiz game with a grand total of {highest_points} points!!"
+            )
+
+    @property
+    def category_embed(self) -> discord.Embed:
+        """Build an embed showing all available trivia categories."""
+        embed = discord.Embed(colour=discord.Colour.blue())
+        embed.title = "The available question categories are:"
+        embed.description = ""
+        for cat, description in self.categories.items():
+            embed.description += f"**- {cat.capitalize()}**\n{description.capitalize()}\n"
+        embed.set_footer(text="If not category is chosen, then a random one will be selected.")
+        return embed
+
+    @staticmethod
+    async def send_answer(channel: discord.TextChannel, question_dict: dict) -> None:
+        """Send the correct answer of a question to the game channel."""
+        answer = question_dict["answer"]
+        info = question_dict["info"]
+        embed = discord.Embed(color=discord.Colour.red())
+        embed.title = f"The correct answer is **{answer}**\n"
+        embed.description = ""
+        if info != "":
+            embed.description += f"**Information**\n{info}\n\n"
+        embed.description += "Lets move to the next question.\nRemaining questions: "
+        await channel.send(embed=embed)
+
+
+def setup(bot: commands.Bot) -> None:
+    """Loading the cog."""
+    bot.add_cog(TriviaQuiz(bot))
+    logger.debug("TriviaQuiz cog loaded")