ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-02-17 16:20:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix docstring of command handler function

Browse Source
This commit is contained in:
Richard Chien 2020-03-14 22:45:17 +08:00
parent c02b9a4ccf
commit c366e6c950
1 changed files with 10 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
nonebot/command/__init__.py
View File

@ -139,25 +139,6 @@ class Command:
return self.__repr__()
class CommandFunc:
__slots__ = ('cmd', 'func')
def __init__(self, cmd: Command, func: CommandHandler_T):
self.cmd = cmd
self.func = func
def __call__(self, session: 'CommandSession') -> Any:
return self.func(session)
def args_parser(self, parser_func: CommandHandler_T) -> CommandHandler_T:
"""
Decorator to register a function as the arguments parser of
the corresponding command.
"""
self.cmd.args_parser_func = parser_func
return parser_func
def on_command(name: Union[str, CommandName_T], *,
aliases: Union[Iterable[str], str] = (),
permission: int = perm.EVERYBODY,
@ -209,7 +190,16 @@ def on_command(name: Union[str, CommandName_T], *,
for alias in aliases:
_aliases[alias] = cmd_name
return CommandFunc(cmd, func)
def args_parser(parser_func: CommandHandler_T) -> CommandHandler_T:
"""
Decorator to register a function as the arguments parser of
the corresponding command.
"""
cmd.args_parser_func = parser_func
return parser_func
func.args_parser = args_parser
return func
return deco