本篇內(nèi)容主要講解“如何使用裝飾器”,感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷,實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“如何使用裝飾器”吧!
創(chuàng)新互聯(lián)建站為企業(yè)級客戶提高一站式互聯(lián)網(wǎng)+設(shè)計(jì)服務(wù),主要包括網(wǎng)站設(shè)計(jì)制作、網(wǎng)站設(shè)計(jì)、成都app開發(fā)、微信小程序、宣傳片制作、LOGO設(shè)計(jì)等,幫助客戶快速提升營銷能力和企業(yè)形象,創(chuàng)新互聯(lián)各部門都有經(jīng)驗(yàn)豐富的經(jīng)驗(yàn),可以確保每一個(gè)作品的質(zhì)量和創(chuàng)作周期,同時(shí)每年都有很多新員工加入,為我們帶來大量新的創(chuàng)意。
下面這是一個(gè)最簡單的裝飾器示例,在運(yùn)行 myfunc 函數(shù)的前后都會(huì)打印一條日志。
def deco(func): def wrapper(*args, **kw): print("Ready to run task") func(*args, **kw) print("Successful to run task") return wrapper @deco def myfunc(): print("Running the task") myfunc()
裝飾器使用起來,似乎有些高端和魔幻,對于一些重復(fù)性的功能,往往我們會(huì)封裝成一個(gè)裝飾器函數(shù)。
在定義一個(gè)裝飾器的時(shí)候,我們都需要像上面一樣機(jī)械性的寫一個(gè)嵌套的函數(shù),對裝飾器原理理解不深的初學(xué)者,往往過段時(shí)間就會(huì)忘記如何定義裝飾器。
有一些比較聰明的同學(xué),會(huì)利用 PyCharm 來自動(dòng)生成裝飾器模板
然后要使用的時(shí)候,直接敲入 deco 就會(huì)生成一個(gè)簡單的生成器代碼,提高編碼的準(zhǔn)備效率
使用 PyCharm 的 Live Template ,雖然能降低編寫裝飾器的難度,但卻要依賴 PyCharm 這一專業(yè)的代碼編輯器。
這里,明哥要教你一個(gè)更加簡單的方法,使用這個(gè)方法呢,你需要先安裝一個(gè)庫 :decorator,使用 pip 可以很輕易地去安裝它
$ python3 -m pip install decorator
從庫的名稱不難看出,這是一個(gè)專門用來解決裝飾器問題的第三方庫。
有了它之后,你會(huì)驚奇的發(fā)現(xiàn),以后自己定義的裝飾器,就再也不需要寫嵌套的函數(shù)了
from decorator import decorator @decorator def deco(func, *args, **kw): print("Ready to run task") func(*args, **kw) print("Successful to run task") @deco def myfunc(): print("Running the task") myfunc()
deco 作為裝飾函數(shù),第一個(gè)參數(shù)是固定的,都是指被裝飾函數(shù),而后面的參數(shù)都固定使用 可變參數(shù) *args 和 **kw 的寫法,代碼被裝飾函數(shù)的原參數(shù)。
這種寫法,不得不說,更加符合直覺,代碼的邏輯也更容易理解。
裝飾器根據(jù)有沒有攜帶參數(shù),可以分為兩種
第一種:不帶參數(shù),最簡單的示例,上面已經(jīng)舉例
def decorator(func): def wrapper(*args, **kw): func(*args, **kw) return wrapper
第二種:帶參數(shù),這就相對復(fù)雜了,理解起來了也不是那么容易。
def decorator(arg1, arg2): def wrapper(func): def deco(*args, **kwargs) func(*args, **kwargs) return deco return wrapper
那么對于需要帶參數(shù)的裝飾器,decorator 是否也一樣能很好的支持呢?
下面是一個(gè)官方的示例
from decorator import decorator @decorator def warn_slow(func, timelimit=60, *args, **kw): t0 = time.time() result = func(*args, **kw) dt = time.time() - t0 if dt > timelimit: logging.warn('%s took %d seconds', func.__name__, dt) else: logging.info('%s took %d seconds', func.__name__, dt) return result @warn_slow(timelimit=600) # warn if it takes more than 10 minutes def run_calculation(tempdir, outdir): pass
可以看到:
裝飾函數(shù)的第一個(gè)參數(shù),還是被裝飾器 func ,這個(gè)跟之前一樣
而第二個(gè)參數(shù) timelimit 寫成了位置參數(shù)的寫法,并且有默認(rèn)值
再往后,就還是跟原來一樣使用了可變參數(shù)的寫法
不難推斷,只要你在裝飾函數(shù)中第二個(gè)參數(shù)開始,使用了非可變參數(shù)的寫法,這些參數(shù)就可以做為裝飾器調(diào)用時(shí)的參數(shù)。
我們在自己寫裝飾器的時(shí)候,通常都會(huì)順手加上一個(gè)叫 functools.wraps 的裝飾器,我想你應(yīng)該也經(jīng)常見過,那他有啥用呢?
先來看一個(gè)例子
def wrapper(func): def inner_function(): pass return inner_function @wrapper def wrapped(): pass print(wrapped.__name__) #inner_function
為什么會(huì)這樣子?不是應(yīng)該返回 func 嗎?
這也不難理解,因?yàn)樯线厛?zhí)行func 和下邊 decorator(func) 是等價(jià)的,所以上面 func.__name__ 是等價(jià)于下面decorator(func).__name__ 的,那當(dāng)然名字是 inner_function
def wrapper(func): def inner_function(): pass return inner_function def wrapped(): pass print(wrapper(wrapped).__name__) #inner_function
目前,我們可以看到當(dāng)一個(gè)函數(shù)被裝飾器裝飾過后,它的簽名信息會(huì)發(fā)生變化(譬如上面看到的函數(shù)名)
那如何避免這種情況的產(chǎn)生?
解決方案就是使用我們前面所說的 functools .wraps 裝飾器。
它的作用就是將 被修飾的函數(shù)(wrapped) 的一些屬性值賦值給 修飾器函數(shù)(wrapper) ,最終讓屬性的顯示更符合我們的直覺。
from functools import wraps def wrapper(func): @wraps(func) def inner_function(): pass return inner_function @wrapper def wrapped(): pass print(wrapped.__name__) # wrapped
那么問題就來了,我們使用了 decorator 之后,是否還會(huì)存在這種簽名的問題呢?
寫個(gè)例子來驗(yàn)證一下就知道啦
from decorator import decorator @decorator def deco(func, *args, **kw): print("Ready to run task") func(*args, **kw) print("Successful to run task") @deco def myfunc(): print("Running the task") print(myfunc.__name__)
輸出的結(jié)果是 myfunc,說明 decorator 已經(jīng)默認(rèn)幫我們處理了一切可預(yù)見的問題。
到此,相信大家對“如何使用裝飾器”有了更深的了解,不妨來實(shí)際操作一番吧!這里是創(chuàng)新互聯(lián)網(wǎng)站,更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢,關(guān)注我們,繼續(xù)學(xué)習(xí)!