python-2.7 - 使用 argparse 时如何处理输出文件？

Question

我有一个程序，我想制作它，以便如果用户指定给定的 2 个输出文件的确切名称，那么这些文件将根据用户的喜好命名。

例如：

-o file1.txt file2.txt

如果未指定输出文件，则脚本将自动生成具有默认名称的文件。

Answer 1

您可以使用 argparse 模块，它提供了处理输入参数的简洁方法。对于您的任务，您可以使用类似

arguments = argparse.ArgumentParser()
arguments.add_argument('fileNames', nargs='*', help='Output file names', default = ['val1', 'val2'])
inputArgs = arguments.parse_args()

# User may decide to give just one file name
outFileName1 = inputArgs.fileNames[0]
outFileName2 = 'val2' if len(inputArgs.fileNames) == 1 else inputArgs.fileNames[1]