从文件读取或STDIN

Question

我已经编写了一个命令行实用程序，它使用getopt来解析命令行中给出的参数。我也想有一个文件名是一个可选的参数，如它在其他实用程序，如grep，剪切等，所以，我想它有以下用途

tool -d character -f integer [filename] 

我如何能实现以下？

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

Answer 1

简单的说：

import sys 
# parse command line 
if file_name_given: 
    inf = open(file_name_given) 
else: 
    inf = sys.stdin 

此时你会使用inf从文件中读取。根据是否给出文件名，这将从给定文件或标准输入读取。

当您需要关闭文件，你可以这样做：

if inf is not sys.stdin: 
    inf.close() 

然而，在大多数情况下，它是无害关闭sys.stdin，如果你用它做。

Answer 2

的fileinput模块可以做你想做的 - 假设非选项参数是args则：

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

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

Answer 3

喜欢的东西：

if input_from_file: 
    f = open(file_name, "rt") 
else: 
    f = sys.stdin 
inL = f.readline() 
while inL: 
    print inL.rstrip() 
    inL = f.readline() 

Answer 4

要使用Python的with语句，可以使用下面的代码：

import sys 
with open(sys.argv[1], 'r') if len(sys.argv) > 1 else sys.stdin as f: 
    # read data using f 
    # ...... 

Answer 5

我更愿意用“ - ”为你应该阅读的指标从stdin，它更明确：

import sys 
with open(sys.argv[1], 'r') if sys.argv[1] is not "-" else sys.stdin as f: 
    pass # do something here 

Answer 6

我喜欢使用上下文管理的一般习语呃，但是当你不在我想要避免的with声明中时，（太）无效的解决方案最终会结束sys.stdin。

从this answer借款，这里是一个解决办法：

import sys 
import contextlib 

@contextlib.contextmanager 
def _smart_open(filename, mode='Ur'): 
    if filename == '-': 
     if mode is None or mode == '' or 'r' in mode: 
      fh = sys.stdin 
     else: 
      fh = sys.stdout 
    else: 
     fh = open(filename, mode) 
    try: 
     yield fh 
    finally: 
     if filename is not '-': 
      fh.close() 

if __name__ == '__main__': 
    args = sys.argv[1:] 
    if args == []: 
     args = ['-'] 
    for filearg in args: 
     with _smart_open(filearg) as handle: 
      do_stuff(handle) 

我想你可以实现something similar with os.dup()，但我做了这样做竟然是更复杂，更神奇的代码，而上面是有点笨重但很直接。