Python代码中STDIN和Fileinput的通用构造

Question

我希望我的代码能够接受来自文件和标准输入的输入.这样做的构造是什么？

我的意思是一个统一的结构

file1 = sys.stdin

和

file1 = fileinput.input(sys.argv[1])

Answer 1

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