Python tương đương với Perl trong khi (<>) {...}?

Question

Tôi viết rất nhiều tập lệnh nhỏ xử lý các tệp trên cơ sở từng dòng. Trong Perl, tôi sử dụng

while (<>) { 
    do stuff; 
} 

Điều này rất tiện lợi vì nó không quan tâm đến đầu vào (tệp hoặc stdin).

Trong Python tôi sử dụng này

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 

mà không có vẻ rất thanh lịch. Có một thành ngữ Python có thể dễ dàng xử lý đầu vào tập tin/stdin?

Answer 1

Module fileinput trong thư viện chuẩn chỉ là những gì bạn muốn:

import fileinput 

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

Answer 2

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

này lặp trên các dòng của tất cả các file được liệt kê trong sys.argv [1:], mặc định cho sys.stdin nếu danh sách trống.

Answer 3

fileinput mặc định là stdin, vì vậy sẽ làm cho nó ngắn gọn hơn một chút.

Nếu bạn thực hiện rất nhiều công cụ dòng lệnh, tuy nhiên, điều này piping hack rất gọn gàng.