From 6cf0d77ef775b24020e857ee27503d1ec73dc69c Mon Sep 17 00:00:00 2001 From: EEKIM10 Date: Tue, 4 Oct 2022 15:06:43 +0100 Subject: [PATCH] Add B number fetch --- cogs/verify.py | 16 ++++++++++++++++ utils/_email.py | 2 +- 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/cogs/verify.py b/cogs/verify.py index 21b9959..8132f52 100644 --- a/cogs/verify.py +++ b/cogs/verify.py @@ -114,6 +114,22 @@ class VerifyCog(commands.Cog): "\N{white heavy check mark} Verification complete!", ) + @commands.user_command(name="B Number") + @discord.guild_only() + async def get_b_number(self, ctx: discord.ApplicationContext, member: discord.Member): + try: + student: Student = await Student.objects.get(user_id=ctx.author.id) + return await ctx.respond( + f"{member.mention}'s B number is saved as {student.id!r}.", + allowed_mentions=discord.AllowedMentions.none() + ) + except orm.NoMatch: + pass + return await ctx.respond( + f"{member.mention} has no saved B number.", + allowed_mentions=discord.AllowedMentions.none() + ) + def setup(bot): bot.add_cog(VerifyCog(bot)) diff --git a/utils/_email.py b/utils/_email.py index c0b5fc0..08ff1f9 100644 --- a/utils/_email.py +++ b/utils/_email.py @@ -13,7 +13,7 @@ async def send_verification_code(user: discord.User, student_number: str, **kwar """Sends a verification code, returning said verification code, to the student.""" code = secrets.token_hex(16) text = ( - f"Hey {user} ({student_number})! The code to join the hi^5 code is '{code}' - use " + f"Hey {user} ({student_number})! The code to join the Unscrupulous Nonsense is '{code}' - use " f"'/verify {code}' in the bot's DMs to continue \N{dancer}\n\n~nex" ) msg = EmailMessage()