Merge pull request #1 from ae-utbm/attribute-president-role

Attribute president role

Author: imperosol

bot.toml.example

@@ -4,13 +4,22 @@ log_level = "INFO"
 
 [guild]
 id = <the id of the server here>
-club_ids = [
-    1,  # AE
-    29,  # Club Welcome
-    38,  # Troll Penché
-    43,  # Reflex
-    79,  # Cook'UT
-]
+
+[guild.club.1]  # AE
+president_role_id = <role id>
+member_role_id = <role id>
+[guild.club.29]  # Club Welcome
+president_role_id = <role id>
+member_role_id = <role id>
+[guild.club.38]  # Troll Penché
+president_role_id = <role id>
+member_role_id = <role id>
+[guild.club.43]  # Reflex
+president_role_id = <role id>
+member_role_id = <role id>
+[guild.club.79]  # Cook'UT
+president_role_id = <role id>
+member_role_id = <role id>
 
 [sith_api]
 api_key = "<your sith api token here>"

src/commands/club.py

@@ -2,20 +2,32 @@
 
 from typing import TYPE_CHECKING
 
-from discord import Interaction, app_commands
-from discord.app_commands import Choice
+from discord import Interaction, Member, app_commands
+from discord.app_commands import Choice, Transform, Transformer
 from discord.ext import commands
+from discord.ext.commands import BadArgument
 
+from src.client import ClubSchema  # noqa TC001
 from src.services.club import ClubService
 from src.settings import Settings
 
 if TYPE_CHECKING:
-    from src.client import ClubSchema, SithClient
+    from src.client import SithClient
     from src.main import AeBot
 
 
+class ClubTransformer(Transformer):
+    async def transform(
+        self, interaction: Interaction[AeBot], value: int
+    ) -> ClubSchema:
+        club = await interaction.client.client.get_club(value)
+        if not club:
+            raise BadArgument("Ce club n'existe pas")
+        return club
+
+
 class ClubCog(commands.GroupCog, group_name="club"):
-    def __init__(self, client: SithClient, bot: "AeBot"):
+    def __init__(self, client: SithClient, bot: AeBot):
         self.club_service = ClubService(client, bot)
         self.settings = Settings()
 
@@ -30,10 +42,31 @@ async def autocomplete_club(
 
     @app_commands.command(name="infos")
     @app_commands.autocomplete(club=autocomplete_club)
-    async def club_infos(self, interaction: Interaction, club: int):
+    async def club_infos(
+        self, interaction: Interaction, club: Transform[ClubSchema, ClubTransformer]
+    ):
         await interaction.response.defer(thinking=True)
-        club: ClubSchema = await self.club_service.get_club(club)
-        if club is None:
-            await interaction.followup.send("Ce club n'a pas été trouvé.")
-            return
         await interaction.followup.send(embed=self.club_service.embed(club))
+
+    @app_commands.command(name="add_member")
+    @app_commands.autocomplete(club=autocomplete_club)
+    async def add_club_member(
+        self,
+        interaction: Interaction,
+        club: int,
+        member: Member,
+    ):
+        await interaction.response.defer(thinking=True)
+        club = next(
+            (c for _id, c in self.settings.guild.clubs.items() if _id == club), None
+        )
+        if not club:
+            await interaction.followup.send("Ce club n'a pas été trouvé")
+            return
+        if not interaction.user.get_role(club.president_role_id):
+            await interaction.followup.send(
+                "Seul le président du club peut utiliser cette commande"
+            )
+            return
+        await member.add_roles(interaction.guild.get_role(club.member_role_id))
+        await interaction.followup.send("Rôle attribué :thumbs_up:")

src/main.py

@@ -19,7 +19,9 @@ def __init__(self, client: SithClient):
         self.settings = Settings()
         self.logger = logging.getLogger("discord")
         self.client = client
-        super().__init__(command_prefix="?", intents=Intents.all())
+        super().__init__(
+            command_prefix=self.settings.bot.command_prefix, intents=Intents.all()
+        )
 
     async def setup_hook(self):
         await self.add_cog(ClubCog(self.client, self))

src/services/club.py

@@ -25,7 +25,7 @@ async def search_club(self, current: str) -> list[SimpleClubSchema]:
         clubs = await self._client.search_clubs(current)
         if clubs is None:
             return []
-        return [club for club in clubs if club.id in self._config.guild.club_ids]
+        return [club for club in clubs if club.id in self._config.guild.clubs]
 
     async def get_club(self, club_id: int) -> ClubSchema | None:
         if club_id not in self._club_cache:

src/settings.py

@@ -17,9 +17,14 @@ class ApiConfig(BaseModel):
     api_key: SecretStr
 
 
+class ClubConfig(BaseModel):
+    president_role_id: int
+    member_role_id: int
+
+
 class GuildConfig(BaseModel):
     id: int
-    club_ids: list[int]
+    clubs: dict[int, ClubConfig]
 
 
 class BotConfig(BaseModel):