軟件開發中的重要一條真理就是“不要重復自己的工作”。通常當我們需要創建高度重復的代碼時,都可以尋找到一個更加優雅的解決方案。
1 給函數添加一個包裝,讓它做一點額外的工作
當我們需要讓一個函數擁有計時統計、打印日志的功能時,往往選擇的方案就是直接在函數體中增加需要的代碼。這在只有一兩個函數的時候還可以接受,但是如果需要讓一個項目中的所有函數都具有這樣的功能時,就會變得十分繁瑣。
這時候,就需要我們使用“裝飾器”了。示例如下:
from functools import wraps
import time
def logit(func):
'''
使用裝飾器來打印函數調用信息
'''
@wraps(func)
def wrapper(*args, **kwargs):
print('start func {}'.format(func.__name__),
time.strftime('at %Y %m %d %H:%M:%S', time.localtime()))
result = func(*args, **kwargs)
print('finish func {}'.format(func.__name__),
time.strftime('at %Y %m %d %H:%M:%S', time.localtime()))
return result
return wrapper
下面是使用演示:
>>> @logit
def countdown(n):
while(n > 0):
n -= 1
>>> countdown(10000000)
start func countdown at 2018 09 16 16:45:23
finish func countdown at 2018 09 16 16:45:24
只需要在函數定義時為它增加一個裝飾器(@logit),這個函數就能告訴我們它開始運行的時間以及結束運行的時間!
裝飾器其實就是一個函數,它可以接受一個函數作為輸入并返回一個新的函數作為輸出。
裝飾器內部的代碼一般會涉及創建一個新的函數,利用*args和**kwargs可以接收任意的參數。在這個函數內部,我們調用原來的輸入函數(即被包裝的那個函數,它是裝飾器的輸入參數)并返回它的結果。此時,這個新創建的wrapper函數就會作為裝飾器的結果返回,取代了原本的函數。
需要強調的一點是,裝飾器一般來說不會修改調用的簽名,也不會修改被包裝函數返回的結果。這里使用了*args和**kwargs來確保可以接受任何形式的輸入參數。裝飾器的返回值幾乎總是同調用func(*args,**kwargs)的結果一致,這里的func就是那個未被包裝過的原始函數。
2 那裝飾器函數中的裝飾器@wraps有什么作用
裝飾器@wraps可以用來保存底層的元數據,比如函數名、文檔字符串、函數注解以及調用簽名。
如果沒有使用該裝飾器,獲取上一個例子中countdown函數的元數據就會看起來像這樣:
>>> countdown.__name__
'wrapper'
>>> countdown.__doc__
>>> countdown.__annotations__
@wraps的另一個重要特性就是可以通過__wrapped__屬性來訪問那個被包裝的函數,而該屬性同時也可以使得裝飾器函數可以合適的將底層被包裝的函數的簽名暴露出來:
>>> from inspect import signature
>>> print(signature(countdown))
(n:int)
另外有需要云服務器可以了解下創新互聯scvps.cn,海內外云服務器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業上云的綜合解決方案,具有“安全穩定、簡單易用、服務可用性高、性價比高”等特點與優勢,專為企業上云打造定制,能夠滿足用戶豐富、多元化的應用場景需求。
新聞標題:編寫裝飾器并保存函數的元數據,代碼簡潔之道-創新互聯
網站地址:http://vcdvsql.cn/article36/pggpg.html
成都網站建設公司_創新互聯,為您提供手機網站建設、移動網站建設、企業網站制作、網站排名、服務器托管、做網站
聲明:本網站發布的內容(圖片、視頻和文字)以用戶投稿、用戶轉載內容為主,如果涉及侵權請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網站立場,如需處理請聯系客服。電話:028-86922220;郵箱:631063699@qq.com。內容未經允許不得轉載,或轉載時需注明來源: 創新互聯