Python 从命令行参数或标准输入中读取

Question

在用 Python 编写面向文本的命令行程序时，我经常想读取命令行上传递的所有文件，或者 (XOR) 标准输入(如 Unix cat 或 Perl 的 <> )。所以，我说

if len(args) == 0:  # result from optparse
    input = sys.stdin
else:
    input = itertools.chain(*(open(a) for a in args))

这是执行此操作的 Pythonic 方式，还是我错过了库的某些部分？

Answer 1

你需要fileinput .

标准用例是:

import fileinput
for line in fileinput.input():
  process(line)