python - 从自定义操作中捕获 ArgumentTypeError 异常

Question

从我自己的自定义操作中抛出 ArgumentTypeError 异常并让 argparse 为我捕获它的最佳做法是什么？

argparse 的 try/except block 似乎没有为我的自定义操作处理此异常。尽管它的内置操作做得很好。

class unique_server_endpoints(argparse.Action):
    """This class avoids having two duplicate OuterIPs in the client argument list"""
    def __call__(self, parser, namespace, values, option_string=None):
        ips = set()
        endpoints = []
        for r in values:
            ep = server_endpoint(r)
            if ep[0] in ips:
                raise argparse.ArgumentTypeError("Duplicate OuterIPs found")
            else:
                ips.add(ep[0])
                endpoints.append(ep)
        setattr(namespace, self.dest, endpoints)

和

group.add_argument('-c', "--client", nargs = 2,
            dest = "servers", action = unique_server_endpoints,

例如，在上面的代码中，如果我有重复的 IP，那么异常将落到 ma​​in 函数并打印丑陋的堆栈跟踪。我不想这样，我也不想在 ma​​in 中放置一个 try/except block 。

Answer 1

在查看 argparse 源代码后，我发现它将 ArgumentTypeError 转换为 ArgumentError 异常。

所以代替:

            raise argparse.ArgumentTypeError("Duplicate OuterIPs found")

我应该:

            raise argparse.ArgumentError(self, "Duplicate OuterIPs found")

argparse 仍然会为我完成剩下的工作(捕获异常并打印使用信息)...