From 6110fe5b1ca19a11a193872bbfd7431644cdd75e Mon Sep 17 00:00:00 2001 From: Rapptz Date: Tue, 1 Mar 2022 07:37:39 -0500 Subject: [PATCH] Add fallback overloads for CommandTree --- discord/app_commands/tree.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/discord/app_commands/tree.py b/discord/app_commands/tree.py index b0fdf997c..52057d2c8 100644 --- a/discord/app_commands/tree.py +++ b/discord/app_commands/tree.py @@ -210,6 +210,17 @@ class CommandTree: ) -> Optional[Union[Command, Group]]: ... + @overload + def remove_command( + self, + command: str, + /, + *, + guild: Optional[Snowflake] = ..., + type: AppCommandType = ..., + ) -> Optional[Union[Command, ContextMenu, Group]]: + ... + def remove_command( self, command: str, @@ -278,6 +289,17 @@ class CommandTree: ) -> Optional[Union[Command, Group]]: ... + @overload + def get_command( + self, + command: str, + /, + *, + guild: Optional[Snowflake] = ..., + type: AppCommandType = ..., + ) -> Optional[Union[Command, ContextMenu, Group]]: + ... + def get_command( self, command: str, @@ -339,6 +361,15 @@ class CommandTree: ) -> List[Union[Command, Group]]: ... + @overload + def get_commands( + self, + *, + guild: Optional[Snowflake] = ..., + type: AppCommandType = ..., + ) -> Union[List[Union[Command, Group]], List[ContextMenu]]: + ... + def get_commands( self, *,