在 python 中调试 argparse

我可以知道调试 argpars 函数的最佳实践是什么。

假设我有一个 py 文件 test_file.py ，其中包含以下几行

# Script start
import argparse
import os
parser = argparse.ArgumentParser()
parser.add_argument(“–output_dir”, type=str, default=”/data/xx”)
args = parser.parse_args()
os.makedirs(args.output_dir)
# Script stop

上述脚本可以通过以下方式从终端执行：

python test_file.py –output_dir data/xx

但是，对于调试过程，我想避免使用终端。因此解决方法是

# other line were commented for debugging process
# Thus, active line are
# Script start
import os
args = {“output_dir”:”data/xx”}
os.makedirs(args.output_dir)
#Script stop

但是，我无法执行修改后的脚本。我可以知道我错过了什么吗？

当用作脚本时，parse_args将产生一个Namespace对象，显示为：

argparse.Namespace(output_dir='data/xx')

then

args.output_dir

将是该属性的值

在测试中，您可以做以下几件事：

args = parser.parse_args([....])  # a 'fake' sys.argv[1:] list

args = argparse.Namespace(output_dir= 'mydata')

and use args像之前一样。或者直接致电

os.makedirs('data/xx')

我建议将脚本组织为：

# Script start
import argparse
import os
# this parser definition could be in a function
parser = argparse.ArgumentParser()
parser.add_argument(“–output_dir”, type=str, default=”/data/xx”)

def main(args):
    os.makedirs(args.output_dir)

if __name__=='__main__':
    args = parser.parse_args()
    main(args)

这样一来parse_args导入文件时不运行该步骤。无论你通过args Namespace to main或传递值，例如args.output_dir，或者一本字典等等都是你的选择。