python argparse 使用[-h]时如何才能不退出程序

Question

argparse 在使用[-h]时显示帮助，然后就退出了，如何才能不退出交互界面，等待下一个输入？

while True:
    cmd = input('>>>')
    parser = argparse.ArgumentParser()
    parser.add_argument('-f', help='foo')
    parser.parse_args(cmd.split())

这样输入[-h]的时候：

>>>-h
usage: test.py [-h] [-f F]

optional arguments:
  -h, --help  show this help message and exit
  -f F        foo

程序就结束了，现在我只要'show this help message'，而不需要'exit'，请问要如何做？谢谢！

Answer 1

根本原因在于, argparse.parse_args() 会在出错和 -h 时执行 sys.exit() . 这个可以通过捕获 SystemExit 异常来解决:

import argparse

while True:
    cmd = input('>>>')
    parser = argparse.ArgumentParser()
    parser.add_argument('-f', help='foo')

    try:
        parser.parse_args(cmd.split())
    except SystemExit:
        print("ignoring SystemExit")

另外1: 建议不要把初始化代码放在循环块中, 我觉得这样写会更好:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-f', help='foo')

while True:
    cmd = input('>>>')
    try:
        parser.parse_args(cmd.split())
    except SystemExit:
        print("ignoring SystemExit")

另外2: 提出 python 问题时, 不妨注明你用的是 python2 还是 python3. 你的代码中, 有些规则在两个版本间有区别.

参考 - argparse.parse_args() 源码片段:

    def parse_args(self, args=None, namespace=None):
        args, argv = self.parse_known_args(args, namespace)
        if argv:
            msg = _('unrecognized arguments: %s')
            self.error(msg % ' '.join(argv))
        return args

class _HelpAction(Action):

    def __init__(self,
                 option_strings,
                 dest=SUPPRESS,
                 default=SUPPRESS,
                 help=None):
        super(_HelpAction, self).__init__(
            option_strings=option_strings,
            dest=dest,
            default=default,
            nargs=0,
            help=help)

    def __call__(self, parser, namespace, values, option_string=None):
        parser.print_help()
        parser.exit()

    def exit(self, status=0, message=None):
        if message:
            self._print_message(message, _sys.stderr)
        _sys.exit(status)

    def error(self, message):
        """error(message: string)

        Prints a usage message incorporating the message to stderr and
        exits.

        If you override this in a subclass, it should not return -- it
        should either exit or raise an exception.
        """
        self.print_usage(_sys.stderr)
        args = {'prog': self.prog, 'message': message}
        self.exit(2, _('%(prog)s: error: %(message)s\n') % args)