有一個除 __init__()
方法外只定義了一個方法的類。爲了簡化代碼,你想將它轉換成一個函數。
大多數情況下,可以使用閉包來將單個方法的類轉換成函數。 舉個例子,下面示例中的類允許使用者根據某個模板方案來獲取到URL鏈接地址。
from urllib.request import urlopen
class UrlTemplate:
def __init__(self, template):
self.template = template
def open(self, **kwargs):
return urlopen(self.template.format_map(kwargs))
# Example use. Download stock data from yahoo
yahoo = UrlTemplate('http://finance.yahoo.com/d/quotes.csv?s={names}&f={fields}')
for line in yahoo.open(names='IBM,AAPL,FB', fields='sl1c1v'):
print(line.decode('utf-8'))
這個類可以被一個更簡單的函數來代替:
def urltemplate(template):
def opener(**kwargs):
return urlopen(template.format_map(kwargs))
return opener
# Example use
yahoo = urltemplate('http://finance.yahoo.com/d/quotes.csv?s={names}&f={fields}')
for line in yahoo(names='IBM,AAPL,FB', fields='sl1c1v'):
print(line.decode('utf-8'))
大部分情況下,你擁有一個單方法類的原因是需要存儲某些額外的狀態來給方法使用。 比如,定義UrlTemplate類的唯一目的就是先在某個地方存儲模板值,以便將來可以在open()方法中使用。
使用一個內部函數或者閉包的方案通常會更優雅一些。簡單來講,一個閉包就是一個函數, 只不過在函數內部帶上了一個額外的變量環境。閉包關鍵特點就是它會記住自己被定義時的環境。 因此,在我們的解決方案中,opener()
函數記住了 template
參數的值,並在接下來的調用中使用它。
任何時候只要你碰到需要給某個函數增加額外的狀態信息的問題,都可以考慮使用閉包。 相比將你的函數轉換成一個類而言,閉包通常是一種更加簡潔和優雅的方案。