Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Attempting to optimize cog

Browse Source
This commit is contained in:
unknown 2023-03-13 13:10:07 +01:00
parent c67b755075
commit 76a7c74a0c
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
recruitmentCog/recruitment.py
View File

@ -18,17 +18,19 @@ class Recruitment(commands.Cog):
self.bot = bot
self.message: str = ''
@commands.group(name="application", usage="[text]", invoke_without_command=True)
async def application(self, ctx: commands.Context, *, _application: str = ""):
author = ctx.author
if await self.is_direct_message(ctx):
await self.interactive_application(author)
# Check if the command was sent in a direct message to the bot
async def is_direct_message(self, ctx: commands.Context) -> bool:
if not isinstance(ctx.channel, discord.DMChannel):
await ctx.send("This command can only be used in a direct message to the bot.")
return
await self.interactive_application(author)
return False
return True
async def interactive_application(self, author: discord.Member):
"""Ask the user several questions to create an application."""
@ -39,30 +41,39 @@ class Recruitment(commands.Cog):
await author.send("You took too long to answer. Please try again later.")
return
embed = await self.format_application(answers, author)
embeddedApplication = await self.format_application(answers, author)
# Send the embed to the application channel
application_channel = self.bot.get_channel(application_channel_id)
message = await application_channel.send(embed=embed)
# Call the sendApplication to check if the author is a member of the guild and send the application if they are
await self.sendApplication(author, embeddedApplication)
# Add reactions to the message
reactions = ["✅", "❌", "❓"]
for reaction in reactions:
await message.add_reaction(reaction)
async def sendApplication(self, author: discord.Member, embeddedApplication: discord.Embed):
# Check if the author is a member of the guild
guild = self.bot.get_guild(274657393936302080)
guild = self.bot.get_guild(guild_id)
member = guild.get_member(author.id)
if member is None:
await author.send("You need to join the server before your application can be processed.")
return
# Send the embed to the application channel
application_channel = self.bot.get_channel(application_channel_id)
message = await application_channel.send(embed=embeddedApplication)
# Add reactions to the message
await self.add_reactions(message)
# Assign the Trial role to the author
role = guild.get_role(531181363420987423)
await member.add_roles(role)
await author.send("Thank you for submitting your application! You have been given the 'Trial' role.")
async def add_reactions(self, message: discord.Message):
reactions = ["✅", "❌", "❓"]
for reaction in reactions:
await message.add_reaction(reaction)
async def format_application(self, answers: List[str], author: discord.Member) -> discord.Embed:
"""Format the application answers into an embed."""
embed = discord.Embed(title=f"Application from {author.display_name}", color=discord.Color.green())
@ -80,12 +91,12 @@ class Recruitment(commands.Cog):
for question in questions:
await author.send(question)
answer = await self.get_answer(author)
answer = await self.get_answers(author)
answers.append(answer.content)
return answers
async def get_answer(self, author: discord.Member) -> discord.Message:
async def get_answers(self, author: discord.Member) -> discord.Message:
"""Wait for the user to send a message."""
return await self.bot.wait_for(
"message", check=lambda m: m.author == author and m.guild is None