4.16 迭代器代替while無限循環(huán)

問題

你在代碼中使用while循環(huán)來迭代處理數(shù)據(jù)，因為它需要調(diào)用某個函數(shù)或者和一般迭代模式不同的測試條件。能不能用迭代器來重寫這個循環(huán)呢？

解決方案

一個常見的IO操作程序可能會想下面這樣：

CHUNKSIZE = 8192

def reader(s):
    while True:
        data = s.recv(CHUNKSIZE)
        if data == b'':
            break
        process_data(data)

這種代碼通?？梢允褂?iter() 來代替，如下所示：

def reader2(s):
    for chunk in iter(lambda: s.recv(CHUNKSIZE), b''):
        pass
        # process_data(data)

如果你懷疑它到底能不能正常工作，可以試驗下一個簡單的例子。比如：

>>> import sys
>>> f = open('/etc/passwd')
>>> for chunk in iter(lambda: f.read(10), ''):
...     n = sys.stdout.write(chunk)
...
nobody:*:-2:-2:Unprivileged User:/var/empty:/usr/bin/false
root:*:0:0:System Administrator:/var/root:/bin/sh
daemon:*:1:1:System Services:/var/root:/usr/bin/false
_uucp:*:4:4:Unix to Unix Copy Protocol:/var/spool/uucp:/usr/sbin/uucico
...
>>>

討論

iter 函數(shù)一個鮮為人知的特性是它接受一個可選的 callable 對象和一個標記(結(jié)尾)值作為輸入?yún)?shù)。當以這種方式使用的時候，它會創(chuàng)建一個迭代器， 這個迭代器會不斷調(diào)用 callable 對象直到返回值和標記值相等為止。

這種特殊的方法對于一些特定的會被重復(fù)調(diào)用的函數(shù)很有效果，比如涉及到I/O調(diào)用的函數(shù)。舉例來講，如果你想從套接字或文件中以數(shù)據(jù)塊的方式讀取數(shù)據(jù)，通常你得要不斷重復(fù)的執(zhí)行 read() 或 recv()，并在后面緊跟一個文件結(jié)尾測試來決定是否終止。這節(jié)中的方案使用一個簡單的 iter() 調(diào)用就可以將兩者結(jié)合起來了。其中l(wèi)ambda函數(shù)參數(shù)是為了創(chuàng)建一個無參的callable對象，并為 recv 或 read() 方法提供了size參數(shù)。