From 17f15aca5bd262195fa78366a1a2e60b9cda8910 Mon Sep 17 00:00:00 2001
From: profitroll <vozhd.kk@gmail.com>
Date: Wed, 27 Dec 2023 13:45:40 +0100
Subject: [PATCH] Initial Pycord addition

---
 requirements/pycord.txt               |  1 +
 requirements/pyrogram.txt             |  2 +-
 src/libbot/__init__.py                |  2 +-
 src/libbot/pycord/classes/__init__.py |  1 +
 src/libbot/pycord/classes/bot.py      | 57 +++++++++++++++++++++++++++
 5 files changed, 61 insertions(+), 2 deletions(-)
 create mode 100644 src/libbot/pycord/classes/__init__.py
 create mode 100644 src/libbot/pycord/classes/bot.py

diff --git a/requirements/pycord.txt b/requirements/pycord.txt
index e38caa7..5c3fd69 100644
--- a/requirements/pycord.txt
+++ b/requirements/pycord.txt
@@ -1 +1,2 @@
+apscheduler~=3.10.4
 py-cord~=2.4.1
\ No newline at end of file
diff --git a/requirements/pyrogram.txt b/requirements/pyrogram.txt
index 9d72624..a7d1e59 100644
--- a/requirements/pyrogram.txt
+++ b/requirements/pyrogram.txt
@@ -1,2 +1,2 @@
-apscheduler~=3.10.1
+apscheduler~=3.10.4
 pyrogram~=2.0.106
\ No newline at end of file
diff --git a/src/libbot/__init__.py b/src/libbot/__init__.py
index c26fb2f..163d106 100644
--- a/src/libbot/__init__.py
+++ b/src/libbot/__init__.py
@@ -2,5 +2,5 @@ __version__ = "2.0.1"
 __license__ = "GPL3"
 __author__ = "Profitroll"
 
-from . import i18n, pyrogram, sync
+from . import i18n, pycord, pyrogram, sync
 from .__main__ import *
diff --git a/src/libbot/pycord/classes/__init__.py b/src/libbot/pycord/classes/__init__.py
new file mode 100644
index 0000000..4294771
--- /dev/null
+++ b/src/libbot/pycord/classes/__init__.py
@@ -0,0 +1 @@
+from .bot import PycordBot
diff --git a/src/libbot/pycord/classes/bot.py b/src/libbot/pycord/classes/bot.py
new file mode 100644
index 0000000..e59dad5
--- /dev/null
+++ b/src/libbot/pycord/classes/bot.py
@@ -0,0 +1,57 @@
+import logging
+from pathlib import Path
+from typing import Any, Dict, Union
+
+try:
+    from apscheduler.schedulers.asyncio import AsyncIOScheduler
+    from apscheduler.schedulers.background import BackgroundScheduler
+    from discord import Bot
+except ImportError as exc:
+    raise ImportError(
+        "You need to install libbot[pycord] in order to use this class."
+    ) from exc
+
+try:
+    from ujson import loads
+except ImportError:
+    from json import loads
+
+from libbot.i18n import BotLocale
+from libbot.i18n.sync import _
+
+logger = logging.getLogger(__name__)
+
+
+class PycordBot(Bot):
+    def __init__(
+        self,
+        config: Union[Dict[str, Any], None] = None,
+        config_path: Union[str, Path] = Path("config.json"),
+        locales_root: Union[str, Path, None] = None,
+        scheduler: Union[AsyncIOScheduler, BackgroundScheduler, None] = None,
+        **kwargs,
+    ):
+        if config is None:
+            with open(config_path, "r", encoding="utf-8") as f:
+                self.config: dict = loads(f.read())
+        else:
+            self.config = config
+
+        super().__init__(
+            debug_guilds=self.config["bot"]["debug_guilds"],
+            owner_ids=self.config["bot"]["owners"],
+            **kwargs,
+        )
+
+        self.bot_locale: BotLocale = BotLocale(
+            default_locale=self.config["locale"],
+            locales_root=(Path("locale") if locales_root is None else locales_root),
+        )
+        self.default_locale: str = self.bot_locale.default
+        self.locales: dict = self.bot_locale.locales
+
+        self._ = self.bot_locale._
+        self.in_all_locales = self.bot_locale.in_all_locales
+        self.in_every_locale = self.bot_locale.in_every_locale
+
+        self.scheduler: Union[AsyncIOScheduler, BackgroundScheduler, None] = scheduler