Improved type-hinting and overall sanity checks implemented.

cogs/custom_channels.py

@@ -1,4 +1,6 @@
-from discord import ApplicationContext, Embed, option
+from typing import Any, Dict, List, Union
+
+from discord import ApplicationContext, Embed, option, TextChannel
 from discord import utils as ds_utils
 from discord.abc import GuildChannel
 from discord.commands import SlashCommandGroup
@@ -8,24 +10,26 @@ from libbot.pycord.classes import PycordBot
 from libbot.sync import config_get as sync_config_get
 
 from classes.holo_user import HoloUser
-from enums.colors import Color
+from enums import Color
 from modules.database import col_users
 from modules.utils_sync import guild_name
 
 
 class CustomChannels(commands.Cog):
     def __init__(self, client: PycordBot):
-        self.client = client
+        self.client: PycordBot = client
 
     @commands.Cog.listener()
-    async def on_guild_channel_delete(self, channel: GuildChannel):
+    async def on_guild_channel_delete(self, channel: GuildChannel) -> None:
         await col_users.find_one_and_update(
             {"customchannel": channel.id}, {"$set": {"customchannel": None}}
         )
 
-    customchannel = SlashCommandGroup("customchannel", "Керування особистим каналом")
+    custom_channel_group: SlashCommandGroup = SlashCommandGroup(
+        "customchannel", "Керування особистим каналом"
+    )
 
-    @customchannel.command(
+    @custom_channel_group.command(
         name="get",
         description="Отримати персональний текстовий канал",
         guild_ids=[sync_config_get("guild")],
@@ -35,8 +39,8 @@ class CustomChannels(commands.Cog):
     @option("threads", description="Дозволити гілки")
     async def customchannel_get_cmd(
         self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
-    ):
-        holo_user_ctx = HoloUser(ctx.user)
+    ) -> None:
+        holo_user_ctx: HoloUser = HoloUser(ctx.user)
 
         # Return if the user already has a custom channel
         if holo_user_ctx.customchannel is not None:
@@ -45,14 +49,14 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Помилка виконання",
                     description="У вас вже є особистий канал.\nДля редагування каналу є `/customchannel edit` або просто відкрийте меню керування вашим каналом.",
-                    color=Color.fail,
+                    color=Color.FAIL,
                 )
             )
             return
 
         await ctx.defer()
 
-        created_channel = await ctx.user.guild.create_text_channel(
+        created_channel: TextChannel = await ctx.user.guild.create_text_channel(
             name=name,
             reason=f"Користувач {guild_name(ctx.user)} отримав власний приватний канал",
             category=ds_utils.get(
@@ -83,11 +87,11 @@ class CustomChannels(commands.Cog):
             embed=Embed(
                 title="Створено канал",
                 description=f"Вітаємо! Ви створили канал {created_channel.mention}. Для керування ним користуйтесь меню налаштувань каналу а також командою `/customchannel edit`",
-                color=Color.success,
+                color=Color.SUCCESS,
             )
         )
 
-        bots = await config_get("bots")
+        bots: List[Dict[str, Any]] = await config_get("bots")
 
         for bot in bots:
             await created_channel.set_permissions(
@@ -95,7 +99,7 @@ class CustomChannels(commands.Cog):
                 view_channel=False,
             )
 
-    @customchannel.command(
+    @custom_channel_group.command(
         name="edit",
         description="Змінити параметри особистого каналу",
         guild_ids=[sync_config_get("guild")],
@@ -105,10 +109,10 @@ class CustomChannels(commands.Cog):
     @option("threads", description="Дозволити гілки")
     async def customchannel_edit_cmd(
         self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
-    ):
-        holo_user_ctx = HoloUser(ctx.user)
+    ) -> None:
+        holo_user_ctx: HoloUser = HoloUser(ctx.user)
 
-        custom_channel = ds_utils.get(
+        custom_channel: Union[TextChannel, None] = ds_utils.get(
             ctx.guild.channels, id=holo_user_ctx.customchannel
         )
 
@@ -118,7 +122,7 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Канал не знайдено",
                     description="Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
-                    color=Color.fail,
+                    color=Color.FAIL,
                 )
             )
             return
@@ -136,11 +140,11 @@ class CustomChannels(commands.Cog):
             embed=Embed(
                 title="Канал змінено",
                 description=f"Назва каналу тепер `{name}`, реакції `{reactions}` та дозволено треди `{threads}`",
-                color=Color.fail,
+                color=Color.FAIL,
             )
         )
 
-    @customchannel.command(
+    @custom_channel_group.command(
         name="remove",
         description="Відібрати канал, знищуючи його, та частково повернути кошти",
         guild_ids=[sync_config_get("guild")],
@@ -148,8 +152,8 @@ class CustomChannels(commands.Cog):
     @option("confirm", description="Підтвердження операції")
     async def customchannel_remove_cmd(
         self, ctx: ApplicationContext, confirm: bool = False
-    ):
-        holo_user_ctx = HoloUser(ctx.user)
+    ) -> None:
+        holo_user_ctx: HoloUser = HoloUser(ctx.user)
 
         # Return if the user does not have a custom channel
         if holo_user_ctx.customchannel is None:
@@ -158,14 +162,14 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Помилка виконання",
                     description="У вас немає особистого каналу.",
-                    color=Color.fail,
+                    color=Color.FAIL,
                 )
             )
             return
 
         await ctx.defer()
 
-        custom_channel = ds_utils.get(
+        custom_channel: Union[TextChannel, None] = ds_utils.get(
             ctx.guild.channels, id=holo_user_ctx.customchannel
         )
 
@@ -175,7 +179,7 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Канал не знайдено",
                     description="Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
-                    color=Color.fail,
+                    color=Color.FAIL,
                 )
             )
             await holo_user_ctx.set("customchannel", None)
@@ -187,7 +191,7 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Підтвердження не надано",
                     description="Для підтвердження операції додайте до команди параметр `confirm` зі значенням `True`.",
-                    color=Color.fail,
+                    color=Color.FAIL,
                 )
             )
             return
@@ -201,10 +205,10 @@ class CustomChannels(commands.Cog):
                 embed=Embed(
                     title="Канал знищено",
                     description="Ви відмовились від каналу та видалили його.",
-                    color=Color.default,
+                    color=Color.DEFAULT,
                 )
             )
 
 
-def setup(client: PycordBot):
+def setup(client: PycordBot) -> None:
     client.add_cog(CustomChannels(client))