From 428ebb98d0ffb19bc979822b33dd5c42563178b9 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@outlook.com>
Date: Mon, 31 May 2021 14:06:51 +0200
Subject: [PATCH] swapping commands for help discord server to his own cog

---
 src/cogs/confreriedukassoulait.py | 115 ++++++++++++++++++++++++++++++
 src/main.py                       |  86 +---------------------
 2 files changed, 116 insertions(+), 85 deletions(-)
 create mode 100644 src/cogs/confreriedukassoulait.py

diff --git a/src/cogs/confreriedukassoulait.py b/src/cogs/confreriedukassoulait.py
new file mode 100644
index 0000000..b471979
--- /dev/null
+++ b/src/cogs/confreriedukassoulait.py
@@ -0,0 +1,115 @@
+import discord
+import re
+import os
+from discord.ext import commands
+from random import choice
+from datetime import datetime
+from pytz import timezone
+customTimezone = os.environ['TIMEZONE']
+
+def setup(client):
+    client.add_cog(ConfrerieDuKassoulait(client))
+
+class ConfrerieDuKassoulait(commands.Cog):
+    """Unique pour le serveur Discord "La confrérie du Kassoulait"."""
+
+    def __init__(self, client):
+        self.client = client
+
+    @commands.Cog.listener()
+    async def on_ready(self):
+        channel = self.client.get_channel(742564480050790512)
+        await channel.send("Le bot a bien démarré.")
+
+    @commands.Cog.listener()
+    async def on_member_join(self, member):
+        if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
+            if member.bot == True:
+                role = discord.utils.get(member.guild.roles, name = "Bot")
+            else:
+                role = discord.utils.get(member.guild.roles, name = "Copain")
+            await member.add_roles(role)
+            try:
+                await member.send(f"Coucou **{member.name}** sur {member.guild.name} ! 🥰\n\nTu as le rôle **{role}** 💖!")
+            except:
+                pass
+            channel = self.client.get_channel(741639570172674120) # salons des arrivées
+            switch = [
+                f"Bienvenue, {member.mention}. On espère que tu as apporté de la pizza.",
+                f"C'est un plaisir de te voir, {member.mention}.",
+                f"{member.mention} vient juste d'arriver !",
+                f"{member.mention} vient juste d'atterrir.",
+                f"{member.mention} vient de se glisser dans le serveur.",
+                f"{member.mention} a bondi dans le serveur.",
+                f"Contents de te voir, {member.mention}.",
+                f"{member.mention} est arrivé(e).",
+                f"Tout le monde, accueillez comme il se doit {member.mention} !",
+                f"Youhou, tu as réussi, {member.mention} !",
+                f"{member.mention} a rejoint le groupe."
+            ]
+            message = await channel.send("...") # évite d'envoyer une notification
+            await message.edit(content = choice(switch))
+
+    @commands.Cog.listener()
+    async def on_member_remove(self, member):
+        if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
+            channel = self.client.get_channel(741639570172674120) # salons des arrivées
+            await channel.send(f"{member.mention} vient de quitter le serveur.")
+
+    @commands.Cog.listener()
+    async def on_raw_reaction_add(self, payload):
+        if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
+            if payload.emoji.name == '✅':
+                role = discord.utils.get(payload.member.guild.roles, name="règles-acceptés")
+                await payload.member.add_roles(role)
+
+    @commands.Cog.listener()
+    async def on_raw_reaction_remove(self, payload):
+        if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
+            if payload.emoji.name == '✅':
+                guild = discord.utils.find(lambda g : g.id == payload.guild_id, self.client.guilds)
+                member = discord.utils.find(lambda m: m.id == payload.user_id, guild.members)
+                role = discord.utils.get(guild.roles, name="règles-acceptés")
+                await member.remove_roles(role)
+
+    @commands.Cog.listener()
+    async def on_message_delete(self, message):
+        if message.author.guild.id == 441208120644075520: # Confrérie du Kassoulait
+            prefix = await self.client.get_prefix(message)
+            if not (
+                message.content.startswith(f"{prefix}note") or
+                message.content.startswith(f"{prefix}memo") or
+                len(re.findall(".com/channels/", message.content)) != 0 or
+                self.client.user.id is message.author.id
+            ):
+                user_suppressed = None
+
+                async for entry in message.guild.audit_logs(limit=1):
+                    if (datetime.now() - entry.created_at).seconds < 5 and str(entry.action) == 'AuditLogAction.message_delete':
+                        user_suppressed = entry.user
+                
+                channel = self.client.get_channel(742588187456831659)
+                embed = discord.Embed(description = f"{message.content}")
+
+                embed.set_author(name = self.user_or_nick(message.author), icon_url = message.author.avatar_url)
+
+                if not user_suppressed:
+                    embed.set_footer(text = f"Channel: #{message.channel.name} | Date : {self.goodTimezone(message.created_at, 1)}\nSupprimé le {datetime.now(timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
+                else:                
+                    embed.set_footer(icon_url = user_suppressed.avatar_url, text = f"Channel: #{message.channel.name} | Date : {self.goodTimezone(message.created_at, 1)}\nSupprimé par {self.user_or_nick(user_suppressed)} le {datetime.now(timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
+                
+                await channel.send(embed = embed)
+                # ne fonctionne pas quand un message a été supprimé avant que le bot ai démarré
+                # info sur la personne qui a supprimé ne fonctionne pas si il a supprimé un message auparavant (les logs se rajoute a un log deja existant)
+
+    def user_or_nick(user):
+        if user.nick:
+            return f"{user.nick} ({user.name}#{user.discriminator})"
+        else:
+            return f"{user.name}#{user.discriminator}"
+
+    def goodTimezone(self, date, type):
+        if type == 0:
+            return str(timezone(customTimezone).fromutc(date))[:-13].replace('-', '/').split()
+        elif type == 1:
+            return str(timezone(customTimezone).fromutc(date))[:-13].replace('-', '/').replace(' ', ' à ')
diff --git a/src/main.py b/src/main.py
index fe9a28f..cfa5ce0 100644
--- a/src/main.py
+++ b/src/main.py
@@ -5,8 +5,6 @@ import re
 import os
 from discord_slash import SlashCommand
 from discord.ext import commands
-from random import choice
-from datetime import datetime
 from pytz import timezone
 customPrefix = os.environ['PREFIX']
 customTimezone = os.environ['TIMEZONE']
@@ -22,6 +20,7 @@ client.load_extension("cogs.games")
 client.load_extension("cogs.fun")
 client.load_extension("cogs.autopublish")
 client.load_extension("cogs.school")
+client.load_extension("cogs.confreriedukassoulait")
 print("Terminé !")
 
 @client.event
@@ -32,59 +31,6 @@ async def on_connect():
 async def on_ready():
     await client.change_presence(status = discord.Status.online, activity = discord.Activity(name = f"{customPrefix}help", type = discord.ActivityType.playing))
     print("Bot prêt.")
-    channel = client.get_channel(742564480050790512)
-    await channel.send("Le bot a bien démarré.")
-
-@client.event
-async def on_member_join(member):
-    if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
-        if member.bot == True:
-            role = discord.utils.get(member.guild.roles, name = "Bot")
-        else:
-            role = discord.utils.get(member.guild.roles, name = "Copain")
-        await member.add_roles(role)
-        try:
-            await member.send(f"Coucou **{member.name}** sur {member.guild.name} ! 🥰\n\nTu as le rôle **{role}** 💖!")
-        except:
-            pass
-        channel = client.get_channel(741639570172674120) # salons des arrivées
-        switch = [
-            f"Bienvenue, {member.mention}. On espère que tu as apporté de la pizza.",
-            f"C'est un plaisir de te voir, {member.mention}.",
-            f"{member.mention} vient juste d'arriver !",
-            f"{member.mention} vient juste d'atterrir.",
-            f"{member.mention} vient de se glisser dans le serveur.",
-            f"{member.mention} a bondi dans le serveur.",
-            f"Contents de te voir, {member.mention}.",
-            f"{member.mention} est arrivé(e).",
-            f"Tout le monde, accueillez comme il se doit {member.mention} !",
-            f"Youhou, tu as réussi, {member.mention} !",
-            f"{member.mention} a rejoint le groupe."
-        ]
-        message = await channel.send("...") # évite d'envoyer une notification
-        await message.edit(content = choice(switch))
-
-@client.event
-async def on_member_remove(member):
-    if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
-        channel = client.get_channel(741639570172674120) # salons des arrivées
-        await channel.send(f"{member.mention} vient de quitter le serveur.")
-
-@client.event
-async def on_raw_reaction_add(payload):
-    if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
-        if payload.emoji.name == '✅':
-            role = discord.utils.get(payload.member.guild.roles, name="règles-acceptés")
-            await payload.member.add_roles(role)
-
-@client.event
-async def on_raw_reaction_remove(payload):
-    if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
-        if payload.emoji.name == '✅':
-            guild = discord.utils.find(lambda g : g.id == payload.guild_id, client.guilds)
-            member = discord.utils.find(lambda m: m.id == payload.user_id, guild.members)
-            role = discord.utils.get(guild.roles, name="règles-acceptés")
-            await member.remove_roles(role)
 
 @client.event
 async def on_command_error(ctx, error):
@@ -181,36 +127,6 @@ async def on_message(message):
                 if not "invalid literal for int() with base 10:" in e or not "404 Not Found (error code: 10008)" in e: # faute de frappe / message supprimé
                     print(e)
 
-@client.event
-async def on_message_delete(message):
-    if message.author.guild.id == 441208120644075520: # Confrérie du Kassoulait
-        prefix = await client.get_prefix(message)
-        if not (
-            message.content.startswith(f"{prefix}note") or
-            message.content.startswith(f"{prefix}memo") or
-            len(re.findall(".com/channels/", message.content)) != 0 or
-            client.user.id is message.author.id
-        ):
-            user_suppressed = None
-
-            async for entry in message.guild.audit_logs(limit=1):
-                if (datetime.now() - entry.created_at).seconds < 5 and str(entry.action) == 'AuditLogAction.message_delete':
-                    user_suppressed = entry.user
-            
-            channel = client.get_channel(742588187456831659)
-            embed = discord.Embed(description = f"{message.content}")
-
-            embed.set_author(name = user_or_nick(message.author), icon_url = message.author.avatar_url)
-
-            if not user_suppressed:
-                embed.set_footer(text = f"Channel: #{message.channel.name} | Date : {goodTimezone(message.created_at, 1)}\nSupprimé le {datetime.now(timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
-            else:                
-                embed.set_footer(icon_url = user_suppressed.avatar_url, text = f"Channel: #{message.channel.name} | Date : {goodTimezone(message.created_at, 1)}\nSupprimé par {user_or_nick(user_suppressed)} le {datetime.now(timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
-            
-            await channel.send(embed = embed)
-            # ne fonctionne pas quand un message a été supprimé avant que le bot ai démarré
-            # info sur la personne qui a supprimé ne fonctionne pas si il a supprimé un message auparavant (les logs se rajoute a un log deja existant)
-
 def user_or_nick(user):
     if user.nick:
         return f"{user.nick} ({user.name}#{user.discriminator})"