Python эквивалент Perl while (<>) {...}?

Question

Я пишу много маленьких скриптов, которые обрабатывают файлы по очереди. В Perl, я использую

while (<>) { 
    do stuff; 
} 

Это удобно, потому что он не заботится, где вход поступает из (файла или стандартного ввода).

В Python Я использую этот

if len(sys.argv) == 2: # there's a command line argument 
    sys.stdin = file(sys.argv[1]) 
for line in sys.stdin.readlines(): 
    do stuff 

, который, кажется, не очень элегантно. Есть ли идиома Python, которая легко обрабатывает входной файл/stdin?

Answer 1

Модуль FileInput в стандартной библиотеке именно то, что вы хотите:

import fileinput 

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

Answer 2

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

Это перебирает строки всех файлов, перечисленных в sys.argv [1:], недобросовестный sys.stdin если список пуст.

Answer 3

fileinput defaults to stdin, так что сделайте его немного более кратким.

Если вы делаете много сообщений в командной строке, то этот piping hack очень опрятный.