使用argparse分析布尔值

Question

我想使用argparse解析写为“--foo true”或“--foo false”的布尔命令行参数。例如：

my_program --my_boolean_flag False

但是，下面的测试代码并没有做我想做的事情：

import argparse
parser = argparse.ArgumentParser(description="My parser")
parser.add_argument("--my_bool", type=bool)
cmd_line = ["--my_bool", "False"]
parsed_args = parser.parse(cmd_line)

遗憾的是，parsed_args.my_bool的计算结果为true。即使我将cmd_line更改为[“--my_bool”,“”]时也是如此，这是令人惊讶的，因为bool（“”）的计算结果为false。

如何使argparse将“false”、“f”及其小写变体解析为false？

Answer 1

我认为更规范的方法是通过：

command --feature

而且

command --no-feature

argparse很好地支持此版本：

parser.add_argument('--feature', dest='feature', action='store_true')
parser.add_argument('--no-feature', dest='feature', action='store_false')
parser.set_defaults(feature=True)

当然，如果您确实想要--arg 版本，您可以将ast.literal_eval作为“类型”或用户定义的函数传递...

def t_or_f(arg):
    ua = str(arg).upper()
    if 'TRUE'.startswith(ua):
       return True
    elif 'FALSE'.startswith(ua):
       return False
    else:
       pass  #error condition maybe?

Answer 2

另一种解决方案使用了前面的建议，但带有argparse中的“正确”解析错误：

def str2bool(v):
    if isinstance(v, bool):
       return v
    if v.lower() in ('yes', 'true', 't', 'y', '1'):
        return True
    elif v.lower() in ('no', 'false', 'f', 'n', '0'):
        return False
    else:
        raise argparse.ArgumentTypeError('Boolean value expected.')

这对于使开关具有默认值非常有用；例如：

parser.add_argument("--nice", type=str2bool, nargs='?',
                        const=True, default=False,
                        help="Activate nice mode.")

允许我使用：

script --nice
script --nice <bool>

并且仍然使用默认值（特定于用户设置）。这种方法的一个（间接相关的）缺点是，'nargs'可能会捕获位置参数--请参阅这个相关问题和这个argparse bug报告。