General improvements and refactoring

cogs/admin.py

@@ -1,4 +1,5 @@
 import logging
+import sys
 
 from discord import ApplicationContext, Embed, User, option, slash_command
 from discord import utils as ds_utils
@@ -14,9 +15,12 @@ logger = logging.getLogger(__name__)
 
 
 class Admin(commands.Cog):
+    """Cog with utility commands for admins."""
+
     def __init__(self, client: PycordBot):
         self.client = client
 
+    # Disabled because warning functionality is temporarily not needed
     # @slash_command(
     #     name="warning",
     #     description="Попередити юзера про порушення правил",
@@ -119,42 +123,49 @@ class Admin(commands.Cog):
                 amount,
                 ctx.channel.id,
             )
+
             await ctx.respond(
                 embed=Embed(description="Видаляю..."), ephemeral=True, delete_after=2.0
             )
-            if user == None:
+
+            if user is None:
                 await ctx.channel.purge(limit=amount)
             else:
                 await ctx.channel.purge(
                     limit=amount, check=lambda msg: msg.author == user
                 )
-        else:
-            logging.warning(
-                "User %s tried to use /clear but permission denied",
-                guild_name(ctx.user),
-            )
-            await ctx.respond(
-                embed=Embed(
-                    title="Відмовлено в доступі",
-                    description="Здається, це команда лише для модераторів",
-                    color=Color.fail,
-                )
-            )
-            mod_role = ds_utils.get(
-                ctx.user.guild.roles, id=await config_get("moderators", "roles")
-            )
-            admin_chan = ds_utils.get(
-                ctx.user.guild.channels,
-                id=await config_get("adminchat", "channels", "text"),
-            )
-            await admin_chan.send(
-                content=f"{mod_role.mention}",
-                embed=Embed(
-                    title="Неавторизований запит",
-                    description=f"Користувач {ctx.user.mention} запитав у каналі {ctx.channel.mention} команду, до якої не повинен мати доступу/бачити.",
-                    color=Color.fail,
-                ),
+
+            return
+
+        logging.warning(
+            "User %s tried to use /clear but permission denied",
+            guild_name(ctx.user),
+        )
+
+        await ctx.respond(
+            embed=Embed(
+                title="Відмовлено в доступі",
+                description="Здається, це команда лише для модераторів",
+                color=Color.fail,
             )
+        )
+
+        mod_role = ds_utils.get(
+            ctx.user.guild.roles, id=await config_get("moderators", "roles")
+        )
+        admin_chan = ds_utils.get(
+            ctx.user.guild.channels,
+            id=await config_get("adminchat", "channels", "text"),
+        )
+
+        await admin_chan.send(
+            content=f"{mod_role.mention}",
+            embed=Embed(
+                title="Неавторизований запит",
+                description=f"Користувач {ctx.user.mention} запитав у каналі {ctx.channel.mention} команду, до якої не повинен мати доступу/бачити.",
+                color=Color.fail,
+            ),
+        )
 
     @slash_command(
         name="reboot",
@@ -163,6 +174,7 @@ class Admin(commands.Cog):
     )
     async def reboot_cmd(self, ctx: ApplicationContext):
         await ctx.defer(ephemeral=True)
+
         if ctx.user.id in self.client.owner_ids:
             logging.info("Calling shutdown initiated by %s", guild_name(ctx.user))
             await ctx.respond(
@@ -173,34 +185,37 @@ class Admin(commands.Cog):
             )
             scheduler.shutdown()
             await self.client.close()
-            exit()
-        else:
-            logging.warning(
-                "User %s tried to use /reboot but permission denied",
-                guild_name(ctx.user),
-            )
-            await ctx.respond(
-                embed=Embed(
-                    title="Відмовлено в доступі",
-                    description="Здається, це команда лише для модераторів",
-                    color=Color.fail,
-                )
-            )
-            mod_role = ds_utils.get(
-                ctx.user.guild.roles, id=await config_get("moderators", "roles")
-            )
-            admin_chan = ds_utils.get(
-                ctx.user.guild.channels,
-                id=await config_get("adminchat", "channels", "text"),
-            )
-            await admin_chan.send(
-                content=f"{mod_role.mention}",
-                embed=Embed(
-                    title="Неавторизований запит",
-                    description=f"Користувач {ctx.user.mention} запитав у каналі {ctx.channel.mention} команду, до якої не повинен мати доступу/бачити.",
-                    color=Color.fail,
-                ),
+            sys.exit()
+
+        logging.warning(
+            "User %s tried to use /reboot but permission denied",
+            guild_name(ctx.user),
+        )
+
+        await ctx.respond(
+            embed=Embed(
+                title="Відмовлено в доступі",
+                description="Здається, це команда лише для модераторів",
+                color=Color.fail,
             )
+        )
+
+        mod_role = ds_utils.get(
+            ctx.user.guild.roles, id=await config_get("moderators", "roles")
+        )
+        admin_chan = ds_utils.get(
+            ctx.user.guild.channels,
+            id=await config_get("adminchat", "channels", "text"),
+        )
+
+        await admin_chan.send(
+            content=f"{mod_role.mention}",
+            embed=Embed(
+                title="Неавторизований запит",
+                description=f"Користувач {ctx.user.mention} запитав у каналі {ctx.channel.mention} команду, до якої не повинен мати доступу/бачити.",
+                color=Color.fail,
+            ),
+        )
 
 
 def setup(client: PycordBot):