【Python标准库:fileinput】优雅的读取文件

每当我写个脚本涉及文件输入时，一般写法都是下面这种

import sys

file_in = sys.argv[1]

for line in open(file_in, 'r'):
    commands

这个操作有一个缺点就是，如果我想从管道里面传入数据给Python的话，就会报错，因为原代码要求文件而不是标准输入。

这个问题可以通过Python的一个标准库: fileinput进行解决。

import fileinput

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

fileinput.intpu()会帮我们自动处理输入。如果sys.argv[1:]里有输入文件，它就会对里面所有的文件进行遍历，如果sys.argv为空，那么它就会从标准输入sys.stdin里读取输入，如果输入文件的文件名是"-", 同样地会从标准输入里读取输入。这样子就省去了我们自己写条件语句进行判断输入类型。