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

⚡ store message in ParserExit exception

Browse Source
This commit is contained in:
yanyongyu 2021-03-26 11:25:13 +08:00
parent b861149e0b
commit 1f585d8bcb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nonebot/rule.py
View File

@ -294,7 +294,8 @@ class ArgumentParser(ArgParser):
setattr(self, "message", old_message) setattr(self, "message", old_message)
def exit(self, status=0, message=None): def exit(self, status=0, message=None):
raise ParserExit(status=status, message=message) raise ParserExit(status=status,
message=message or getattr(self, "message", None))
def parse_args(self, def parse_args(self,
args: Optional[Sequence[str]] = None, args: Optional[Sequence[str]] = None,
@ -369,7 +370,7 @@ def shell_command(*cmds: Union[str, Tuple[str, ...]],
args = parser.parse_args(state["argv"]) args = parser.parse_args(state["argv"])
state["args"] = args state["args"] = args
except ParserExit as e: except ParserExit as e:
state["args"] = getattr(parser, "message", None) or e state["args"] = e
return True return True
else: else:
return False return False