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
|
from discord.ext.commands import Cog, Bot
from discord.app_commands import command
from discord import Interaction, Embed, Color, User
from prisma import Prisma
from common.types import category_field_translations
from typing import Optional
reverse_categories = {v: k for k, v in category_field_translations.items()}
class Stats(Cog):
def __init__(self, bot: Bot) -> None:
self.bot = bot
super().__init__()
@command(description="Get your statistics for qbb")
async def stats(self, ctx: Interaction, user: Optional[User] = None):
db = Prisma()
await db.connect()
person = user if user is not None else ctx.user
stats = await db.user.find_first(where={'id': person.id})
cb = await db.categorybreakdown.find_first(where={'userId': person.id})
if stats is None or cb is None:
embed = Embed(title="No Stats!", description=f"@{person.name} has no stats! Trying using the bot and then running this command", color=Color.red())
return await ctx.response.send_message(embed=embed)
embed = Embed(title=f"@{person.name}'s qbb stats", description=f"""**Number of correct tossups:** {stats.questions_correct}
**Number of incorrect tossups:** {stats.questions_incorrect}
""")
for cat in category_field_translations.values():
corr = getattr(cb, f'{cat}_correct')
incorr = getattr(cb, f'{cat}_incorrect')
embed.add_field(name=reverse_categories[cat], value=f'{corr+incorr} heard ({corr} correct)')
await ctx.response.send_message(embed=embed)
await db.disconnect()
async def setup(bot: Bot) -> None:
await bot.add_cog(Stats(bot))
|