Support str type for aliases

This commit is contained in:
Richard Chien 2020-02-14 16:44:22 +08:00
parent cfa9fc3936
commit f7e4223dc6
2 changed files with 5 additions and 2 deletions

View File

@ -151,7 +151,7 @@ class CommandFunc:
def on_command(name: Union[str, CommandName_T], *,
aliases: Iterable[str] = (),
aliases: Union[Iterable[str], str] = (),
permission: int = perm.EVERYBODY,
only_to_me: bool = True,
privileged: bool = False,
@ -189,6 +189,9 @@ def on_command(name: Union[str, CommandName_T], *,
current_parent = current_parent[parent_key]
current_parent[cmd_name[-1]] = cmd
nonlocal aliases
if isinstance(aliases, str):
aliases = (aliases,)
for alias in aliases:
_aliases[alias] = cmd_name

View File

@ -17,7 +17,7 @@ class CommandGroup:
shell_like: bool = ...): ...
def command(self, name: Union[str, CommandName_T], *,
aliases: Iterable[str] = ...,
aliases: Union[Iterable[str], str] = ...,
permission: int = ...,
only_to_me: bool = ...,
privileged: bool = ...,