如何不调用argparse.Namespace私有方法覆盖带默认值的列表型参数?
解决方案
一、仅靠命令行实现的限制
因为你设置了nargs="+",这个参数规则要求必须至少传入一个值,所以直接在命令行不给--list传任何值会触发报错。如果只能修改命令行,唯一可行的方式是传空字符串(比如python test.py --list ""),但此时args.list会是[""],还是需要代码过滤空值才能得到空列表——不过代码过滤完全不需要调用私有方法。
二、无需私有方法的代码实现
别用args._get_kwargs()这种私有API,直接针对已知的参数名显式处理就行,简单直接还合规:
parser.add_argument("--list", type=str, nargs="+", default=["arg"]) args = parser.parse_args() # 直接处理目标参数,完全避开私有方法 args.list = list(filter(None, args.list))
如果有多个列表型参数需要处理,手动列出参数名遍历即可,同样不依赖私有方法:
parser.add_argument("--list", type=str, nargs="+", default=["arg"]) parser.add_argument("--other-list", type=str, nargs="+", default=["foo"]) args = parser.parse_args() # 手动指定要过滤的列表参数名 list_params = ["list", "other_list"] for param in list_params: val = getattr(args, param) if isinstance(val, list): setattr(args, param, list(filter(None, val)))
额外选项:彻底靠命令行得到空列表(需修改代码)
如果想完全不用代码过滤,直接通过命令行得到空列表,需要调整参数定义:把nargs="+"改成nargs="*"。这样命令行执行python test.py --list(后面不加任何参数)时,args.list就会是空列表,直接覆盖默认值。但这属于修改代码的操作,若不能改动代码,就只能用前面命令行传空字符串+显式过滤的方式。
内容的提问来源于stack exchange,提问作者aweeeezy
相关产品推荐
相关产品推荐

