summary refs log tree commit diff stats
diff options
context:
space:
mode:
Diffstat
-rw-r--r--.env2
-rw-r--r--bot.py41
-rw-r--r--cogs/__pycache__/solo.cpython-311.pycbin0 -> 3787 bytes
-rw-r--r--cogs/__pycache__/tossup.cpython-311.pycbin0 -> 3805 bytes
-rw-r--r--cogs/solo.py41
-rw-r--r--cogs/tossup.py42
-rw-r--r--common/__pycache__/types.cpython-311.pycbin0 -> 403 bytes
-rw-r--r--common/types.py16
-rw-r--r--components/AnswerModal.py85
-rw-r--r--components/TossupButtons.py99
-rw-r--r--components/__pycache__/AnswerModal.cpython-311.pycbin0 -> 7397 bytes
-rw-r--r--components/__pycache__/TossupButtons.cpython-311.pycbin0 -> 8765 bytes
-rw-r--r--test.py22
13 files changed, 348 insertions, 0 deletions
diff --git a/.env b/.env
new file mode 100644
index 0000000..acab8b5
--- /dev/null
+++ b/.env
@@ -0,0 +1,2 @@
+TOKEN=ODMyMjU1MjU4NDAyNjE5Mzky.GLf39U.JmrRbXvobEvpRzG1f9pUd6lvou2xhgKtThQzUk
+OWNER_ID=435206857276260353
diff --git a/bot.py b/bot.py
new file mode 100644
index 0000000..9240d32
--- /dev/null
+++ b/bot.py
@@ -0,0 +1,41 @@
+import discord
+from discord import Interaction, Intents, Embed
+from discord.ext.commands import Bot, Context, when_mentioned
+from dotenv import load_dotenv
+from os import getenv
+from sys import exit
+from typing import Literal
+import httpx
+import nltk
+import asyncio
+
+load_dotenv()
+
+if getenv('TOKEN') is None or getenv('OWNER_ID') is None:
+    print('please set a TOKEN and OWNER_ID environment variable')
+    exit(1)
+
+class QBBBot(Bot):
+    def __init__(self) -> None:
+        super().__init__(command_prefix=when_mentioned, intents=Intents.default())
+
+    async def setup_hook(self):
+        await self.load_extension('cogs.tossup')
+        await self.load_extension('cogs.solo')
+        await self.load_extension('jishaku')
+
+
+bot = QBBBot()
+bot.qb_categories = Literal['Literature', 'History', 'Science', 'Fine Arts', 'Religion', 'Mythology', 'Philosophy',
+                            'Social Science', 'Current Events', 'Geography', 'Other Academic', 'Trash']
+
+
+@bot.command()
+async def sync(ctx: Context):
+    if str(ctx.author.id) == getenv('OWNER_ID'):
+        await bot.tree.sync()
+        await ctx.send('ok')
+    else:
+        await ctx.send('no')
+
+bot.run(getenv('TOKEN'))
diff --git a/cogs/__pycache__/solo.cpython-311.pyc b/cogs/__pycache__/solo.cpython-311.pyc
new file mode 100644
index 0000000..6ede884
--- /dev/null
+++ b/cogs/__pycache__/solo.cpython-311.pyc
Binary files differdiff --git a/cogs/__pycache__/tossup.cpython-311.pyc b/cogs/__pycache__/tossup.cpython-311.pyc
new file mode 100644
index 0000000..08b770b
--- /dev/null
+++ b/cogs/__pycache__/tossup.cpython-311.pyc
Binary files differdiff --git a/cogs/solo.py b/cogs/solo.py
new file mode 100644
index 0000000..aab0b77
--- /dev/null
+++ b/cogs/solo.py
@@ -0,0 +1,41 @@
+from discord.ext.commands import GroupCog, Bot
+from discord.ui import View, Button, Modal, button, TextInput
+from discord.app_commands import command
+from discord import Interaction, Embed, User, Member, ButtonStyle, Color
+from httpx import AsyncClient
+from nltk import sent_tokenize
+from typing import Union, Optional
+from common.types import question_category
+from components.TossupButtons import SoloTossupButtons
+
+
+class Solo(GroupCog, name="solo"):
+    def __init__(self, bot: Bot) -> None:
+        self.bot = bot
+        super().__init__()
+
+    @command(description="Do a tossup in solo mode (only you can control the tossup)")
+    async def tossup(self, ctx: Interaction, category: Optional[question_category] = None):
+        c = AsyncClient()
+        params = {'difficulties': [2,3,4,5]}
+        if category is not None:
+            params['categories'] = category
+
+        req = await c.get(
+            "https://qbreader.org/api/random-tossup", params=params
+        )
+        tossup: dict = req.json()["tossups"][0]
+        tossup['sentences'] = await self.bot.loop.run_in_executor(None, sent_tokenize, tossup['question'])
+
+
+        view: SoloTossupButtons = SoloTossupButtons(tossup, ctx.user)
+        embed = Embed(title="Random Tossup", description=tossup["sentences"][0])
+        embed.set_author(
+            name=f"{tossup['set']['name']} Packet {tossup['packetNumber']} Question {tossup['questionNumber']}"
+        )
+        embed.set_footer(text="Questions obtained from qbreader.org")
+        await ctx.response.send_message(embed=embed, view=view)
+        await c.aclose()
+
+async def setup(bot: Bot) -> None:
+    await bot.add_cog(Solo(bot))
diff --git a/cogs/tossup.py b/cogs/tossup.py
new file mode 100644
index 0000000..ae01519
--- /dev/null
+++ b/cogs/tossup.py
@@ -0,0 +1,42 @@
+from discord.ext.commands import Cog, Bot
+from discord.app_commands import command, describe
+from discord import Interaction, Embed, ButtonStyle
+from discord.ui import View, Button, Modal, button, TextInput
+from nltk import sent_tokenize
+from httpx import AsyncClient
+from typing import Literal, Optional
+from common.types import question_category
+from components.AnswerModal import Answer
+from components.TossupButtons import TossupButtons
+
+class Tossup(Cog):
+    def __init__(self, bot: Bot) -> None:
+        self.bot = bot
+
+    @command(name="tossup", description="gives a random tossup")
+    @describe(category="The category to choose the question from (optional)")
+    async def tossup(self, ctx: Interaction, category: Optional[question_category] = None):
+        c = AsyncClient()
+
+        params = {"difficulties": [2, 3, 4, 5]}
+        if category is not None:
+            params["categories"] = category
+
+        req = await c.get("https://qbreader.org/api/random-tossup", params=params)
+        tossup: dict = req.json()["tossups"][0]
+        tossup["sentences"] = await self.bot.loop.run_in_executor(
+            None, sent_tokenize, tossup["question"]
+        )
+
+        view: TossupButtons = TossupButtons(tossup)
+        embed = Embed(title="Random Tossup", description=tossup["sentences"][0])
+        embed.set_author(
+            name=f"{tossup['set']['name']} Packet {tossup['packetNumber']} Question {tossup['questionNumber']} (Category: {tossup['category']})"
+        )
+        embed.set_footer(text="Questions obtained from qbreader.org")
+        await ctx.response.send_message(embed=embed, view=view)
+        await c.aclose()
+
+
+async def setup(bot: Bot) -> None:
+    await bot.add_cog(Tossup(bot))
diff --git a/common/__pycache__/types.cpython-311.pyc b/common/__pycache__/types.cpython-311.pyc
new file mode 100644
index 0000000..3d7bbd1
--- /dev/null
+++ b/common/__pycache__/types.cpython-311.pyc
Binary files differdiff --git a/common/types.py b/common/types.py
new file mode 100644
index 0000000..e9716b1
--- /dev/null
+++ b/common/types.py
@@ -0,0 +1,16 @@
+from typing import Literal
+
+question_category = Literal[
+    "Literature",
+    "History",
+    "Science",
+    "Fine Arts",
+    "Religion",
+    "Mythology",
+    "Philosophy",
+    "Social Science",
+    "Current Events",
+    "Geography",
+    "Other Academic",
+    "Trash",
+]
diff --git a/components/AnswerModal.py b/components/AnswerModal.py
new file mode 100644
index 0000000..9bd5f17
--- /dev/null
+++ b/components/AnswerModal.py
@@ -0,0 +1,85 @@
+from discord.ui import Modal, TextInput, View
+from httpx import AsyncClient
+from discord import Interaction, Color
+
+
+class Answer(Modal, title="Submit Answer"):
+    def __init__(self, correct_answer: str, view: View) -> 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":
+            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'])
+            e.add_field(name='Answered by', value=interaction.user.mention)
+            items = self.view.children
+            for item in items:
+                item.disabled = True
+            await interaction.response.edit_message(embed=e, view=self.view)
+        elif answer_check_data['directive'] == 'prompt':
+            await interaction.response.send_message("Prompt! Try answering the question again", ephemeral=True)
+        else:
+            await interaction.response.send_message(f"Incorrect! You've been locked out from the question. The correct answer was {self.view.tossup['answer']}", ephemeral=True)
+            self.view.already_answered.append(interaction.user.id)
+        await c.aclose()
+
+class SoloAnswer(Modal, title="Submit Answer"):
+    def __init__(self, correct_answer: str, view: View) -> 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":
+            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:
+                item.disabled = True
+            await interaction.response.edit_message(embed=e, view=self.view)
+        else:
+            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:
+                item.disabled = True
+            await interaction.response.edit_message(embed=e, view=self.view)
+        await c.aclose()
diff --git a/components/TossupButtons.py b/components/TossupButtons.py
new file mode 100644
index 0000000..7238cac
--- /dev/null
+++ b/components/TossupButtons.py
@@ -0,0 +1,99 @@
+from discord.ui import View, button, Button
+from discord import ButtonStyle, Interaction, User, Member, Color
+from .AnswerModal import Answer, SoloAnswer
+from typing import Union
+
+class TossupButtons(View):
+    def __init__(self, tossup) -> None:
+        self.tossup = tossup
+        self.already_answered = []
+        self.already_voted = []
+        self.i = 0
+        self.final_answer_votes = 0
+        super().__init__()
+
+    @button(label="Answer!", style=ButtonStyle.green)
+    async def answer(self, interaction: Interaction, button: Button):
+        if interaction.user.id not in self.already_answered:
+            await interaction.response.send_modal(
+                Answer(self.tossup.get('formatted_answer', self.tossup['answer']), self)
+            )
+        else:
+            await interaction.response.send_message(
+                "you've already answered!", ephemeral=True
+            )
+
+    @button(label="Add sentence", style=ButtonStyle.blurple)
+    async def add_sentence(self, interaction: Interaction, button: Button):
+        if interaction.user.id in self.already_answered:
+            return await interaction.response.send_message(
+                "you've already answered!", ephemeral=True
+            )
+        e = interaction.message.embeds[0]
+        self.i += 1
+        if self.i == len(self.tossup["sentences"]):
+            button.disabled = True
+        e.description = "\n".join(self.tossup["sentences"][0 : self.i + 1])
+        await interaction.response.edit_message(embed=e, view=self)
+
+    @button(label="Vote to Reveal Answer (0/3)", style=ButtonStyle.red)
+    async def reveal_answer(self, interaction: Interaction, button: Button):
+        if interaction.user.id in self.already_voted:
+            return await interaction.response.send_message(
+                "you've already voted!", ephemeral=True
+            )
+        self.already_voted.append(interaction.user.id)
+        self.final_answer_votes += 1
+        button.label = f"vote to reveal answer ({self.final_answer_votes}/3)"
+        if self.final_answer_votes >= 3:
+            e = interaction.message.embeds[0]
+            e.title = '[SKIPPED] Random Tossup'
+            e.color = Color.orange()
+            e.description = self.tossup["question"]
+            e.add_field(name="Answer", value=self.tossup["answer"])
+            for item in self.children:
+                item.disabled = True
+            return await interaction.response.edit_message(embed=e, view=self)
+        await interaction.response.edit_message(view=self)
+        await interaction.followup.send("you've voted!", ephemeral=True)
+
+class SoloTossupButtons(View):
+    def __init__(self, tossup, user: Union[User, Member]) -> None:
+        self.user = user
+        self.tossup = tossup
+        self.i = 0
+        super().__init__()
+
+    @button(label="Answer!", style=ButtonStyle.green)
+    async def answer(self, interaction: Interaction, button: Button):
+        if interaction.user.id != self.user.id:
+            return interaction.response.send_message('This is not your tossup!', ephemeral=True)
+        await interaction.response.send_modal(SoloAnswer(self.tossup.get('formatted_answer', self.tossup['answer']), self))
+
+    @button(label="Add sentence", style=ButtonStyle.blurple)
+    async def add_sentence(self, interaction: Interaction, button: Button):
+        if interaction.user.id != self.user.id:
+            return await interaction.response.send_message(
+                "not your tossup!", ephemeral=True
+            )
+        e = interaction.message.embeds[0]
+        self.i += 1
+        if self.i == len(self.tossup["sentences"]) - 1:
+            button.disabled = True
+        e.description = "\n".join(self.tossup["sentences"][0 : self.i + 1])
+        await interaction.response.edit_message(embed=e, view=self)
+
+    @button(label="Reveal Answer", style=ButtonStyle.red)
+    async def reveal_answer(self, interaction: Interaction, button: Button):
+        if interaction.user.id != self.user.id:
+            return await interaction.response.send_message(
+                "not your tossup!", ephemeral=True
+            )
+        e = interaction.message.embeds[0]
+        e.title = f'[SKIPPED] Random Tossup'
+        e.color = Color.orange()
+        e.description = self.tossup["question"]
+        e.add_field(name="Answer", value=self.tossup["answer"])
+        for item in self.children:
+            item.disabled = True
+        return await interaction.response.edit_message(embed=e, view=self)
diff --git a/components/__pycache__/AnswerModal.cpython-311.pyc b/components/__pycache__/AnswerModal.cpython-311.pyc
new file mode 100644
index 0000000..0ca3cb1
--- /dev/null
+++ b/components/__pycache__/AnswerModal.cpython-311.pyc
Binary files differdiff --git a/components/__pycache__/TossupButtons.cpython-311.pyc b/components/__pycache__/TossupButtons.cpython-311.pyc
new file mode 100644
index 0000000..fc443d0
--- /dev/null
+++ b/components/__pycache__/TossupButtons.cpython-311.pyc
Binary files differdiff --git a/test.py b/test.py
new file mode 100644
index 0000000..19a51fa
--- /dev/null
+++ b/test.py
@@ -0,0 +1,22 @@
+from bs4 import BeautifulSoup, Tag
+import httpx as h
+from tabulate import tabulate
+
+r = h.get('https://www.naqt.com/stats/school/players.jsp?org_id=69304')
+
+html = BeautifulSoup(r.text, 'lxml')
+
+tbl = html.find('section', attrs={'id': 'players'}).table
+
+headers = [e.text for e in tbl.thead.tr.find_all('th')]
+
+data = []
+
+elem: Tag
+for elem in tbl.tbody.find_all('tr'):
+    temp = []
+    for z in elem.find_all():
+        temp.append(z.text)
+    data.append(temp[1:])
+
+print(tabulate(data, headers, showindex=False))
generated by cgit-pink 1.4.1-2-gfad0 (git 2.36.2.497.gbbea4dcf42) at 2024-11-22 07:51:44 +0000