import asyncio from typing import Union, List from datetime import timedelta from copy import copy import contextlib import discord from redbot.core import Config, checks, commands from redbot.core.utils import AsyncIter from redbot.core.utils.chat_formatting import pagify, box from redbot.core.utils.antispam import AntiSpam from redbot.core.bot import Red from redbot.core.utils.predicates import MessagePredicate application_channel_id = 1023172488143839252 class Recruitment(commands.Cog): """A cog that lets a user send a membership application. Users can open an application using `[!]apply`. These are then sent to a channel in the server for staff, and the application creator gets a DM. Both can be used to communicate. """ def __init__(self, bot): self.bot = bot self.message: str = '' self.active = True @commands.group(name="application", usage="[text]", invoke_without_command=True) async def application(self, ctx: commands.Context, *, _application: str = ""): """Send an application. Use without arguments for an interactive application creation flow, or do `[p]application [text]` to use it non-interactively. """ author = ctx.author # Check if the command was sent in a direct message to the bot if not isinstance(ctx.channel, discord.DMChannel): await ctx.send("This command can only be used in a direct message to the bot.") return # If there is no text argument, use an interactive flow if not _application: # Ask the user the questions and get the answers questions = ["What's your name?", "What's your age?", "Why do you want to join our community?"] answers = await ask_questions(self.bot, author, questions) # Create an embed with the application information application_embed = await format_application(answers, author) # Send the embed to the application channel application_channel = self.bot.get_channel(application_channel_id) await application_channel.send(embed=application_embed) # Send a confirmation message to the author await author.send("Thank you for submitting your application!") else: # If there is a text argument, use a non-interactive flow application_channel = self.bot.get_channel(application_channel_id) await application_channel.send(_application) await author.send("Thank you for submitting your application!") async def format_application(answers: List[str], author: discord.Member) -> discord.Embed: embed = discord.Embed(title="Application", color=discord.Color.green()) embed.add_field(name="Name", value=author.display_name) embed.add_field(name="Discord ID", value=author.id) embed.add_field(name="Age", value=answers[1]) embed.add_field(name="Reason for joining", value=answers[2]) return embed async def ask_questions(bot, author, questions): """Ask the user a list of questions and return the answers.""" answers = [] # Ask the user the questions for question in questions: await author.send(question) # Wait for the user to respond answer = await bot.wait_for( "message", check=lambda m: m.author == author and m.guild is None ) answers.append(answer.content) return answers