Changed the Client structure

cogs/admin.py

@@ -14,9 +14,9 @@ from discord import (
 from discord import utils as ds_utils
 from discord.ext import commands
 from libbot import config_get
-from libbot.pycord.classes import PycordBot
 from libbot.sync import config_get as sync_config_get
 
+from classes.holo_bot import HoloBot
 from enums import Color
 from modules.scheduler import scheduler
 from modules.utils_sync import guild_name
@@ -28,8 +28,8 @@ logger = logging.getLogger(__name__)
 class Admin(commands.Cog):
     """Cog with utility commands for admins."""
 
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     # Disabled because warning functionality is temporarily not needed
     # @slash_command(
@@ -236,5 +236,5 @@ class Admin(commands.Cog):
             )
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(Admin(client))

cogs/analytics.py

@@ -3,16 +3,16 @@ from typing import Dict, List, Any
 
 from discord import Cog, Message
 from discord.ext import commands
-from libbot.pycord.classes import PycordBot
 
+from classes.holo_bot import HoloBot
 from modules.database import col_analytics
 
 logger = logging.getLogger(__name__)
 
 
 class Analytics(commands.Cog):
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     @Cog.listener()
     async def on_message(self, message: Message) -> None:
@@ -60,5 +60,5 @@ class Analytics(commands.Cog):
             )
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(Analytics(client))

cogs/custom_channels.py

@@ -7,9 +7,9 @@ from discord.abc import GuildChannel
 from discord.commands import SlashCommandGroup
 from discord.ext import commands
 from libbot import config_get
-from libbot.pycord.classes import PycordBot
 from libbot.sync import config_get as sync_config_get
 
+from classes.holo_bot import HoloBot
 from classes.holo_user import HoloUser
 from enums import Color
 from modules.database import col_users
@@ -19,8 +19,8 @@ logger = logging.getLogger(__name__)
 
 
 class CustomChannels(commands.Cog):
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     @commands.Cog.listener()
     async def on_guild_channel_delete(self, channel: GuildChannel) -> None:
@@ -234,5 +234,5 @@ class CustomChannels(commands.Cog):
             )
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(CustomChannels(client))

cogs/data.py

@@ -1,4 +1,5 @@
 import logging
+from logging import Logger
 from os import makedirs
 from pathlib import Path
 from typing import Union, List, Dict, Any
@@ -9,21 +10,21 @@ from discord import utils as ds_utils
 from discord.commands import SlashCommandGroup
 from discord.ext import commands
 from libbot import config_get
-from libbot.pycord.classes import PycordBot
 from libbot.sync import config_get as sync_config_get
 from libbot.sync import json_write as sync_json_write
 
+from classes.holo_bot import HoloBot
 from classes.holo_user import HoloUser
 from enums import Color
 from modules.database import col_users
 from modules.utils_sync import guild_name
 
-logger = logging.getLogger(__name__)
+logger: Logger = logging.getLogger(__name__)
 
 
 class Data(commands.Cog):
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     data: SlashCommandGroup = SlashCommandGroup("data", "Керування даними користувачів")
 
@@ -179,5 +180,5 @@ class Data(commands.Cog):
         )
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(Data(client))

cogs/fun.py

@@ -3,9 +3,9 @@ import logging
 from discord import ApplicationContext, Embed, User, option, slash_command
 from discord.ext import commands
 from libbot import config_get
-from libbot.pycord.classes import PycordBot
 from libbot.sync import config_get as sync_config_get
 
+from classes.holo_bot import HoloBot
 from modules.utils_sync import guild_name
 from modules.waifu_pics import waifu_pics
 
@@ -13,8 +13,8 @@ logger = logging.getLogger(__name__)
 
 
 class Fun(commands.Cog):
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     @slash_command(
         name="action",
@@ -54,5 +54,5 @@ class Fun(commands.Cog):
         await ctx.respond(embed=embed)
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(Fun(client))

cogs/logger.py

@@ -4,14 +4,14 @@ from discord import Member, Message, TextChannel
 from discord import utils as ds_utils
 from discord.ext import commands
 from libbot import config_get
-from libbot.pycord.classes import PycordBot
 
+from classes.holo_bot import HoloBot
 from modules.database import col_users
 
 
 class Logger(commands.Cog):
-    def __init__(self, client: PycordBot):
-        self.client: PycordBot = client
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
 
     @commands.Cog.listener()
     async def on_message(self, message: Message):
@@ -65,5 +65,5 @@ class Logger(commands.Cog):
                 await col_users.insert_one(document=user)
 
 
-def setup(client: PycordBot) -> None:
+def setup(client: HoloBot) -> None:
     client.add_cog(Logger(client))

cogs/utility.py

@@ -0,0 +1,57 @@
+import logging
+from logging import Logger
+
+from discord import Activity, ActivityType
+from discord.ext import commands
+from libbot import config_get
+
+from classes.holo_bot import HoloBot
+
+logger: Logger = logging.getLogger(__name__)
+
+
+class Utility(commands.Cog):
+    def __init__(self, client: HoloBot):
+        self.client: HoloBot = client
+
+    @commands.Cog.listener()
+    async def on_ready(self) -> None:
+        logger.info("Logged in as %s", self.client.user)
+
+        activity_type: str = await config_get("type", "status")
+        activity_message: str = await config_get("message", "status")
+
+        if activity_type == "playing":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.playing, name=activity_message)
+            )
+        elif activity_type == "watching":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.watching, name=activity_message)
+            )
+        elif activity_type == "listening":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.listening, name=activity_message)
+            )
+        elif activity_type == "streaming":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.streaming, name=activity_message)
+            )
+        elif activity_type == "competing":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.competing, name=activity_message)
+            )
+        elif activity_type == "custom":
+            await self.client.change_presence(
+                activity=Activity(type=ActivityType.custom, name=activity_message)
+            )
+        else:
+            return
+
+        logger.info(
+            "Set activity type to %s with message %s", activity_type, activity_message
+        )
+
+
+def setup(client: HoloBot) -> None:
+    client.add_cog(Utility(client))