diff options
| -rw-r--r-- | cogs/stats.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/cogs/stats.py b/cogs/stats.py index 575dab5..54037c2 100644 --- a/cogs/stats.py +++ b/cogs/stats.py @@ -1,8 +1,9 @@ from discord.ext.commands import Cog, Bot -from discord.app_commands import command -from discord import Interaction, Embed, Color +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()} @@ -12,15 +13,16 @@ class Stats(Cog): super().__init__() @command(description="Get your statistics for qbb") - async def stats(self, ctx: Interaction): + async def stats(self, ctx: Interaction, user: Optional[User] = None): db = Prisma() await db.connect() - stats = await db.user.find_first(where={'id': ctx.user.id}) - cb = await db.categorybreakdown.find_first(where={'userId': ctx.user.id}) + 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="You have no stats! Trying using the bot and then running this command", color=Color.red()) + 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="Your Stats!", description=f"""**Number of correct tossups:** {stats.questions_correct} + embed = Embed(title="@{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(): |