main.py

import nextcord
from nextcord.ext import commands
from typing import Optional         #Used to make slash commands optional
import urllib.request               #Used to open urls of opentdb.com
import random                       #Used to randomize choices of each question
import asyncio                      #Used to keep a waiting time between questions, by default 10s
import json                         #Used to convert bytes to json while reading content from opentdb.com
import html                         #Used to convert few special characters to readable characters when reading from json, fetched from the api
import time

import config_quiz_bot

prefix = "&"
bot = commands.Bot(command_prefix = prefix, intents = nextcord.Intents.all())

@bot.event
async def on_ready():
    print(f'We have logged in as {bot.user}')


#Default Values
api_url = "https://opentdb.com/api.php"         #API used to fetch Questions
duration = 30                                   #Standard duration used in various places

#Function to handle users joining the quiz about to start
def participate():
    duration = 30
    points = {}
    users = set()
    class Participate(nextcord.ui.View):
        global embed
        embed = nextcord.Embed(title= f"Quiz starting in {duration} seconds", description= "React below to join!", color= 0x0c97f6)

        @nextcord.ui.button(label="Join the Quiz!",style=nextcord.ButtonStyle.blurple)
        async def callback(self, button:nextcord.Button, interaction: nextcord.Interaction):
            if interaction.user.id not in users:
                users.add(interaction.user.id)
                points[interaction.user.id] = 0
            else:
                await interaction.send("You have already joined the Quiz!",ephemeral=True)
        embed.add_field(name = "Countdown", value = f"Starting in: <t:{int(time.time() + 30)}:R>")
    view = Participate()
    return embed, view, points


class Options:
    #Adds category to api url, optional
    def category(category:str):
        api_key_dict = {"General Knowledge": 9, "Books": 10, "Films":11, "Music":12, "Musicals and Theatre":13, "Television":14, "Video Games":15,"Board Games":16, "Science & Nature":17, "Computers":18, "Mathematics":19, "Mythology":20, "Sports":21, "Geography":22,"History":23, "Politics":24, "Art":25, "Celebrities":26, "Animals":27, "Vehicles":28, "Comics":29, "Gadgets":30, "Japanese Anime & Manga":31, "Cartoon & Animations":32}
        global api_url
        api_url+="&category="+str(api_key_dict[category])
        pass

    #Adds difficulty to api url, optional
    def difficulty(difficulty:str):
        global api_url
        api_url+="&difficulty="+ difficulty.lower()
        pass

    #Required parameter, defines the number of questions
    def amount(amount:int):
        global api_url
        api_url += '?amount='+str(amount)
        pass

class Quiz:
        #Converts html special characters to readable content
        def replace(arg):
            arg = html.unescape(arg)
            return arg

        #Main Function which handles Questions
        def quiz(stuff:dict, count:int):
            category = stuff['category']
            question_type = stuff['type']
            question = stuff['question']
            question = Quiz.replace(question)
            correct_ans = stuff['correct_answer']
            incorrect_ans = stuff['incorrect_answers']

            options = list(incorrect_ans)
            options.append(correct_ans)

            if len(options) == 2:
                options = ["True","False"]
            else:
                random.shuffle(options)

            
            description = f'''**Category: {category} **\n\n'''
            embed = nextcord.Embed(title= f'Question No. {count} : {question}', description=description, color= 0x21f9fd)
            correct = options.index(correct_ans)

            for i in range(len(options)):
                options[i] = Quiz.replace(options[i])

            if question_type == 'boolean':
                embed.add_field(name = f":regional_indicator_a: {options[0]}", value= "",inline=False)
                embed.add_field(name = f":regional_indicator_b: {options[1]}", value= "",inline=False)
            else:
                embed.add_field(name = f":regional_indicator_a: {options[0]}", value= "",inline=False)
                embed.add_field(name = f":regional_indicator_b: {options[1]}", value= "",inline=False)
                embed.add_field(name = f":regional_indicator_c: {options[2]}", value= "",inline=False)
                embed.add_field(name = f":regional_indicator_d: {options[3]}", value= "",inline=False)

            return embed, options,correct, question_type, question

@bot.slash_command(name = 'start',description = "Start a Quiz!")
async def start(
    interaction: nextcord.Interaction,

    amount: int = nextcord.SlashOption(
        required=True, 
        description="Number of Questions",
    ),
    type: str = nextcord.SlashOption(
        required=True,
        description= "Type of Quiz",
        choices= ["Fastest Finger First", "Trivia"],
    ),
    question_duration: Optional[int] = nextcord.SlashOption(
        description= "Time for which a certain question will be visible before revealing the answers.",
    ),
    category: Optional[str] = nextcord.SlashOption(
        name="category",
        choices=["General Knowledge", "Books", "Films", "Music", "Musicals and Theatre", "Television", "Video Games","Board Games", "Science & Nature", "Computers", "Mathematics", "Mythology", "Sports", "Geography","History", "Politics", "Art", "Celebrities", "Animals", "Vehicles", "Comics", "Gadgets", "Japanese Anime & Manga", "Cartoon & Animations"],
        required= False,
        description="Choose Category of the Quiz.", 
    ),

    difficulty: Optional[str] = nextcord.SlashOption(
        name = "difficulty",
        choices = ["Easy","Medium","Difficult"],
        required= False,
        description="Choose Difficulty of the Quiz.", 
    ),

):
    global api_url
    api_url = "https://opentdb.com/api.php"
    if amount <0 or amount>50:
        await interaction.response.send_message("Please enter a Valid Amount.Max Questions allowed is 50")
        return
    else:
        Options.amount(amount)

    if category:
        Options.category(category)
    else:
        pass
    if difficulty:
        Options.difficulty(difficulty)
    else:
        pass

    if question_duration:
        if question_duration<5 or question_duration>100:
            await interaction.send("Duration for which Question is visible should be between 5 and 100 seconds",ephemeral=True)
            question_duration = 20
        else:
            question_duration = question_duration
    else:
        question_duration = 20
    await interaction.response.send_message(f"Starting quiz! Each Question will be visible for **{question_duration} seconds**")


    

    #Opening up the api link, and converting into readable format (json)
    webUrl = urllib.request.urlopen(api_url)
    questions = webUrl.read()

    questions = str(questions,'UTF-8')
    questions.replace("'",'"')
    my_json = json.loads(questions)

    embed,view,points = participate()
    await interaction.send(embed = embed, view= view, delete_after=duration)
    await asyncio.sleep(duration)

    
    
    count = 0
    for i in my_json['results']:
        count +=1
        embed, options,correct, question_type, question= Quiz.quiz(i, count)
        global correct_answers, incorrect_answers
        correct_answers = {}
        incorrect_answers = {}
            
        ans_embed = nextcord.Embed(title = f'🥁The CORRECT ANSWER IS 🥁', description= "\n\n",color= 0x10eb38)
        ans_embed.add_field(name = f'Question: {question} \n\n**:regional_indicator_{chr(97+correct)}: {options[correct]}**' , value = " ", inline=False)
        ans_embed.set_thumbnail(url = "https://media.tenor.com/J5a7alhbQx0AAAAC/clapping-hands.gif")

        #View Component for Question of type Fastest Finger First
        class Question_fff(nextcord.ui.View):
            def __init__(self) -> None:
                super().__init__()
                self.done = []

            if question_type == 'boolean':
                @nextcord.ui.button(label = f'{options[0]}', style=nextcord.ButtonStyle.grey, emoji="🇦")
                async def option_0(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,0)

                @nextcord.ui.button(label = f'{options[1]}', style=nextcord.ButtonStyle.grey, emoji="🇧")
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,1)
                    
            else:
                @nextcord.ui.button(label = f'{options[0]}', style=nextcord.ButtonStyle.grey, emoji="🇦")
                async def option_0(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,0)

                @nextcord.ui.button(label = f'{options[1]}', style=nextcord.ButtonStyle.grey, emoji="🇧")
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,1)

                @nextcord.ui.button(label = f'{options[2]}', style=nextcord.ButtonStyle.grey, emoji= "🇨")
                async def option_2(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,2)

                @nextcord.ui.button(label = f'{options[3]}', style=nextcord.ButtonStyle.grey, emoji= "🇩")
                async def option_3(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_fff.correct_or_not(self,interaction,3)

            #Checks's whether the button clicked by the user is the correct option or not
            async def correct_or_not(self, interaction:nextcord.Interaction, value:int):

                try:
                    points[interaction.user.id]
                except:
                    points[interaction.user.id] = 0

                if interaction.user.id not in self.done:
                    if value == correct:
                        points[interaction.user.id] += 1
                        self.stop()
                        await interaction.send(f'{interaction.user.mention} got the Correct Answer First! 🎊 🎊')
                        await interaction.message.edit(embed=ans_embed, view=Answer())
                    else:
                        await interaction.send(f'{interaction.user.mention} Incorrect! 😔', ephemeral=True)
                    self.done.append(interaction.user.id)
                else:
                    await interaction.send('You have already responded!', ephemeral=True)

            embed.set_thumbnail(url="https://clipart-library.com/images/ATbr7A4Xc.jpg")

        #View Component for Question of type Trivia
        class Question_trivia(nextcord.ui.View):
            def __init__(self) -> None:
                super().__init__()
                self.answers = {"correct": set(),"incorrect":set()}
                self.done = []
                ans_embed.add_field(name = "Correct",value="",inline=True)
                ans_embed.add_field(name="Incorrect", value="", inline=True)

            if question_type == 'boolean':
                @nextcord.ui.button(label = f'{options[0]}', style=nextcord.ButtonStyle.grey, emoji="🇦")
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_trivia.correct_or_not(self, interaction,0,interaction.user.id)
                    


                @nextcord.ui.button(label = f'{options[1]}', style=nextcord.ButtonStyle.grey, emoji="🇧")
                async def option_2(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_trivia.correct_or_not(self, interaction,1,interaction.user.id)
                    
            else:
                @nextcord.ui.button(label = f'{options[0]}', style=nextcord.ButtonStyle.grey, emoji="🇦")
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    await Question_trivia.correct_or_not(self, interaction,0,interaction.user.id)
                    

                @nextcord.ui.button(label = f'{options[1]}', style=nextcord.ButtonStyle.grey, emoji="🇧")
                async def option_2(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    
                    await Question_trivia.correct_or_not(self, interaction,1,interaction.user.id)

                @nextcord.ui.button(label = f'{options[2]}', style=nextcord.ButtonStyle.grey, emoji= "🇨")
                async def option_3(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    
                    await Question_trivia.correct_or_not(self, interaction,2,interaction.user.id)

                @nextcord.ui.button(label = f'{options[3]}', style=nextcord.ButtonStyle.grey, emoji= "🇩")
                async def option_4(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    
                    await Question_trivia.correct_or_not(self, interaction,3,interaction.user.id)

            #Checks's whether the button clicked by the user is the correct option or not
            async def correct_or_not(self, interaction:nextcord.Interaction,value:int,id):
                try:
                    points[id]
                except KeyError:
                    points[id] = 0
                    pass
                    #await interaction.send(f'{interaction.user.mention}, You have not joined the Quiz, please wait for the next round.', ephemeral=True)
                    #return 
                
                if id not in self.done:
                    if value == correct:
                        points[id] +=1
                        self.answers["correct"].add(id)
                    else:
                        self.answers["incorrect"].add(id)
                    self.done.append(id)
                elif id in self.done:
                    await interaction.send("You have already answered!",ephemeral=True)
                    return
                global correct_answers, incorrect_answers
                correct_answers = self.answers["correct"]
                incorrect_answers = self.answers["incorrect"]
            
           


            embed.set_thumbnail(url="https://clipart-library.com/images/ATbr7A4Xc.jpg")
        
        #View component for Answer (sent after duration between questions)
        class Answer(nextcord.ui.View):
            style = [nextcord.ButtonStyle.grey, nextcord.ButtonStyle.grey, nextcord.ButtonStyle.grey, nextcord.ButtonStyle.grey]
            style[correct] = nextcord.ButtonStyle.green

            def __init__(self) -> None:
                super().__init__()


            if question_type == 'boolean':
                @nextcord.ui.button(label = f'{options[0]}', style= style[0] ,emoji="🇦", disabled=True)
                async def option_0(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass

                @nextcord.ui.button(label = f'{options[1]}',  style= style[1] ,emoji="🇧",disabled=True)
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass
            else:
                @nextcord.ui.button(label = f'{options[0]}', style= style[0] ,emoji="🇦", disabled=True)
                async def option_0(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass

                @nextcord.ui.button(label = f'{options[1]}', style= style[1] ,emoji="🇧", disabled=True)
                async def option_1(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass

                @nextcord.ui.button(label = f'{options[2]}', style= style[2] ,emoji= "🇨", disabled=True)
                async def option_2(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass

                @nextcord.ui.button(label = f'{options[3]}', style= style[3] ,emoji= "🇩", disabled=True)
                async def option_3(self, button: nextcord.Button, interaction:nextcord.Interaction):
                    pass

        embed.add_field(name="Countdown",value=f"Ending in: <t:{int(time.time() + question_duration)}:R>")
        if type == "Fastest Finger First":
            await interaction.send(embed= embed, view=Question_fff(), delete_after= question_duration-1)
            await asyncio.sleep(question_duration-1)
            await interaction.send(embed=ans_embed, view=Answer())
        elif type == "Trivia":
            await interaction.send(embed = embed,view = Question_trivia(), delete_after=question_duration-1)
            await asyncio.sleep(question_duration-1)
            incorrect_value = ""
            correct_value = ""
            if (len(correct_answers)+len(incorrect_answers) != 0):
                for i in correct_answers:
                    correct_value += f'<@{i}>\n'
                for t in incorrect_answers:
                    incorrect_value += f'<@{t}>\n'
                if correct_value == "":
                    ans_embed.set_field_at(name="Correct",index=-2, value= "No one got it correct! 😔")
                else:
                    ans_embed.set_field_at(name="Correct",index=-2, value= correct_value)
                if incorrect_value == "":
                    ans_embed.set_field_at(name="Incorrect",index=-1, value= "No incorrect answers! 🎊 🎊")
                else:
                    ans_embed.set_field_at(name="Incorrect",index=-1, value= incorrect_value)
            else:
                ans_embed.set_field_at(name="Correct",index=-2, value= "No Responses")
                ans_embed.set_field_at(name="Incorrect",index=-1, value= "No Responses")
                pass
            await interaction.send(embed=ans_embed, view=Answer())
    leaderboard = nextcord.Embed(title= " 🏆 Leaderboard 🏆", color= 0x00ffdc)

    #winners = sorted(points.keys(), key= lambda x:x[1], reverse=True)
    points_  = sorted(points.items(), key = lambda x:x[1], reverse=True)


    i = 0
    while (i<10):
        emoji = ""
        if i+1 <=3:
            if i+1 == 1:
                emoji = '🥇 '
            elif i+1 == 2:
                emoji = '🥈 '
            elif i+1 == 3:
                emoji = '🥉 '
        try:
            leaderboard.add_field(name = f'{emoji}#{i+1}', value= f"<@{points_[i][0]}> with **{points_[i][1]} POINTS!**\n", inline= False)
        except:
            if i == 0:
                leaderboard.add_field(name=f'Sadly no one joined the Quiz 😔', value="\n")
            break
        i+=1
    await interaction.send(embed=leaderboard)

    exit


bot.run(config_quiz_bot.quiz_bot_token)