python - 从文件或 STDIN 读取

Question

我编写了一个命令行实用程序，它使用 getopt 来解析命令行上给出的参数。我还希望文件名是一个可选参数，例如它在其他实用程序中，如 grep、cut 等。所以，我希望它具有以下用法

tool -d character -f integer [filename]

如何实现以下功能？

• 如果给出了文件名，则从文件中读取。
• 如果没有给出文件名，则从 STDIN 读取。

Answer 1

fileinput模块可以做你想做的事——假设非选项参数在 args那么:

import fileinput
for line in fileinput.input(args):
    print line

如果 args为空则 fileinput.input()将从标准输入读取；否则它会依次从每个文件中读取，与 Perl 的 while(<>) 类似。 .