moderation/moderation.py

@@ -1,7 +1,6 @@
 import logging
 import time
 from datetime import datetime, timedelta, timezone
-from typing import Union
 import discord
 import humanize
 import mysql.connector
@@ -12,7 +11,7 @@ from redbot.core.app_commands import Choice
 
 
 class Moderation(commands.Cog):
-    """Custom moderation cog.
+    """Custom cog moderation cog, meant to copy GalacticBot.
     Developed by SeaswimmerTheFsh."""
 
     def __init__(self, bot):
@@ -46,8 +45,7 @@ class Moderation(commands.Cog):
         guilds: list[discord.Guild] = self.bot.guilds
         try:
             for guild in guilds:
-                if not await self.bot.cog_disabled_in_guild(self, guild):
+                await self.create_guild_table(guild)
-                    await self.create_guild_table(guild)
         except ConnectionRefusedError:
             return
 
@@ -57,54 +55,52 @@ class Moderation(commands.Cog):
     @commands.Cog.listener('on_guild_join')
     async def db_generate_guild_join(self, guild: discord.Guild):
         """This method prepares the database schema whenever the bot joins a guild."""
-        if not await self.bot.cog_disabled_in_guild(self, guild):
+        conf = await self.check_conf([
-            conf = await self.check_conf([
+            'mysql_address',
-                'mysql_address',
+            'mysql_database',
-                'mysql_database',
+            'mysql_username',
-                'mysql_username',
+            'mysql_password'
-                'mysql_password'
+        ])
-            ])
+        if conf:
-            if conf:
+            self.logger.error("Failed to create a table for %s, due to MySQL connection configuration being unset.", guild.id)
-                self.logger.error("Failed to create a table for %s, due to MySQL connection configuration being unset.", guild.id)
+            return
-                return
+        try:
-            try:
+            await self.create_guild_table(guild)
-                await self.create_guild_table(guild)
+        except ConnectionRefusedError:
-            except ConnectionRefusedError:
+            return
-                return
 
     @commands.Cog.listener('on_audit_log_entry_create')
     async def autologger(self, entry: discord.AuditLogEntry):
         """This method automatically logs moderations done by users manually ("right clicks")."""
-        if not await self.bot.cog_disabled_in_guild(self, entry.guild):
+        if await self.config.guild(entry.guild.id).ignore_other_bots() is True:
-            if await self.config.guild(entry.guild.id).ignore_other_bots() is True:
+            if entry.user.bot or entry.target.bot:
-                if entry.user.bot or entry.target.bot:
-                    return
-            else:
-                if entry.user.id == self.bot.user.id:
-                    return
-            duration = "NULL"
-            if entry.reason:
-                reason = entry.reason + " (This action was performed without the bot.)"
-            else:
-                reason = "This action was performed without the bot."
-            if entry.action == discord.AuditLogAction.kick:
-                moderation_type = 'KICK'
-            elif entry.action == discord.AuditLogAction.ban:
-                moderation_type = 'BAN'
-            elif entry.action == discord.AuditLogAction.unban:
-                moderation_type = 'UNBAN'
-            elif entry.action == discord.AuditLogAction.member_update:
-                if entry.after.timed_out_until is not None:
-                    timed_out_until_aware = entry.after.timed_out_until.replace(tzinfo=timezone.utc)
-                    duration_datetime = timed_out_until_aware - datetime.now(tz=timezone.utc)
-                    minutes = round(duration_datetime.total_seconds() / 60)
-                    duration = timedelta(minutes=minutes)
-                    moderation_type = 'MUTE'
-                else:
-                    moderation_type = 'UNMUTE'
-            else:
                 return
-            await self.mysql_log(entry.guild.id, entry.user.id, moderation_type, entry.target.id, duration, reason)
+        else:
+            if entry.user.id == self.bot.user.id:
+                return
+        duration = "NULL"
+        if entry.reason:
+            reason = entry.reason + " (This action was performed without the bot.)"
+        else:
+            reason = "This action was performed without the bot."
+        if entry.action == discord.AuditLogAction.kick:
+            moderation_type = 'KICK'
+        elif entry.action == discord.AuditLogAction.ban:
+            moderation_type = 'BAN'
+        elif entry.action == discord.AuditLogAction.unban:
+            moderation_type = 'UNBAN'
+        elif entry.action == discord.AuditLogAction.member_update:
+            if entry.after.timed_out_until is not None:
+                timed_out_until_aware = entry.after.timed_out_until.replace(tzinfo=timezone.utc)
+                duration_datetime = timed_out_until_aware - datetime.now(tz=timezone.utc)
+                minutes = round(duration_datetime.total_seconds() / 60)
+                duration = timedelta(minutes=minutes)
+                moderation_type = 'MUTE'
+            else:
+                moderation_type = 'UNMUTE'
+        else:
+            return
+        await self.mysql_log(entry.guild.id, entry.user.id, moderation_type, entry.target.id, duration, reason)
 
     async def connect(self):
         """Connects to the MySQL database, and returns a connection object."""
@@ -177,17 +173,6 @@ class Moderation(commands.Cog):
                 not_found_list.append(item)
         return not_found_list
 
-    def check_permissions(self, member: discord.Member, permissions: list, ctx: Union[commands.Context, discord.Interaction] = None):
-        """Checks if the bot has a specific permission (or a list of permissions) in a channel."""
-        if ctx:
-            resolved_permissions = ctx.channel.permissions_for(member)
-        else:
-            resolved_permissions = member.guild_permissions
-        for permission in permissions:
-            if not getattr(resolved_permissions, permission, False) and not resolved_permissions.administrator is True:
-                return permission
-        return False
-
     async def mysql_log(self, guild_id: str, author_id: str, moderation_type: str, target_id: int, duration, reason: str):
         timestamp = int(time.time())
         if duration != "NULL":
@@ -316,14 +301,6 @@ class Moderation(commands.Cog):
             Why are you noting this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
         await interaction.response.send_message(content=f"{target.mention} has recieved a note!\n**Reason** - `{reason}`")
         if silent is None:
             silent = not await self.config.guild(interaction.guild).dm_users()
@@ -347,14 +324,6 @@ class Moderation(commands.Cog):
             Why are you warning this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
         await interaction.response.send_message(content=f"{target.mention} has been warned!\n**Reason** - `{reason}`")
         if silent is None:
             silent = not await self.config.guild(interaction.guild).dm_users()
@@ -380,18 +349,6 @@ class Moderation(commands.Cog):
             Why are you unbanning this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
-        permissions = self.check_permissions(interaction.client.user, ['moderate_members'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         if target.is_timed_out() is True:
             await interaction.response.send_message(f"{target.mention} is already muted!", allowed_mentions=discord.AllowedMentions(users=False), ephemeral=True)
             return
@@ -427,18 +384,6 @@ class Moderation(commands.Cog):
             Why are you unmuting this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
-        permissions = self.check_permissions(interaction.client.user, ['moderate_members'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         if target.is_timed_out() is False:
             await interaction.response.send_message(f"{target.mention} is not muted!", allowed_mentions=discord.AllowedMentions(users=False), ephemeral=True)
             return
@@ -470,18 +415,6 @@ class Moderation(commands.Cog):
             Why are you kicking this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
-        permissions = self.check_permissions(interaction.client.user, ['kick_members'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         await interaction.response.send_message(content=f"{target.mention} has been kicked!\n**Reason** - `{reason}`")
         if silent is None:
             silent = not await self.config.guild(interaction.guild).dm_users()
@@ -518,18 +451,6 @@ class Moderation(commands.Cog):
             How many days of messages to delete?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
-        permissions = self.check_permissions(interaction.client.user, ['ban_members'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         try:
             await interaction.guild.fetch_ban(target)
             await interaction.response.send_message(content=f"{target.mention} is already banned!", ephemeral=True)
@@ -575,18 +496,6 @@ class Moderation(commands.Cog):
             Why are you unbanning this user?
         silent: bool
             Should the user be messaged?"""
-        if interaction.guild.get_member(target.id):
-            target_member = interaction.guild.get_member(target.id)
-            if interaction.guild.get_member(interaction.client.user.id).top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a role higher than the bot!", ephemeral=True)
-                return
-            if interaction.user.top_role <= target_member.top_role:
-                await interaction.response.send_message(content="You cannot moderate members with a higher role than you!", ephemeral=True)
-                return
-        permissions = self.check_permissions(interaction.client.user, ['moderate_members'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         try:
             await interaction.guild.fetch_ban(target)
         except discord.errors.NotFound:
@@ -624,10 +533,6 @@ class Moderation(commands.Cog):
             Page to select
         epheremal: bool
             Hide the command response"""
-        permissions = self.check_permissions(interaction.client.user, ['embed_links'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         database = await self.connect()
         cursor = database.cursor()
         if target:
@@ -690,10 +595,6 @@ class Moderation(commands.Cog):
             Case number of the case you're trying to resolve
         reason: str
             Reason for resolving case"""
-        permissions = self.check_permissions(interaction.client.user, ['embed_links'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         conf = await self.check_conf(['mysql_database'])
         if conf:
             raise(LookupError)
@@ -754,10 +655,6 @@ class Moderation(commands.Cog):
             What case are you looking up?
         ephemeral: bool
             Hide the command response"""
-        permissions = self.check_permissions(interaction.client.user, ['embed_links'], interaction)
-        if permissions:
-            await interaction.response.send_message(f"I do not have the `{permissions}` permission, required for this action.", ephemeral=True)
-            return
         database = await self.connect()
         cursor = database.cursor()
         query = "SELECT * FROM moderation_%s WHERE moderation_id = %s;"
@@ -782,25 +679,24 @@ class Moderation(commands.Cog):
         db = await self.config.mysql_database()
         guilds: list[discord.Guild] = self.bot.guilds
         for guild in guilds:
-            if not await self.bot.cog_disabled_in_guild(self, guild):
+            tempban_query = f"SELECT target_id, moderation_id FROM moderation_{guild.id} WHERE end_timestamp != 0 AND end_timestamp <= %s AND moderation_type = 'TEMPBAN' AND expired = 0"
-                tempban_query = f"SELECT target_id, moderation_id FROM moderation_{guild.id} WHERE end_timestamp != 0 AND end_timestamp <= %s AND moderation_type = 'TEMPBAN' AND expired = 0"
+            try:
+                cursor.execute(tempban_query, (time.time(),))
+                result = cursor.fetchall()
+            except mysql.connector.errors.ProgrammingError:
+                continue
+            target_ids = [row[0] for row in result]
+            moderation_ids = [row[1] for row in result]
+            for target_id, moderation_id in zip(target_ids, moderation_ids):
+                user: discord.User = await self.bot.fetch_user(target_id)
+                await guild.unban(user, reason=f"Automatic unban from case #{moderation_id}")
+                embed = await self.embed_factory('message', guild, f'Automatic unban from case #{moderation_id}', 'unbanned')
                 try:
-                    cursor.execute(tempban_query, (time.time(),))
+                    await user.send(embed=embed)
-                    result = cursor.fetchall()
+                except discord.errors.HTTPException:
-                except mysql.connector.errors.ProgrammingError:
+                    pass
-                    continue
+            expiry_query = f"UPDATE `{db}`.`moderation_{guild.id}` SET expired = 1 WHERE (end_timestamp != 0 AND end_timestamp <= %s AND expired = 0) OR (expired = 0 AND resolved = 1)"
-                target_ids = [row[0] for row in result]
+            cursor.execute(expiry_query, (time.time(),))
-                moderation_ids = [row[1] for row in result]
-                for target_id, moderation_id in zip(target_ids, moderation_ids):
-                    user: discord.User = await self.bot.fetch_user(target_id)
-                    await guild.unban(user, reason=f"Automatic unban from case #{moderation_id}")
-                    embed = await self.embed_factory('message', guild, f'Automatic unban from case #{moderation_id}', 'unbanned')
-                    try:
-                        await user.send(embed=embed)
-                    except discord.errors.HTTPException:
-                        pass
-                expiry_query = f"UPDATE `{db}`.`moderation_{guild.id}` SET expired = 1 WHERE (end_timestamp != 0 AND end_timestamp <= %s AND expired = 0) OR (expired = 0 AND resolved = 1)"
-                cursor.execute(expiry_query, (time.time(),))
         database.commit()
         cursor.close()
         database.close()