Equivalente Python di Perl's while (<>) {...}?

Question

Scrivo un sacco di piccoli script che elaborano i file riga per riga. In Perl, io uso

while (<>) { 
    do stuff; 
} 

Questo è comodo perché non importa dove l'input proviene da (un file o stdin).

In Python Io uso questo

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 

che non sembra molto elegante. Esiste un idioma Python che gestisce facilmente l'input di file/stdin?

Answer 1

Il modulo fileinput nella libreria standard è proprio quello che si vuole:

import fileinput 

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

Answer 2

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

Questo itera oltre le linee di tutti i file elencati in sys.argv [1:], inadempiente a sys.stdin se la lista è vuota

Answer 3

fileinput predefinito per lo stdin, quindi lo renderebbe leggermente più conciso.

Se si esegue molta roba da riga di comando, tuttavia, questo piping hack è molto ordinato.