python argparse传入布尔参数不生效的解决方法

(相关资料图)

在一个需要用到flag作为信号控制代码中一些代码片段是否运行的，比如"--flag True"或者"--flag False"。但是古怪的是无法传入False，无论传入True还是False，程序里面都是True的参数，所以这个flag并没有生效，也就失去了意义。

参考代码：

import argparsedef test_bool():    parser = argparse.ArgumentParser(description="This code is used to test bool value.")    parser.add_argument("--flag",                        type=bool,                        default=True)    args = parser.parse_args()    print("# The type of flag: ", type(args.flag))    print(args.flag is False)    if args.flag:        print(f"# Flag: {True}")    else:        print(f"# Flag: {False}")if __name__ == "__main__":    test_bool()

执行上面的代码：python3 test.py --flag False按照执行的期望是输出# Flag: False，但是实际输出的结果是True：

(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag False# The type of flag:  False# Flag: True

为什么会出现这种情况呢？因为，在命令行的输入会被python认为是字符串，并不是布尔类型。

解决方式一【推荐】

更改代码为：

# ***parser.add_argument("--flag",                        action="store_true",                        help="Run or not.")# ***

这样的话，当你不输入--flag的时候，默认为False；输入--flag的时候，才会触发True值。看执行结果：

(data_parse)  ~/Desktop/code/cython&numba  python3 test.py# The type of flag:  True# Flag: False(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag# The type of flag:  False# Flag: True

当然这种方式，你也可以指定action为store_false，不输入的时候--flag默认True，输入--flag触发False。

解决方式二

这种方式稍微复杂一些，就是需要自己去写一个函数对输入的字符串值进行一个转换，由字符串转换为布尔值。代码：

import argparsedef 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.")def test_bool():    parser = argparse.ArgumentParser(description="This code is used to test bool value.")    parser.add_argument("--flag",                        type=str2bool,                        default=True,                        help="Run or not.")    args = parser.parse_args()    print("# The type of flag: ", type(args.flag))    print(args.flag is False)    if args.flag:        print(f"# Flag: {True}")    else:        print(f"# Flag: {False}")if __name__ == "__main__":    test_bool()

这样的话，当我们只要输入自定义函数中的那些对应TrueorFalse的选项，就能能触发对应的布尔类型。看执行情况：

(data_parse)  ~/Desktop/code/cython&numba  python3 test.py         # The type of flag:  False# Flag: True(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag True# The type of flag:  False# Flag: True(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag False# The type of flag:  True# Flag: False(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag 0    # The type of flag:  True# Flag: False(data_parse)  ~/Desktop/code/cython&numba  python3 test.py --flag 1# The type of flag:  False# Flag: True

由于默认了default的值为True，所以不输入--flag的时候就默认为True，然后输入--flag对应自定义func: str2bool()中的选项时，就能达到想要的开关效果。

关键词：