Formatted with black

classes/holo_user.py

@@ -11,20 +11,28 @@ try:
 except ImportError:
     from typing_extensions import Literal
 
+
 class NotEnoughMoneyError(Exception):
     """User does not have enough money to do that"""
+
     pass
 
+
 class UserNotFoundError(Exception):
     """HoloUser could not find user with such an ID in database"""
+
     def __init__(self, user, user_id):
         self.user = user
         self.user_id = user_id
-        super().__init__(f"User of type {type(self.user)} with id {self.user_id} was not found")
+        super().__init__(
+            f"User of type {type(self.user)} with id {self.user_id} was not found"
+        )
 
-class HoloUser():
 
-    def __init__(self, user: Union[discord.User, discord.Member, discord.member.Member, int]) -> None:
+class HoloUser:
+    def __init__(
+        self, user: Union[discord.User, discord.Member, discord.member.Member, int]
+    ) -> None:
         """Get an object that has a proper binding between Discord ID and database
 
         ### Args:
@@ -35,7 +43,7 @@ class HoloUser():
         """
 
         if hasattr(user, "id"):
-            self.id = user.id # type: ignore
+            self.id = user.id  # type: ignore
         else:
             self.id = user
 
@@ -168,9 +176,12 @@ class HoloUser():
         """
         warns = col_warnings.find_one({"user": self.id})
         if warns != None:
-            col_warnings.update_one(filter={"_id": self.db_id}, update={ "$set": { "warns": warns["warns"]+count } })
+            col_warnings.update_one(
+                filter={"_id": self.db_id},
+                update={"$set": {"warns": warns["warns"] + count}},
+            )
         else:
-            col_warnings.insert_one(document={ "user": self.id, "warns": count })
+            col_warnings.insert_one(document={"user": self.id, "warns": count})
         logWrite(f"User {self.id} was warned {count} times due to: {reason}")
 
     # def cooldown_go(self, kind: Literal["work", "daily", "weekly", "monthly", "steal"]) -> None:
@@ -192,7 +203,9 @@ class HoloUser():
         if not hasattr(self, key):
             raise AttributeError()
         setattr(self, key, value)
-        col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { key: value } }, upsert=True)
+        col_users.update_one(
+            filter={"_id": self.db_id}, update={"$set": {key: value}}, upsert=True
+        )
         logWrite(f"Set attribute {key} of user {self.id} to {value}")
 
     def purge(self) -> None:

cogs/custom_channels.py

@@ -6,69 +6,152 @@ from enums.colors import Color
 from modules.utils import config_get
 from modules.utils_sync import config_get_sync, guild_name
 
-class CustomChannels(commands.Cog):
 
+class CustomChannels(commands.Cog):
     def __init__(self, client):
         self.client = client
 
     customchannel = SlashCommandGroup("customchannel", "Керування особистим каналом")
 
-    @customchannel.command(name="buy", description="Отримати персональний текстовий канал", guild_ids=[config_get_sync("guild")])
+    @customchannel.command(
+        name="buy",
+        description="Отримати персональний текстовий канал",
+        guild_ids=[config_get_sync("guild")],
+    )
     @option("name", description="Назва каналу")
     @option("reactions", description="Дозволити реакції")
     @option("threads", description="Дозволити гілки")
-    async def customchannel_buy_cmd(self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool):
+    async def customchannel_buy_cmd(
-            
+        self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
+    ):
         holo_user_ctx = HoloUser(ctx.user)
 
         if holo_user_ctx.customchannel == None:
             await ctx.defer()
-            created_channel = await ctx.user.guild.create_text_channel(name=name,
+            created_channel = await ctx.user.guild.create_text_channel(
+                name=name,
                 reason=f"Користувач {guild_name(ctx.user)} купив канал",
-                category=utils.get(ctx.author.guild.categories, id=await config_get("customchannels", "categories")),
+                category=utils.get(
+                    ctx.author.guild.categories,
+                    id=await config_get("customchannels", "categories"),
+                ),
+            )
+            await created_channel.set_permissions(
+                ctx.user.guild.default_role,
+                send_messages=False,
+                add_reactions=reactions,
+                create_public_threads=threads,
+                create_private_threads=threads,
+            )
+            await created_channel.set_permissions(
+                ctx.user,
+                attach_files=True,
+                manage_messages=True,
+                send_messages=True,
+                embed_links=True,
+                manage_channels=True,
             )
-            await created_channel.set_permissions(ctx.user.guild.default_role, send_messages=False, add_reactions=reactions, create_public_threads=threads, create_private_threads=threads)
-            await created_channel.set_permissions(ctx.user, attach_files=True, manage_messages=True, send_messages=True, embed_links=True, manage_channels=True)
             holo_user_ctx.set("customchannel", created_channel.id)
-            await ctx.respond(embed=Embed(title="Створено канал", description=f"Вітаємо! Ви створили канал {created_channel.mention}. Для керування ним користуйтесь меню налаштувань каналу а також командою `/customchannel edit`", color=Color.success))
+            await ctx.respond(
+                embed=Embed(
+                    title="Створено канал",
+                    description=f"Вітаємо! Ви створили канал {created_channel.mention}. Для керування ним користуйтесь меню налаштувань каналу а також командою `/customchannel edit`",
+                    color=Color.success,
+                )
+            )
             bots = await config_get("bots")
             for bot in bots:
-                await created_channel.set_permissions(utils.get(ctx.user.guild.roles, id=bots[bot]["role"]), view_channel=False)
+                await created_channel.set_permissions(
+                    utils.get(ctx.user.guild.roles, id=bots[bot]["role"]),
+                    view_channel=False,
+                )
         else:
             await ctx.defer(ephemeral=True)
-            await ctx.respond(embed=Embed(title="Помилка виконання", description=f"У вас вже є особистий канал.\nДля редагування каналу є `/customchannel edit` або просто відкрийте меню керування вашим каналом.", color=Color.fail))
+            await ctx.respond(
+                embed=Embed(
+                    title="Помилка виконання",
+                    description=f"У вас вже є особистий канал.\nДля редагування каналу є `/customchannel edit` або просто відкрийте меню керування вашим каналом.",
+                    color=Color.fail,
+                )
+            )
 
-    @customchannel.command(name="edit", description="Змінити параметри особистого каналу", guild_ids=[config_get_sync("guild")])
+    @customchannel.command(
+        name="edit",
+        description="Змінити параметри особистого каналу",
+        guild_ids=[config_get_sync("guild")],
+    )
     @option("name", description="Назва каналу")
     @option("reactions", description="Дозволити реакції")
     @option("threads", description="Дозволити гілки")
-    async def customchannel_edit_cmd(self, ctx: ApplicationContext, name:str, reactions: bool, threads: bool):
+    async def customchannel_edit_cmd(
-            
+        self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
+    ):
         holo_user_ctx = HoloUser(ctx.user)
 
         custom_channel = utils.get(ctx.guild.channels, id=holo_user_ctx.customchannel)
         if custom_channel is None:
-            await ctx.respond(embed=Embed(title="Канал не знайдено", description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.", color=Color.fail))
+            await ctx.respond(
+                embed=Embed(
+                    title="Канал не знайдено",
+                    description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
+                    color=Color.fail,
+                )
+            )
             return
         await custom_channel.edit(name=name)
-        await custom_channel.set_permissions(ctx.user.guild.default_role, send_messages=False, add_reactions=reactions, create_public_threads=threads, create_private_threads=threads)
+        await custom_channel.set_permissions(
-        await ctx.respond(embed=Embed(title="Канал змінено", description=f"Назва каналу тепер `{name}`, реакції `{reactions}` та дозволено треди `{threads}`", color=Color.fail))
+            ctx.user.guild.default_role,
+            send_messages=False,
+            add_reactions=reactions,
+            create_public_threads=threads,
+            create_private_threads=threads,
+        )
+        await ctx.respond(
+            embed=Embed(
+                title="Канал змінено",
+                description=f"Назва каналу тепер `{name}`, реакції `{reactions}` та дозволено треди `{threads}`",
+                color=Color.fail,
+            )
+        )
 
-    @customchannel.command(name="refund", description="Відібрати канал, знищуючи його, та частково повернути кошти", guild_ids=[config_get_sync("guild")])
+    @customchannel.command(
+        name="refund",
+        description="Відібрати канал, знищуючи його, та частково повернути кошти",
+        guild_ids=[config_get_sync("guild")],
+    )
     async def customchannel_refund_cmd(self, ctx: ApplicationContext):
-            
         holo_user_ctx = HoloUser(ctx.user)
 
         if holo_user_ctx.customchannel is not None:
             await ctx.defer()
-            custom_channel = utils.get(ctx.guild.channels, id=holo_user_ctx.customchannel)
+            custom_channel = utils.get(
+                ctx.guild.channels, id=holo_user_ctx.customchannel
+            )
             if custom_channel is None:
-                await ctx.respond(embed=Embed(title="Канал не знайдено", description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.", color=Color.fail))
+                await ctx.respond(
+                    embed=Embed(
+                        title="Канал не знайдено",
+                        description=f"Канал, вказаний як ваш, не існує. Можливо, його було вручну видалено раніше.",
+                        color=Color.fail,
+                    )
+                )
                 holo_user_ctx.set("customchannel", None)
                 return
             await custom_channel.delete(reason="Повернення коштів")
             holo_user_ctx.set("customchannel", None)
-            await ctx.respond(embed=Embed(title="Канал знищено", description=f"Ви відмовились від каналу.", color=Color.default))
+            await ctx.respond(
+                embed=Embed(
+                    title="Канал знищено",
+                    description=f"Ви відмовились від каналу.",
+                    color=Color.default,
+                )
+            )
         else:
             await ctx.defer(ephemeral=True)
-            await ctx.respond(embed=Embed(title="Помилка виконання", description=f"У вас немає особистого каналу.", color=Color.fail))
+            await ctx.respond(
+                embed=Embed(
+                    title="Помилка виконання",
+                    description=f"У вас немає особистого каналу.",
+                    color=Color.fail,
+                )
+            )

modules/database.py

@@ -6,12 +6,12 @@ with open("config.json", "r", encoding="utf-8") as f:
     f.close()
 
 db_client = MongoClient(
-    'mongodb://{0}:{1}@{2}:{3}/{4}'.format(
+    "mongodb://{0}:{1}@{2}:{3}/{4}".format(
         db_config["user"],
         db_config["password"],
         db_config["host"],
         db_config["port"],
-        db_config["name"]
+        db_config["name"],
     )
 )
 db = db_client.get_database(name=db_config["name"])

modules/utils_sync.py

@@ -35,6 +35,7 @@ def config_set_sync(key: str, value: Any, *path: str) -> None:
     json_write_sync(this_dict, "config.json")
     return
 
+
 def guild_name(member: Member):
     if member.nick == None:
         return member.name