从文件或STDIN读取

我写了一个命令行实用程序,它使用getopt来parsing命令行中给出的参数。 我也想有一个文件名是一个可选的参数,如它在其他工具如grep,剪切等。所以,我希望它有以下用法

tool -d character -f integer [filename] 

我怎样才能实现以下?

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

6 Solutions collect form web for “从文件或STDIN读取”

最简单的说法是:

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

此时,您将使用inf从文件读取。 根据是否给出文件名,这将从给定的文件或标准input读取。

当你需要closures文件时,你可以这样做:

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

但是,在大多数情况下,closuressys.stdin是完全无害的。

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

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

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

我喜欢使用上下文pipe理器的一般习惯用法,但是当你不在with语句中时,(太)小问题的解决scheme最终会closuressys.stdin ,这是我想避免的。

从这个答案借用, 这是一个解决方法:

 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) 

我想你可以用os.dup()来实现类似的function,但是我准备做的代码变得更复杂,更神奇,而上面的代码有些笨重,但是很直接。

要使用Python的语句,可以使用下面的代码:

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

我更喜欢使用“ – ”作为一个指标,你应该从标准input读取,这是更明确的:

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

就像是:

 if input_from_file: f = open(file_name, "rt") else: f = sys.stdin inL = f.readline() while inL: print inL.rstrip() inL = f.readline() 
  • 在Java中读取System.in最快的方法是什么?
  • nodejs如何从标准input读取按键
  • 如何在CLI应用程序中“截取”Ctrl + C?
  • 在Python中运行一个交互命令
  • 欺骗一个应用程序,认为它的标准input是交互式的,而不是pipe道
  • pythonsubprocess中的多个input和输出进行通信
  • 如何将variables的值传递给命令的stdin?
  • 如何让SSH从标准input接收密码
  • 发送string到标准input
  • 如何从外部进程将数据写入现有进程的STDIN?
  • 如何从C中的控制台读取一行?