Python argparse：参数的可选值和位置值

Question

考虑以下用法：

usage: do.py [-h] [-s | -m] filename

这不是完整的用法。但我实际上希望 filename 成为文件的实际值，而不是：

--filename FILENAME

而且，filename 应该是可选的，所以我可以从标准输入中读取。想想 UNIX 上的 cat 程序。

你只是说：

cat filename

或

cat

编辑：现在，如果我在没有任何命令行选项的情况下执行程序do.py，我将得到一个error: too few arguments。相反，即使我没有给它一个有效的filename，我仍然希望它执行。我该怎么做？

Answer 1

更新 2：来自 ArgParse 文档，

nargs='?' 的更常见用法之一是允许可选的输入和输出文件：

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('infile', nargs='?', type=argparse.FileType('r'),
...                     default=sys.stdin)
>>> parser.add_argument('outfile', nargs='?', type=argparse.FileType('w'),
...                     default=sys.stdout)
>>> parser.parse_args(['input.txt', 'output.txt'])
Namespace(infile=<open file 'input.txt', mode 'r' at 0x...>,
          outfile=<open file 'output.txt', mode 'w' at 0x...>)
>>> parser.parse_args([])
Namespace(infile=<open file '<stdin>', mode 'r' at 0x...>,
          outfile=<open file '<stdout>', mode 'w' at 0x...>)

原答案：

这很简单：只需添加一个带有默认值和nargs='*' 的位置参数。如果参数为零，则使用默认值，否则将使用命令行中的参数：

>>> p = argparse.ArgumentParser()
>>> p.add_argument('filename', nargs='*', default=['-'])
>>> p.parse_args([])
Namespace(filename=['-'])
>>> p.parse_args(['abc'])
Namespace(filename=['abc'])

通常，- 用于指代标准输入/标准输出。

然后你做这样的事情：

def get_inputs(ns):
    """Iterate over input files."""
    for path in ns.filename:
        if path == '-':
            yield sys.stdin
        else:
            yield open(path, 'r')

更新：我假设你想要多个文件名，因为 cat 需要多个文件名。如果你想要零个或一个文件名，你可以使用nargs='?'。