寫一個簡單的無參裝飾器,計算math_info所需要的時間
# 1, 基礎函數,計算出2的16次方的值
def math_info():
print("2**16的結果是:",2**64)
# 2, 計算wrApper所需要的時間
import time
def wrapper(func):
start_time=time.time()
res = math_info()
print("用時: ",float(time.time() - start_time))
return res
wrapper(math_info)
2**64的結果是: 18446744073709551616
用時: 0.004000186920166016
步驟2改變了原函數的調用方式,所以不能使用(違反了開放封閉原則)
# 3, 換一種為函數體傳值的方式,即將值包給函數
def timer(func):
def wrapper(): # 引用外部作用域的變量func
start_time=time.time()
res=func()
stop_time=time.time()
print("用時: ",float(time.time() - start_time))
return res
return wrapper
# 得到math_info=wrapper,wrapper攜帶對外作用域的引用:func=原始的math_info
math_info=timer(math_info)
# 執行的是wrapper(),在wrapper的函數體內再執行最原始的math_info
math_info()
步驟3寫法我們不推薦,因為代碼量變多了,最主要的是Python/ target=_blank class=infotextkey>Python為我們提供了一個語法糖,我們可以使用語法糖更簡便的寫出同樣的功能
# 完整的無參裝飾器
import time
def timer(func):
def wrapper(): # 引用外部作用域的變量func
start_time=time.time()
res=func()
stop_time=time.time()
print("用時: ",float(time.time() - start_time))
return res
return wrapper
@timer # @timer就相當于math_info=timer(math_info)
def math_info():
time.sleep(2)
print("2**64的結果是:",2**64)
math_info()
2**64的結果是: 18446744073709551616
用時: 2.004000186920166016
其實只要記住,裝飾器的目的是在不改變原函數代碼及調用方法的前提下添加新的功能