from __future__ import annotations
from discord.ui import Modal, TextInput
from httpx import AsyncClient
from discord import Interaction, Color
from prisma import Prisma
from prisma.models import User
from typing import TYPE_CHECKING, Literal
if TYPE_CHECKING:
from .TossupButtons import TossupButtons, SoloTossupButtons
from discord.ui.button import Button
from common.types import category_field_translations
class Answer(Modal, title="Submit Answer"):
def __init__(self, correct_answer: str, view: TossupButtons) -> None:
self.correct_answer = correct_answer
self.view = view
super().__init__()
answer = TextInput(label="Answer", placeholder="Your answer here")
async def on_submit(self, interaction: Interaction) -> None:
c = AsyncClient()
db = Prisma()
answer_check_resp = await c.get(
"https://qbreader.org/api/check-answer",
params={
"answerline": self.correct_answer,
"givenAnswer": self.answer.value,
},
)
answer_check_data = answer_check_resp.json()
await db.connect()
if answer_check_data["directive"] == "accept":
if interaction.message is None:
return
e = interaction.message.embeds[0]
e.color = Color.green()
e.title = '[CORRECT!] Random Tossup'
e.description = ".".join(self.view.tossup['sentences'][0:self.view.i+1]) + " **(BUZZ)** " + ".".join(self.view.tossup['sentences'][self.view.i+1:])
e.add_field(name='Correct answer', value=self.answer.value, inline=True)
e.add_field(name='Official answer', value=self.view.tossup['answer'], inline=True)
e.add_field(name='Answered by', value=interaction.user.mention, inline=True)
items = self.view.children
for item in items:
if isinstance(item, Button):
item.disabled = True
await interaction.response.edit_message(embed=e, view=self.view)
await db.user.upsert(where={'id': interaction.user.id}, data={
'create': {'questions_correct': 1, 'id': interaction.user.id, 'questions_incorrect': 0},
'update': {'questions_correct': {'increment': 1}}
})
category = category_field_translations[self.view.tossup['category']]
await db.execute_raw(f"""
INSERT INTO CategoryBreakdown(userId, {category}_correct) VALUES (?, 0)
ON CONFLICT(userId) DO UPDATE SET {category}_correct={category}_correct+1
""", interaction.user.id)
elif answer_check_data['directive'] == 'prompt':
await interaction.response.send_message("Prompt! Try answering the question again", ephemeral=True)
else:
if interaction.message is None:
return
e = interaction.message.embeds[0]
self.view.already_answered.append(interaction.user.id)
e.add_field(name="Incorrect answer", value=self.answer.value, inline=False)
await interaction.response.edit_message(embed=e)
await interaction.followup.send(f"Incorrect! You've been locked out from the question. The correct answer was {self.view.tossup['answer']}", ephemeral=True)
await db.user.upsert(
where={'id': interaction.user.id},
data={
'create': {'questions_correct': 0, 'id': interaction.user.id, 'questions_incorrect': 1},
'update': {'questions_incorrect': {'increment': 1}}
}
)
category = category_field_translations[self.view.tossup['category']]
await db.execute_raw(f"""
INSERT INTO CategoryBreakdown(userId, {category}_incorrect) VALUES (?, 0)
ON CONFLICT(userId) DO UPDATE SET {category}_incorrect={category}_incorrect+1
""", interaction.user.id)
await db.disconnect()
await c.aclose()
class SoloAnswer(Modal, title="Submit Answer"):
def __init__(self, correct_answer: str, view: SoloTossupButtons) -> None:
self.correct_answer = correct_answer
self.view = view
super().__init__()
answer = TextInput(label="Answer", placeholder="your answer here!")
async def on_submit(self, interaction: Interaction) -> None:
c = AsyncClient()
answer_check_resp = await c.get(
"https://qbreader.org/api/check-answer",
params={
"answerline": self.correct_answer,
"givenAnswer": self.answer.value,
},
)
answer_check_data = answer_check_resp.json()
if answer_check_data["directive"] == "accept":
if interaction.message is None:
return
e = interaction.message.embeds[0]
e.color = Color.green()
e.title = '[CORRECT!] Random Tossup'
e.description = ".".join(self.view.tossup['sentences'][0:self.view.i+1]) + " **(BUZZ)** " + ".".join(self.view.tossup['sentences'][self.view.i+1:])
e.add_field(name='Your answer', value=self.answer.value)
e.add_field(name='Official answer', value=self.view.tossup['answer'])
items = self.view.children
for item in items:
if isinstance(item, Button):
item.disabled = True
await interaction.response.edit_message(embed=e, view=self.view)
else:
if interaction.message is None:
return
e = interaction.message.embeds[0]
e.color = Color.red()
e.title = '[INCORRECT] Random Tossup'
e.description = ".".join(self.view.tossup['sentences'][0:self.view.i+1]) + " **(BUZZ)** " + ".".join(self.view.tossup['sentences'][self.view.i+1:])
e.add_field(name='Correct answer was...', value=self.view.tossup['answer'])
e.add_field(name='Your answer', value=self.answer.value)
items = self.view.children
for item in items:
if isinstance(item, Button):
item.disabled = True
await interaction.response.edit_message(embed=e, view=self.view)
await c.aclose()