ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-02-21 01:55:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix #98

Browse Source
This commit is contained in:
Richard Chien 2020-02-10 13:26:33 +08:00
parent 3d030fdd03
commit 76f1dfc2a5
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
nonebot/argparse.py
View File

@ -19,7 +19,11 @@ class ArgumentParser(ArgumentParser):
self.session = kwargs.pop('session', None)
super().__init__(*args, **kwargs)
def _print_message(self, *args, **kwargs):
def _session_finish(self, message):
if self.session and isinstance(self.session, CommandSession):
self.session.finish(message)
def _print_message(self, message, file=None):
# do nothing
pass
@ -27,18 +31,11 @@ class ArgumentParser(ArgumentParser):
raise ParserExit(status=status, message=message)
def parse_args(self, args=None, namespace=None):
def finish(msg):
if self.session and isinstance(self.session, CommandSession):
self.session.finish(msg)
if not args:
finish(self.usage)
else:
try:
return super().parse_args(args=args, namespace=namespace)
except ParserExit as e:
if e.status == 0:
# --help
finish(self.usage)
else:
finish('参数不足或不正确，请使用 --help 参数查询使用帮助')
try:
return super().parse_args(args=args, namespace=namespace)
except ParserExit as e:
if e.status == 0:
# --help
self._session_finish(self.usage or self.format_help())
else:
self._session_finish('参数不足或不正确，请使用 --help 参数查询使用帮助')