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

問(wèn)題

你在代碼中使用while循環(huán)來(lái)迭代處理數(shù)據(jù)，因?yàn)樗枰{(diào)用某個(gè)函數(shù)或者和一般迭代模式不同的測(cè)試條件。能不能用迭代器來(lái)重寫(xiě)這個(gè)循環(huán)呢？

解決方案

一個(gè)常見(jiàn)的IO操作程序可能會(huì)想下面這樣：

CHUNKSIZE = 8192

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

這種代碼通常可以使用 iter() 來(lái)代替，如下所示：

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

如果你懷疑它到底能不能正常工作，可以試驗(yàn)下一個(gè)簡(jiǎn)單的例子。比如：

>>> 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ù)一個(gè)鮮為人知的特性是它接受一個(gè)可選的 callable 對(duì)象和一個(gè)標(biāo)記(結(jié)尾)值作為輸入?yún)?shù)。當(dāng)以這種方式使用的時(shí)候，它會(huì)創(chuàng)建一個(gè)迭代器， 這個(gè)迭代器會(huì)不斷調(diào)用 callable 對(duì)象直到返回值和標(biāo)記值相等為止。

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