From f5b0717661e309cfd8ea46bc84fa4460c3e7a0d4 Mon Sep 17 00:00:00 2001 From: Soheab Date: Mon, 1 Aug 2022 12:22:02 +0200 Subject: [PATCH] [commands] Add get_app_commands and walk_app_commands to Cog --- discord/ext/commands/cog.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/discord/ext/commands/cog.py b/discord/ext/commands/cog.py index 477e65623..0b70c66f5 100644 --- a/discord/ext/commands/cog.py +++ b/discord/ext/commands/cog.py @@ -375,6 +375,17 @@ class Cog(metaclass=CogMeta): """ return [c for c in self.__cog_commands__ if c.parent is None] + def get_app_commands(self) -> List[Union[app_commands.Command[Self, ..., Any], app_commands.Group]]: + r"""Returns the app commands that are defined inside this cog. + + Returns + -------- + List[Union[:class:`discord.app_commands.Command`, :class:`discord.app_commands.Group`]] + A :class:`list` of :class:`discord.app_commands.Command`\s and :class:`discord.app_commands.Group`\s that are + defined inside this cog, not including subcommands. + """ + return [c for c in self.__cog_app_commands__ if c.parent is None] + @property def qualified_name(self) -> str: """:class:`str`: Returns the cog's specified name, not the class name.""" @@ -405,6 +416,19 @@ class Cog(metaclass=CogMeta): if isinstance(command, GroupMixin): yield from command.walk_commands() + def walk_app_commands(self) -> Generator[Union[app_commands.Command[Self, ..., Any], app_commands.Group], None, None]: + """An iterator that recursively walks through this cog's app commands and subcommands. + + Yields + ------ + Union[:class:`discord.app_commands.Command`, :class:`discord.app_commands.Group`] + An app command or group from the cog. + """ + for command in self.__cog_app_commands__: + yield command + if isinstance(command, app_commands.Group): + yield from command.walk_commands() + @property def app_command(self) -> Optional[app_commands.Group]: """Optional[:class:`discord.app_commands.Group`]: Returns the associated group with this cog.