1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
from random import choice, randrange
from time import perf_counter
from typing import Union
import discord
from discord import Embed, Interaction
from discord.ui import Button, View
from bot.constants import Colours, NEGATIVE_REPLIES
from ._scoreboard import Scoreboard
class QuestionButton(Button):
"""Button subclass for the options of the questions."""
def __init__(self, label: str, users_picked: dict):
self.users_picked = users_picked
super().__init__(label=label, style=discord.ButtonStyle.green)
def set_time(self) -> None:
"""Sets an instance attribute to a perf counter simulating the question beginning."""
self._time = perf_counter()
async def callback(self, interaction: Interaction) -> None:
"""When a user interacts with the button, this will be called."""
if interaction.user.id not in self.users_picked.keys():
self.users_picked[interaction.user.id] = [self.label, 1, perf_counter() - self._time]
elif self.users_picked[interaction.user.id][1] < 2:
self.users_picked[interaction.user.id] = [
self.label, self.users_picked[interaction.user.id][1] + 1, perf_counter() - self._time
]
else:
await interaction.response.send_message(
embed=Embed(
title=choice(NEGATIVE_REPLIES),
description="You've already changed your answer more than once!",
color=Colours.soft_red
),
ephemeral=True
)
class QuestionView(View):
"""View for the questions."""
def __init__(self):
super().__init__()
self.current_question = {}
self.users_picked = {}
self.buttons = [QuestionButton(label, self.users_picked) for label in ("A", "B", "C", "D")]
for button in self.buttons:
self.add_item(button)
def create_current_question(self) -> Embed:
"""Helper function to create the embed for the current question."""
question_embed = Embed(
title=f"Question {self.current_question['number']}",
description=self.current_question["description"],
color=Colours.python_yellow
)
for label, answer in zip(("A", "B", "C", "D"), self.current_question["answers"]):
question_embed.add_field(name=f"Choice {label}", value=answer, inline=False)
for button in self.buttons:
button.set_time()
return question_embed
def end_question(self) -> tuple[dict, Embed]:
"""Returns the dictionaries from the corresponding buttons for those who got it correct."""
labels = ("A", "B", "C", "D")
label = labels[self.current_question["answers"].index(self.current_question["correct"])]
return_dict = {name: info for name, info in self.users_picked.items() if info[0] == label}
self.users_picked = {}
for button in self.buttons:
button.users_picked = self.users_picked
answer_embed = Embed(
title=f"The correct answer for Question {self.current_question['number']} was",
description=self.current_question["correct"],
color=Colours.soft_green
)
return return_dict, answer_embed
class Questions:
"""An interface to use from the TriviaNight cog for questions."""
def __init__(self, scoreboard: Scoreboard):
self.scoreboard = scoreboard
self.questions = []
self._ptr = -1
def set_questions(self, questions: list) -> None:
"""Setting `self.questions` dynamically via a function to set it."""
self.questions = questions
def next_question(self) -> Union[Embed, None]:
"""Uses another, new question."""
if all("visited" in question.keys() for question in self.questions.values()):
return Embed(
title=choice(NEGATIVE_REPLIES),
description="All of the questions in the question bank have been used.",
color=Colours.soft_red
)
while "visited" in self.questions[self._ptr].keys():
self._ptr = randrange(0, len(self.questions))
self.questions[self._ptr]["visited"] = True
self.view.current_question = self.questions[self._ptr]
def current_question(self) -> tuple[Embed, QuestionView]:
"""Returns an embed entailing the current question as an embed with a view."""
return self.view.create_current_question(), self.view
def end_question(self) -> Embed:
"""Terminates answering of the question and displays the correct answer."""
scores, answer_embed = self.view.end_question()
for user, score in scores.items():
self.scoreboard[f"points: {user}"] = score[1]
self.scoreboard[f"speed: {user}"] = score[2]
return answer_embed
|
