我想使用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解析“假”，“F”，和他们的小写变体为假?
当前回答
我认为最规范的方式是:
parser.add_argument('--ensure', nargs='*', default=None)
ENSURE = config.ensure is None
其他回答
oneliner:
parser.add_argument('--is_debug', default=False, type=lambda x: (str(x).lower() == 'true'))
下面是另一种变体，没有额外的行/秒来设置默认值。布尔值总是被赋值，这样它就可以在逻辑语句中使用，而无需事先检查:
import argparse
parser = argparse.ArgumentParser(description="Parse bool")
parser.add_argument("--do-something", default=False, action="store_true",
help="Flag to do something")
args = parser.parse_args()
if args.do_something:
print("Do something")
else:
print("Don't do something")
print(f"Check that args.do_something={args.do_something} is always a bool.")
我认为更规范的做法是:
command --feature
and
command --no-feature
Argparse很好地支持这个版本:
Python 3.9 +:
parser.add_argument('--feature', action=argparse.BooleanOptionalAction)
Python < 3.9:
parser.add_argument('--feature', action='store_true')
parser.add_argument('--no-feature', dest='feature', action='store_false')
parser.set_defaults(feature=True)
当然，如果你真的想要——arg <True|False>版本，你可以传递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?
只要执行以下操作，就可以使用——test = True
Python文件名——test
parser.add_argument("--test" , default=False ,help="test ?", dest='test', action='store_true')
简单快捷，但仅适用于参数0或1:
parser.add_argument("mybool", default=True,type=lambda x: bool(int(x)))
myargs=parser.parse_args()
print(myargs.mybool)
从终端调用后输出将为“False”:
python myscript.py 0
