From b728061522bfee847af6742449e6e0c43a840ada Mon Sep 17 00:00:00 2001 From: Rapptz Date: Sat, 16 Mar 2019 09:23:34 -0400 Subject: [PATCH] [commands] Add HelpCommand.get_bot_mapping helper. --- discord/ext/commands/help.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/discord/ext/commands/help.py b/discord/ext/commands/help.py index 1898be4e1..664485f37 100644 --- a/discord/ext/commands/help.py +++ b/discord/ext/commands/help.py @@ -294,6 +294,16 @@ class HelpCommand: self._command_impl._eject_cog() self._command_impl = None + def get_bot_mapping(self): + """Retrieves the bot mapping passed to :meth:`send_bot_help`.""" + bot = self.context.bot + mapping = { + cog: cog.get_commands() + for cog in bot.cogs.values() + } + mapping[None] = [c for c in bot.all_commands.values() if c.cog is None] + return mapping + @property def clean_prefix(self): """The cleaned up invoke prefix. i.e. mentions are ``@name`` instead of ``<@id>``.""" @@ -717,11 +727,7 @@ class HelpCommand: bot = ctx.bot if command is None: - mapping = { - cog: cog.get_commands() - for cog in bot.cogs.values() - } - mapping[None] = [c for c in bot.all_commands.values() if c.cog is None] + mapping = self.get_bot_mapping() return await self.send_bot_help(mapping) # Check if it's a cog