通过稍微改变你的代码:
import argparse
import sys
parser = argparse.ArgumentParser()
parser.add_argument('integers', metavar='N', type=int, nargs='+',
help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
const=sum, default=sum,
help='sum the integers (default: find the max)')
parser.add_argument("--output_dir", type=str, default="data/xx")
def main(args):
#os.makedirs(args.output_dir) # XXX: Commented out for debugging
print(args.accumulate(args.integers))
if __name__=='__main__':
print(sys.argv)
args = parser.parse_args() # Disable during debugging @ Run through terminal
# args = argparse.Namespace(integers = 1, output_dir= 'mydata_223ss32') # Disable when run through terminal: For debugging process
print(args)
main(args)
我们可以看到,如果我们使用以下命令调用脚本:./test3.py --output_dir foo 1 2 3
我们的输出是:
['test3.py', '--output_dir', 'foo', '1', '2', '3']
Namespace(accumulate=<built-in function sum>, integers=[1, 2, 3], output_dir='foo')
6
因此,如果您想模拟命令行参数,您有两个选择:
1) Edit sys.argv
(个人首选):
if __name__=='__main__':
# Override sys.argv
sys.argv = ['test3.py', '--output_dir', 'foo', '1', '2', '3']
args = parser.parse_args()
main(args)
2)创建一个Namespace
object:
if __name__=='__main__':
#args = parser.parse_args() # Disable during debugging @ Run through terminal
args = argparse.Namespace(accumulate=sum, integers=[1,2,3], output_dir='foo')
main(args)
第二种方法的问题有两个。首先,通过执行此操作,您将跳过调试过程中非常重要的部分,调试 argparse 配置并确保按照您的预期解析参数。其次,必须在此处定义由 argparse 配置添加的隐式默认值(例如,您的--sum
争论/accumulate
)。如果您采用第一种方法,argparse 将处理sys.argv
并添加累积,而无需进行任何更改。在第二种方法中,我们必须添加accumulate=sum
让它按预期运行。