mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 19:28:54 -05:00
51ce2d82d6
Since these helpers will mostly be used by packages, it's important to make sure that the data they create stays contained inside them. This also brings the additional benefit of being able to manipulate data inside a package without knowing the name of the package's folder itself
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from discord.ext import commands
|
|
from collections import Counter
|
|
from core.settings import CoreDB
|
|
from enum import Enum
|
|
import os
|
|
|
|
|
|
class Red(commands.Bot):
|
|
def __init__(self, cli_flags, **kwargs):
|
|
self._shutdown_mode = ExitCodes.CRITICAL
|
|
self.db = CoreDB("core/data/settings.json",
|
|
relative_path=False)
|
|
|
|
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.db.get_global("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 shutdown(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"""
|
|
if not restart:
|
|
self._shutdown_mode = ExitCodes.SHUTDOWN
|
|
else:
|
|
self._shutdown_mode = ExitCodes.RESTART
|
|
|
|
await self.logout()
|
|
|
|
def list_packages(self):
|
|
"""Lists packages present in the cogs the folder"""
|
|
return os.listdir("cogs")
|
|
|
|
async def save_packages_status(self):
|
|
loaded = []
|
|
for package in self.extensions:
|
|
if package.startswith("cogs."):
|
|
loaded.append(package)
|
|
await self.db.set_global("packages", loaded)
|
|
|
|
|
|
class ExitCodes(Enum):
|
|
CRITICAL = 1
|
|
SHUTDOWN = 0
|
|
RESTART = 26 |