First commit

core/bot.py

@@ -0,0 +1,61 @@
+from discord.ext import commands
+from collections import Counter
+from core.utils.helpers import JsonGuildDB
+import os
+
+
+class Red(commands.Bot):
+    def __init__(self, cli_flags, **kwargs):
+        self._shutdown_mode = None
+        self.db = JsonGuildDB("core/data/settings.json",
+                              autosave=True,
+                              create_dirs=True)
+
+        def prefix_manager(bot, message):
+            global_prefix = self.db.get_global("prefix", [])
+            if message.guild is None:
+                return global_prefix
+            server_prefix = self.db.get(message.guild, "prefix", [])
+            return server_prefix if server_prefix else global_prefix
+
+        # Priority: args passed > cli flags > db
+        if "command_prefix" not in kwargs:
+            if cli_flags.prefix:
+                kwargs["command_prefix"] = lambda bot, message: cli_flags.prefix
+            else:
+                kwargs["command_prefix"] = None
+
+        if kwargs["command_prefix"] is None:
+            kwargs["command_prefix"] = prefix_manager
+
+        self.counter = Counter()
+        self.uptime = None
+        super().__init__(**kwargs)
+
+    async def is_owner(self, user, allow_coowners=True):
+        if allow_coowners:
+            if user.id in self.settings.coowners:
+                return True
+        return await super().is_owner(user)
+
+    async def send_cmd_help(self, ctx):
+        if ctx.invoked_subcommand:
+            pages = await self.formatter.format_help_for(ctx, ctx.invoked_subcommand)
+            for page in pages:
+                await ctx.send(page)
+        else:
+            pages = await self.formatter.format_help_for(ctx, ctx.command)
+            for page in pages:
+                await ctx.send(page)
+
+    async def logout(self, *, restart=False):
+        """Gracefully quits Red with exit code 0
+
+        If restart is True, the exit code will be 26 instead
+        Upon receiving that exit code, the launcher restarts Red"""
+        self._shutdown_mode = not restart
+        await super().logout()
+
+    def list_packages(self):
+        """Lists packages present in the cogs the folder"""
+        return os.listdir("cogs")