db api

Author: fourjr

.gitignore

@@ -108,6 +108,9 @@ ENV/
 #MacOS
 .DS_Store
 
+# custom
+.vscode/
+
 # Modmail
 config.json
 plugins/

bot.py

@@ -40,7 +40,7 @@
 from motor.motor_asyncio import AsyncIOMotorClient
 
 from core.changelog import ChangeLog
-from core.clients import ModmailApiClient, SelfHostedClient
+from core.clients import ModmailApiClient, SelfHostedClient, PluginDatabaseClient
 from core.config import ConfigManager
 from core.models import Bot
 from core.thread import ThreadManager
@@ -65,6 +65,7 @@ def __init__(self):
             self._api = SelfHostedClient(self)
         else:
             self._api = ModmailApiClient(self)
+        self.plugin_db = PluginDatabaseClient(self)
 
         self.data_task = self.loop.create_task(self.data_loop())
         self.autoupdate_task = self.loop.create_task(self.autoupdate_loop())

cogs/plugins.py

@@ -52,7 +52,7 @@ async def download_initial_plugins(self):
     async def download_plugin_repo(username, repo):
         try:
             cmd = f'git clone https://github.com/{username}/{repo} '
-            cmd += 'plugins/{username}-{repo} -q'
+            cmd += f'plugins/{username}-{repo} -q'
             subprocess.run(cmd, check=True, capture_output=True)
             # -q (quiet) so there's no terminal output unless there's an error
         except subprocess.CalledProcessError as exc:

core/checks.py

@@ -1,5 +1,6 @@
 from discord.ext import commands
 
+
 def has_permissions(**perms):
     """Check if the author has required permissions.
     This will always return ``True`` if the author is a bot owner, or

core/clients.py

@@ -381,3 +381,12 @@ async def get_user_info(self):
                 'url': user.url
             }
         }
+
+
+class PluginDatabaseClient:
+    def __init__(self, bot: Bot):
+        self.bot = bot
+
+    def get_partition(self, cog_cls):
+        cls_name = cog_cls.__class__.__name__
+        return self.bot.db.plugins[cls_name]