Python argparse切换标志

在argparse中有什么方法可以解析像 [+-]a,b,c,d 这样的标志吗?

foo.py +s -b

应该在 destdest 中存储True,在 bdest 中存储False,就像在Windows attrib 或Linux chmod 中完成的那样 .

目前,我分别使用2个单独的参数 +s-sstore_truestore_false . 但它创造了一个丑陋的帮助,它列出每个标志两次(a&-a)

另一种解决方法是使用正则表达式手动解析扩展的arg(这在某种程度上似乎更容易并且使用自定义描述,但在此之前我只想环顾四周是否有任何使用argparse本身可以执行相同操作的东西 .

回答(1)

2 years ago

您可以通过将 -s+s 传递给单个 add_argument 调用并使用自定义操作来执行此操作:

class ToggleAction(argparse.Action):
    def __call__(self, parser, ns, values, option):
        setattr(ns, self.dest, bool("-+".index(option[0])))
ap = ArgumentParser(prefix_chars='-+')
ap.add_argument('-s', '+s', action=ToggleAction, nargs=0)

ap.parse_args(['+s'])
Namespace(s=True)

ap.parse_args(['-s'])
Namespace(s=False)