Modal Form for signup

Author: drcrazy

slao_bot/cogs/signup.py

@@ -1,17 +1,87 @@
-from discord import Colour, DMChannel, Embed, Member, Message
+import discord
+from discord import Colour, Embed, Member, app_commands
 from discord.ext import commands
 from discord.ext.commands import Context
+
+from slaobot import SlaoBot
 from utils.config import settings
 
 
+class SignUpModal(discord.ui.Modal, title='Информация о себе'):
+    name = discord.ui.TextInput(
+        label='Твой имя',
+        placeholder='Как тебя зовут',
+        max_length=100,
+        min_length=2,
+    )
+
+    character = discord.ui.TextInput(
+        label='Имя персонажа',
+        placeholder='Имя основного персонажа',
+        max_length=100,
+        min_length=2,
+    )
+
+    class_spec = discord.ui.TextInput(
+        label='Класс и специализация',
+        placeholder='Класс и основная специализация',
+        max_length=100,
+        min_length=2,
+    )
+
+    trade_skill = discord.ui.TextInput(
+        label='Профессии',
+        placeholder='Профессии и специализация в профессии, если есть',
+        style=discord.TextStyle.long,
+        required=False,
+        max_length=100,
+        min_length=2,
+    )
+
+    twinks = discord.ui.TextInput(
+        label='Твинки',
+        placeholder='Имена твинков и их класс',
+        style=discord.TextStyle.long,
+        required=False,
+        max_length=100,
+        min_length=2,
+    )
+
+    async def on_submit(self, interaction: discord.Interaction):
+
+        signup_channel = interaction.client.get_channel(settings.signup_channel_id)
+
+        if signup_channel:
+            signup_embed = Embed(title='Новая заявка',
+                                 description='Больше Адептов!!',
+                                 colour=Colour.light_gray())
+            signup_embed.add_field(name='Имя', value=self.name.value)
+            signup_embed.add_field(name='Персонаж', value=self.character.value)
+            signup_embed.add_field(name='Класс', value=self.class_spec.value)
+            signup_embed.add_field(name='Профы', value=self.trade_skill.value)
+            signup_embed.add_field(name='Твинки', value=self.twinks.value)
+            signup_embed.add_field(name='Пользователь', value=format(interaction.user.mention))
+            if interaction.user.avatar:
+                signup_embed.set_thumbnail(url=interaction.user.avatar.url)
+            await signup_channel.send(embed=signup_embed)
+
+        await interaction.response.send_message(
+            f'Спасибо, {self.name.value}! Офицеры в скором времени выдадут права в Discord',
+            ephemeral=True
+        )
+
+    async def on_error(self, interaction: discord.Interaction, error: Exception) -> None:
+        await interaction.response.send_message('Oops! Something went wrong.', ephemeral=True)
+
+
 class SignUp(commands.Cog):
-    def __init__(self, bot):
+    def __init__(self, bot: SlaoBot):
         """
         Cog to greet newcomers and give them some basic questionnaire.
 
         :param bot: Bot instance
         """
-        self.bot = bot
+        self.bot: SlaoBot = bot
 
     @commands.Cog.listener()
     async def on_member_join(self, member: Member) -> None:
@@ -25,47 +95,24 @@ async def on_member_join(self, member: Member) -> None:
         intro_message = (f'Привет, {format(member.mention)}! \r\n'
                          'Приветствуем тебя на Discord сервере гильдии <Адепты Катаклизма>! \r\n'
                          'Мы играем в Wrath of the Lich King Classic за Альянс на Пламегоре. \r\n'
-                         'Если хочешь вступить к нам в гильдию, то напиши мне + в ответ. \r\n'
+                         'Если хочешь вступить к нам в гильдию, то напиши /signup в любом  \r\n'
+                         'канале Discord сервера Адептов. \r\n'
                          'Удачного времяпрепровождения!')
 
         await dm_channel.send(content=intro_message)
 
-    @commands.Cog.listener()
-    async def on_message(self, message: Message) -> None:
-        if not isinstance(message.channel, DMChannel):
-            return
-        if message.author.bot:
-            return
-
-        if message.content == '+':
-            signup_form = (
-                'Для получения доступа к гильдейским каналам мы бы хотели узнать тебя поближе. \r\n'
-                'Пришли мне, пожалуйста, следующую информацию: \r\n '
-                '1. Твоё имя \r\n'
-                '2. Имя персонажа \r\n'
-                '3. Класс, специализация \r\n'
-                '4. Профессии, специализация \r\n'
-                '5. Имена персонажей твинков'
-            )
-
-            await message.channel.send(signup_form)
-        elif len(message.content) > 5:
-            await message.channel.send('Спасибо! Офицеры в скором времени выдадут права в Discord')
-            signup_channel = self.bot.get_channel(settings.signup_channel_id)
-            if signup_channel:
-                signup_embed = Embed(title='Новая заявка',
-                                     description=message.content,
-                                     colour=Colour.light_gray())
-                signup_embed.add_field(name='Пользователь', value=format(message.author.mention))
-                if message.author.avatar:
-                    signup_embed.set_thumbnail(url=message.author.avatar.url)
-                await signup_channel.send(embed=signup_embed)
-
     @commands.command(hidden=True)
     async def emit_join(self, ctx: Context) -> None:
         """Emit member_join event."""
         self.bot.dispatch('member_join', ctx.author)
 
+    @app_commands.command(description='SignUp Form')
+    async def signup(self, interaction: discord.Interaction):
+        # Send the modal with an instance of our `Feedback` class
+        # Since modals require an interaction, they cannot be done as a response to a text command.
+        # They can only be done as a response to either an application command or a button press.
+        await interaction.response.send_modal(SignUpModal())
+
 
 async def setup(bot):
     await bot.add_cog(SignUp(bot))

slao_bot/slaobot.py

@@ -14,6 +14,8 @@
     'cogs.damage',
 )
 
+TEST_GUILD = discord.Object(873894278110265404)
+
 
 class SlaoBot(commands.Bot):
     def __init__(self):
@@ -25,6 +27,7 @@ def __init__(self):
             message_content=True,
             reactions=True,
         )
+        logging.basicConfig(level=logging.INFO)
 
         super().__init__(
             command_prefix=f'{settings.command_prefix}',
@@ -37,7 +40,10 @@ async def setup_hook(self) -> None:
             try:
                 await self.load_extension(extension)
             except Exception:
-                logging.info(f'Failed to load extension {extension}.')
+                logging.exception(f'Failed to load extension {extension}.')
+
+        synced = await self.tree.sync()
+        logging.info(synced)
 
     async def on_ready(self) -> None:
         logging.info(f'We have logged in as {self.user}')