add register as embed

Author: zefanjajobse

cogs/admin.py

@@ -1,6 +1,8 @@
 """User management"""
 
 import logging
+import os
+import sys
 
 import discord
 from discord import Role, app_commands
@@ -13,9 +15,42 @@
 from database.dto.users import User
 from database.error_handling import is_unique_violation
 from utils.kd_roles import get_kd_roles
+from utils.register import register
 from utils.role_management import RoleManagement
 
 
+class RegisterModal(discord.ui.Modal, title="Register your EA account"):
+    def __init__(self, bot: KDRBot):
+        self.bot = bot
+        super().__init__()
+
+    username = discord.ui.TextInput(
+        label="What is your EA id",
+        style=discord.TextStyle.short,
+        max_length=500,
+        placeholder="test",
+        required=True,
+    )
+
+    async def on_submit(self, interaction: discord.Interaction[KDRBot]) -> None:
+        await interaction.response.defer()
+        await register(self.bot, interaction, self.username.value)
+
+
+class RegisterView(discord.ui.View):
+    def __init__(self, bot: KDRBot):
+        self.bot = bot
+        super().__init__(timeout=None)
+
+    @discord.ui.button(
+        label="Register",
+        custom_id="register_user",
+        style=discord.ButtonStyle.primary,
+    )
+    async def verify_callback(self, interaction: discord.Interaction, button):
+        await interaction.response.send_modal(RegisterModal(self.bot))
+
+
 class Admin(commands.Cog):
     def __init__(self, bot: KDRBot):
         self.bot = bot
@@ -170,7 +205,25 @@ async def remove_kd_role(
             await session.commit()
             await interaction.followup.send("KD-role has been removed", ephemeral=True)
 
+    kdroles_group = app_commands.Group(
+        name="generate", description="Generate a message", parent=group
+    )
+
+    @kdroles_group.command(name="register", description="Generate the register button")
+    @app_commands.guild_only()
+    @app_commands.default_permissions(administrator=True)
+    @app_commands.checks.has_permissions(administrator=True)
+    async def generate_register(self, interaction: discord.Interaction) -> None:
+        """Generate the register button"""
+        embed = discord.Embed(
+            title="Register with your EA account to get your KD-role",
+        )
+        await interaction.response.send_message(
+            embed=embed, view=RegisterView(self.bot)
+        )
+
 
 async def setup(bot: KDRBot) -> None:
     """Setup the cog within discord.py lib"""
+    bot.add_view(RegisterView(bot))
     await bot.add_cog(Admin(bot))

cogs/users.py

@@ -8,10 +8,7 @@
 
 from bot import KDRBot
 from database.dto.users import User
-from database.error_handling import is_unique_violation
-from utils.kd_roles import get_kd_roles
-from utils.role_management import RoleManagement
-from sqlalchemy.exc import IntegrityError
+from utils.register import register
 
 
 class Users(commands.Cog):
@@ -24,45 +21,14 @@ def __init__(self, bot: KDRBot):
     @app_commands.guild_only()
     async def register(self, interaction: discord.Interaction, username: str) -> None:
         """Register a user."""
-        async with self.bot.db.create_session() as session:
-            if interaction.guild is None:
-                await interaction.followup.send(
-                    "Command has to be used in the server", ephemeral=True
-                )
-                return
-
-            await interaction.response.defer()
-            players = await self.bot.gametools_api.find_player(interaction, username)
-            if len(players) <= 0:
-                await interaction.followup.send("User not found", ephemeral=True)
-                return
-            found_player = players[0]
-
-            stats = await self.bot.gametools_api.get_stats(found_player)
-            if len(players) <= 0:
-                await interaction.followup.send(
-                    "No stats found for user", ephemeral=True
-                )
-                return
-
-            stat = stats[0]
-            kd_roles = await get_kd_roles(session, stat["user"].server_id)
-            kdr_role_id = await RoleManagement().update_kdr_role(
-                self.bot, stat["user"], stat["gamemodes"], kd_roles
+        await interaction.response.defer()
+        if interaction.guild is None:
+            await interaction.followup.send(
+                "Command has to be used in the server", ephemeral=True
             )
+            return
 
-            found_player.kdr_role_id = kdr_role_id
-            try:
-                session.add(found_player)
-                await session.commit()
-            except IntegrityError as ex:
-                if is_unique_violation(ex):
-                    await interaction.followup.send(
-                        "You are already registered within this discord!",
-                        ephemeral=True,
-                    )
-
-            await interaction.followup.send("Registered", ephemeral=True)
+        await register(self.bot, interaction, username)
 
     @app_commands.command(name="kdr", description="Get your current kdr")
     @app_commands.guild_only()

utils/register.py

@@ -0,0 +1,41 @@
+import discord
+from sqlalchemy.exc import IntegrityError
+
+from bot import KDRBot
+from database.error_handling import is_unique_violation
+from utils.kd_roles import get_kd_roles
+from utils.role_management import RoleManagement
+
+
+async def register(bot: KDRBot, interaction: discord.Interaction, username: str):
+    async with bot.db.create_session() as session:
+        players = await bot.gametools_api.find_player(interaction, username)
+        if len(players) <= 0:
+            await interaction.followup.send("User not found", ephemeral=True)
+            return
+        found_player = players[0]
+
+        stats = await bot.gametools_api.get_stats(found_player)
+        if len(players) <= 0:
+            await interaction.followup.send("No stats found for user", ephemeral=True)
+            return
+
+        stat = stats[0]
+        kd_roles = await get_kd_roles(session, stat["user"].server_id)
+        kdr_role_id = await RoleManagement().update_kdr_role(
+            bot, stat["user"], stat["gamemodes"], kd_roles
+        )
+
+        found_player.kdr_role_id = kdr_role_id
+        try:
+            session.add(found_player)
+            await session.commit()
+        except IntegrityError as ex:
+            if is_unique_violation(ex):
+                await interaction.followup.send(
+                    "You are already registered within this discord!",
+                    ephemeral=True,
+                )
+                return
+
+        await interaction.followup.send("Registered", ephemeral=True)