Enforcing PEP 8, Refactored Code

cog/admin_tools.py

@@ -0,0 +1,334 @@
+from discord.commands import SlashCommandGroup
+from discord.commands import Option
+from discord.ext import commands
+import aiosqlite
+import traceback
+import datetime
+import discord
+
+
+class ModerationSystem(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    admin = SlashCommandGroup("admin")
+    warn_group = admin.create_subgroup("warn")
+
+    def create_user_remove_action_embed(
+            self,
+            source: discord.User,
+            target: discord.Member,
+            action: str,
+            reason: str
+    ):
+        guild = target.guild
+        embed = discord.Embed(
+            title=f"`✅` {target.name}#{target.discriminator} {action}",
+            description=f"Du hast den User {target.mention} aus dem Server **{guild.name}** {action}.",
+            color=discord.Color.green(),
+            timestamp=datetime.datetime.utcnow()
+        )
+        embed.add_field(name="Moderator:", value=f"{source}", inline=False)
+        embed.add_field(name="Grund:", value=f"{reason}", inline=False)
+        embed.set_author(name=f"{guild.name}", icon_url=target.avatar.url)
+        embed.set_thumbnail(url=target.avatar.url)
+        embed.set_footer(
+            text=f"{self.bot.user.name}#{self.bot.user.discriminator} | Oppro.net Development",
+            icon_url=self.bot.user.avatar.url
+        )
+
+        return embed
+
+    def create_warn_embed(
+            self,
+            source: discord.User,
+            target: discord.Member,
+            warn_id: int,
+            reason: str,
+            is_direct_message: bool = False,
+            is_remove: bool = False
+    ):
+        guild = target.guild
+
+        embed = discord.Embed(
+            title="`⚠️` Warn",
+            description=(
+                f"Ein Warn von dir vom Server **{guild.name}** wurde zurückgezogen."
+                if is_direct_message else
+                f"Du hast den {target.mention} aus dem Server **{guild.name}** unwarned."
+            ) if is_remove else (
+                f"Du wurdest auf dem Server **{guild.name}** gewarnt."
+                if is_direct_message else
+                f"Du hast den User {target.mention} auf dem Server **{guild.name}** gewarnt."
+            ),
+            color=discord.Color.green(),
+            timestamp=datetime.datetime.utcnow()
+        )
+        embed.add_field(name="Moderator:", value=f"```{source}```", inline=False)
+        embed.add_field(name="Warn ID:", value=f"```{warn_id}```", inline=False)
+        embed.add_field(name="Grund:", value=f"```{reason}```", inline=False)
+        embed.set_author(name=f"{guild.name}", icon_url=target.avatar.url)
+        embed.set_thumbnail(url=target.avatar.url if is_direct_message else source.avatar.url)
+        embed.set_footer(
+            text=f"{self.bot.user.name}#{self.bot.user.discriminator} | Oppro.net Development",
+            icon_url=self.bot.user.avatar.url
+        )
+
+        return embed
+
+    def create_error_embed(
+            self,
+            source: discord.User,
+            target: discord.Member,
+            error: discord.ClientException
+    ):
+        guild = target.guild
+        embed = discord.Embed(
+            title="`⚠️` Error",
+            description=f"Es ist ein Fehler aufgetreten.",
+            color=discord.Color.green(),
+            timestamp=datetime.datetime.utcnow()
+        )
+        embed.add_field(
+            name=f"Beim Kicken von {target.mention} ist ein Fehler aufgetreten.",
+            value=f"Bitte versuche es später erneut.", inline=False
+        )
+        embed.add_field(name=f"Fehler Code:", value=f"```{error}```", inline=False)
+        embed.set_author(name=f"{guild.name}", icon_url=self.bot.user.avatar.url)
+        embed.set_footer(
+            text=f"{self.bot.user.name}#{self.bot.user.discriminator} | Oppro.net Development",
+            icon_url=self.bot.user.avatar.url
+        )
+
+        return embed
+
+    @commands.Cog.listener()
+    async def on_ready(self):
+        async with aiosqlite.connect("mod_sys.db") as db:
+            await db.executescript(
+                """
+                CREATE TABLE IF NOT EXISTS WarnList (
+                warn_id INTEGER PRIMARY KEY,
+                mod_id INTEGER,
+                guild_id INTEGER,
+                user_id INTEGER,
+                warns INTEGER DEFAULT 0,
+                warn_reason TEXT,
+                warn_time TEXT
+                )
+                """
+            )
+
+    @admin.command(description="Kicke einen User aus dem Server")
+    @discord.default_permissions(kick_members=True)
+    @discord.guild_only()
+    async def kick(
+            self,
+            ctx: discord.ApplicationContext,
+            member: Option(discord.Member, "Wähle den User aus, den du kicken willst", required=True),
+            reason: Option(
+                str,
+                "Gib einen Grund an, warum du den User kicken willst",
+                required=False,
+                default="Kein Grund angegeben"
+            )
+    ):
+        try:
+            await member.kick(reason=reason)
+        except (discord.Forbidden, discord.HTTPException) as error:
+            traceback.print_exception(error)
+            return await ctx.respond(
+                embed=self.create_error_embed(ctx.user, member, error),
+                ephemeral=True
+            )
+
+        await ctx.respond(
+            embed=self.create_user_remove_action_embed(ctx.user, member, "gekickt", reason),
+            ephemeral=False
+        )
+
+    @commands.command(description="Banne einen User aus dem Server")
+    @discord.default_permissions(ban_members=True)
+    @discord.guild_only()
+    async def ban(
+            self,
+            ctx: discord.ApplicationContext,
+            member: Option(discord.Member, "Wähle den User aus, den du Bannen willst", required=True),
+            reason: Option(
+                str,
+                "Gib einen Grund an, warum du den User Bannen willst",
+                required=False,
+                default="Kein Grund angegeben"
+            )
+    ):
+        try:
+            await member.ban(reason=reason)
+        except (discord.Forbidden, discord.HTTPException) as error:
+            traceback.print_exception(error)
+            return await ctx.respond(
+                embed=self.create_error_embed(ctx.user, member, error),
+                ephemeral=True
+            )
+
+        await ctx.respond(
+            embed=self.create_user_remove_action_embed(ctx.user, member, "gebannt", reason),
+            ephemeral=False
+        )
+
+    @warn_group.command(name="add", description="Warne einen User aus dem Server")
+    @discord.default_permissions(kick_members=True)
+    @discord.guild_only()
+    async def add_warning(
+            self,
+            ctx: discord.ApplicationContext,
+            member: Option(discord.Member, "Wähle den User aus, den du warnen willst", required=True),
+            reason: Option(
+                str,
+                "Gib einen Grund an, warum du den User warnen willst",
+                required=False, default="Kein Grund angegeben"
+            )
+    ):
+        warn_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        async with aiosqlite.connect("mod_sys.db") as db:
+            await db.execute(
+                "INSERT INTO WarnList "
+                "(user_id, guild_id, warns, warn_reason, mod_id, warn_time) VALUES (?, ?, ?, ?, ?, ?)",
+                (member.id, ctx.guild.id, 1, reason, ctx.author.id, warn_time),
+            )
+            await db.commit()
+
+            async with db.execute(
+                    "SELECT warn_id FROM WarnList WHERE user_id = ? AND guild_id = ? ORDER BY warn_id DESC LIMIT 1",
+                    (member.id, ctx.guild.id),
+            ) as cursor:
+                row = await cursor.fetchone()
+                warn_id = row[0]
+
+        await member.send(
+            embed=self.create_warn_embed(ctx.user, member, warn_id, reason, True)
+        )
+        await ctx.respond(
+            embed=self.create_warn_embed(ctx.user, member, warn_id, reason, False),
+            ephemeral=False
+        )
+
+    @warn_group.command(name="remove", description="Unwarn einen User aus dem Server")
+    @discord.default_permissions(kick_members=True)
+    @discord.guild_only()
+    async def remove_warning(
+            self,
+            ctx: discord.ApplicationContext,
+            member: Option(discord.Member, "Wähle den User aus, den du unwarnen willst", required=True),
+            warn_id: Option(int, "Wähle die Warn ID aus, die du zurückziehen willst", required=True),
+            reason: Option(
+                str,
+                "Gib einen Grund an, warum du den User warnen willst",
+                required=False, default="Kein Grund angegeben"
+            )
+    ):
+        async with aiosqlite.connect("mod_sys.db") as db:
+            await db.execute(
+                "DELETE FROM WarnList WHERE user_id = ? AND guild_id = ? AND warn_id = ?",
+                (member.id, ctx.guild.id, warn_id)
+            )
+            await db.commit()
+
+        await member.send(
+            embed=self.create_warn_embed(ctx.user, member, warn_id, reason, True, True)
+        )
+        await ctx.respond(
+            embed=self.create_warn_embed(ctx.user, member, warn_id, reason, False, True),
+            ephemeral=False
+        )
+
+    @warn_group.command(name="list", description="Zeige alle Warns eines Users aus dem Server an")
+    @discord.default_permissions(kick_members=True)
+    @discord.guild_only()
+    async def warnings(self, ctx: discord.ApplicationContext, member: discord.Member):
+        async with aiosqlite.connect("mod_sys.db") as db:
+            async with db.execute(
+                "SELECT warn_id, mod_id, guild_id, user_id, warns, warn_reason, warn_time "
+                "FROM WarnList WHERE user_id = ? AND guild_id = ?",
+                (member.id, ctx.guild.id)
+            ) as cursor:
+                warnings = [
+                    f"**Warn-ID:** __{warn_id}__\n"
+                    f"""**Warn ausgestellt am:** {
+                    datetime.datetime.strptime(warn_time, '%Y-%m-%d %H:%M:%S').strftime('%Y-%m-%d %H:%M:%S')
+                    }\n"""
+                    f"**Moderator:** <@{mod_id}>\n"
+                    f"**Mod-ID**: __{mod_id}__\n"
+                    f"**> Grund:**\n```{warn_reason}```"
+                    for warn_id, mod_id, guild_id, user_id, warns, warn_reason, warn_time in await cursor.fetchall()
+                ]
+
+        if not warnings:
+            warnings_embed = discord.Embed(
+                title="`⚠️` The user has no warns!",
+                description=f"User: {member.mention}",
+                color=discord.Color.red(),
+            )
+        else:
+            warnings_embed = discord.Embed(
+                title=f"`⚠️` Warn Liste {member.name}#{member.discriminator}",
+                description=f"__**Liste der Warns**__",
+                color=discord.Color.green(),
+                timestamp=datetime.datetime.utcnow()
+            )
+        warnings_embed.add_field(name="", value="\n".join(warnings), inline=False)
+        warnings_embed.set_author(name=f"{ctx.guild.name}", icon_url=ctx.guild.icon.url)
+        warnings_embed.set_thumbnail(url=member.avatar.url)
+        warnings_embed.set_footer(
+            text=f"{ctx.bot.user.name}#{ctx.bot.user.discriminator}  | Oppro.net Development",
+            icon_url=ctx.bot.user.avatar.url
+        )
+
+        await ctx.respond(embed=warnings_embed, ephemeral=False)
+
+    @admin.command(description="Lösche Nachrichten aus dem Channel")
+    @commands.has_permissions(administrator=True)
+    async def purge(
+            self,
+            ctx: discord.ApplicationContext,
+            amount: Option(int, "Anzahl an Nachrichten (min. 1 | max. 100)", required=True)
+    ):
+        if amount > 101:
+            error_embed = discord.Embed(
+                title="`❌` Fehler!",
+                description="`Ich kann nicht mehr als 100 Nachrichten Löschen!`",
+                color=discord.Color.red(),
+                timestamp=datetime.datetime.utcnow()
+            )
+            error_embed.set_thumbnail(url=ctx.guild.icon.url)
+            error_embed.set_footer(text=f"| {ctx.bot.user.name}#{ctx.bot.user.discriminator}",
+                                   icon_url=ctx.bot.user.avatar.url)
+            error_embed.set_author(
+                name=f"Purge | Moderation System | Oppro.net Development",
+                icon_url=ctx.bot.user.avatar.url
+            )
+
+            return await ctx.respond(embed=error_embed, delete_after=6, ephemeral=True)
+
+        deleted = await ctx.channel.purge(limit=amount)
+
+        success_embed = discord.Embed(
+            title="`✅` Erfolgreich!",
+            description="**{}** `Nachrichten gelöscht!`".format(len(deleted)),
+            color=discord.Color.green(),
+            timestamp=datetime.datetime.utcnow()
+        )
+        success_embed.set_thumbnail(url=ctx.guild.icon.url)
+        success_embed.set_footer(
+            text=f"| {ctx.bot.user.name}#{ctx.bot.user.discriminator}",
+            icon_url=ctx.bot.user.avatar.url
+        )
+        success_embed.set_author(
+            name=f"Purge | Moderation System | Oppro.net Development",
+            icon_url=ctx.bot.user.avatar.url
+        )
+
+        await ctx.respond(embed=success_embed, delete_after=3, ephemeral=True)
+
+
+def setup(bot):
+    bot.add_cog(ModerationSystem(bot))