add auto join

Author: TitouanDor

src/commands/club.py

@@ -125,6 +125,13 @@ async def create_club(
         if Club.filter(Club.sith_id == club.id).exists():
             await interaction.followup.send(f"Le club : {club.name} existe déjà...")
         else:
+            guild = interaction.guild
+            channel_club = utils.get(guild.channels, id=1428320894610309121)
+            mess = await channel_club.send(
+                f"Réagi à ce message pour rejoindre le club {club.name}"
+            )
+
+            await mess.add_reaction("✅")
             await self.club_service.create_club(club, interaction.guild)
             await interaction.followup.send(f"Le club : {club.name} à été créé")
 

src/commands/role.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from discord.ext import commands
+
+from src.client import ClubSchema  # noqa TC001
+from src.db.models import Club
+from src.services.club import ClubService
+from src.settings import Settings
+
+if TYPE_CHECKING:
+    from src.main import AeBot
+
+
+class RoleCog(commands.GroupCog, group_name="role"):
+    def __init__(self, bot: AeBot):
+        self.settings = Settings()
+        self.club_service = ClubService(bot)
+        self.bot = bot
+
+    @commands.Cog.listener()
+    async def on_raw_reaction_add(self, payload):
+        if payload.guild_id is None:
+            return
+
+        guild = self.bot.get_guild(payload.guild_id)
+        member = guild.get_member(payload.user_id)
+        channel = self.bot.get_channel(payload.channel_id)
+        message = await channel.fetch_message(payload.message_id)
+        emoji = str(payload.emoji)
+
+        db_club = Club.get_or_none(Club.message_autorole_id, message.id)
+
+        if not db_club:
+            return
+
+        elif emoji != "✅":
+            await message.remove_reaction(payload.emoji, member)
+
+        await self.club_service.add_member(db_club, member)
+
+    @commands.Cog.listener()
+    async def on_raw_reaction_remove(self, payload):
+        if payload.guild_id is None:
+            return
+
+        guild = self.bot.get_guild(payload.guild_id)
+        member = guild.get_member(payload.user_id)
+        channel = self.bot.get_channel(payload.channel_id)
+        message = await channel.fetch_message(payload.message_id)
+        emoji = str(payload.emoji)
+
+        db_club = Club.get_or_none(Club.message_autorole_id, message.id)
+
+        if not db_club or emoji != "✅":
+            await member.send("nonon")
+            return
+
+        await self.club_service.remove_member(db_club, member, _former=0)

src/db/models.py

@@ -36,6 +36,7 @@ class Club(DbBaseModel):
     treasurer_role_id = peewee.IntegerField(unique=True)
     member_role_id = peewee.IntegerField(unique=True)
     former_member_role_id = peewee.IntegerField(unique=True)
+    message_autorole_id = peewee.IntegerField(unique=True)
 
 
 def init():

src/main.py

@@ -15,6 +15,7 @@
 from src.commands.club import ClubCog
 from src.commands.misc import MiscCog
 from src.commands.news import NewsCog
+from src.commands.role import RoleCog
 from src.settings import BASE_DIR, Settings
 
 if TYPE_CHECKING:
@@ -38,6 +39,7 @@ async def setup_hook(self):
         await self.add_cog(NewsCog(self))
         await self.add_cog(AdminCog(self))
         await self.add_cog(MiscCog())
+        await self.add_cog(RoleCog(self))
 
     async def on_ready(self):
         await self.wait_until_ready()

src/services/club.py

@@ -130,11 +130,14 @@ async def add_member(self, club: Club, member: Member):
             )
         await member.add_roles(role, reason=f"{member.name} joined club {club.name}")
 
-    async def remove_member(self, club: Club, member: Member):
+    async def remove_member(self, club: Club, member: Member, _former=1):
         role = utils.get(member.guild.roles, id=club.member_role_id)
         former = utils.get(member.guild.roles, id=club.former_member_role_id)
         await member.remove_roles(role, reason=f"{member.name} left club {club.name}")
-        await member.add_roles(former, reason=f"{member.name} left club {club.name}")
+        if _former == 1:
+            await member.add_roles(
+                former, reason=f"{member.name} left club {club.name}"
+            )
 
     async def handover(
         self, club: ClubSchema, new_pres: Member, new_treso: Member, guild: Guild