5

我正在使用 argparse 处理 cmd args，我想如果没有指定 args，则打印帮助消息，但现在解析会输出错误，然后退出。我的代码是：

def main():
    print "in abing/start/main"
    parser = argparse.ArgumentParser(prog="abing")#, usage="%(prog)s <command> [args] [--help]")
    parser.add_argument("-v", "--verbose", action="store_true", default=False, help="show verbose output")

    subparsers = parser.add_subparsers(title="commands")

    bkr_subparser = subparsers.add_parser("beaker", help="beaker inspection")
    bkr_subparser.set_defaults(command=beaker_command)
    bkr_subparser.add_argument("-m", "--max", action="store", default=3, type=int, help="max resubmit count")
    bkr_subparser.add_argument("-g", "--grain", action="store", default="J", choices=["J", "RS", "R", "T", "job", "recipeset", "recipe", "task"], type=str, help="resubmit selection granularity")
    bkr_subparser.add_argument("job_ids", nargs=1, action="store", help="list of job id to be monitored")

    et_subparser = subparsers.add_parser("errata", help="errata inspection")
    et_subparser.set_defaults(command=errata_command)
    et_subparser.add_argument("-w", "--workflows", action="store_true", help="generate workflows for the erratum")
    et_subparser.add_argument("-r", "--run", action="store_true", help="generate workflows, and run for the erratum")
    et_subparser.add_argument("-s", "--start-monitor", action="store_true", help="start monitor the errata system")
    et_subparser.add_argument("-d", "--daemon", action="store_true", help="run monitor into daemon mode")
    et_subparser.add_argument("erratum", action="store", nargs=1, metavar="ERRATUM", help="erratum id")

    if len(sys.argv) == 1:
        parser.print_help()
        return

    args = parser.parse_args()
    args.command(args)

    return

我怎样才能做到这一点？谢谢。

4

1 回答 1

3

解决方案在于argparse.ArgumentParser对其error()方法进行子类化和重新定义。事实上，一旦出错，ArgumentParser就会调用它的error()方法。然后可以通过子类而不是argparse.ArgumentParser. 在源代码中找到了一个模型error()函数argparse：

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)
    self.exit(2, '%s: error: %s\n' % (self.prog, message))

例如，可以在 中引发异常error()，而不是打印消息，以便代码调用parse_args()负责用户参数的问题。

原始答案：根据评论中的澄清，以下不起作用。但是，它提供了一种从子命令功能访问帮助消息的机制：

您几乎拥有它：在您的每个函数中，如果没有足够的参数*_command(args)，您可以测试大小并打印错误消息。args

如果您想使用自动生成的帮助，在您的命令函数中，您可以通过将子解析器传递给每个命令来获取它，如下所示：

args.command(subparsers, args)  # Instead of args.command(args)

然后，每个*_command()函数应该只接受两个参数而不是一个。可以通过以下方式访问自动生成的帮助：

subparsers.choices['beaker'].print_help()  # or print_usage()

例如。

您也可以选择将特定的子解析器直接传递给每个子命令例程*_command()：

args.command(subparsers.choices[sys.argv[1]], args)

然后，在每个 中*_command(subparser, args)，使用 打印帮助subparser.print_help()。

于 2010-12-30T10:40:01.480 回答