1. 什么是裝飾器
知乎大佬如是說(shuō):
內(nèi)褲可以用來(lái)遮羞,但是到了冬天它沒(méi)法為我們防風(fēng)御寒,聰明的人們發(fā)明了長(zhǎng)褲堤瘤,有了長(zhǎng)褲后寶寶再也不冷了,裝飾器就像我們這里說(shuō)的長(zhǎng)褲浆熔,在不影響內(nèi)褲作用的前提下本辐,給我們的身子提供了保暖的功效。
裝飾器本質(zhì)上是Python函數(shù)医增,可以為已存在的對(duì)象添加額外的功能慎皱,同時(shí)裝飾器還可以抽離出與函數(shù)無(wú)關(guān)的重用代碼。具體應(yīng)用場(chǎng)景如:插入日志叶骨、性能測(cè)試茫多、事務(wù)處理、緩存忽刽、權(quán)限校驗(yàn)等天揖。
換言之
裝飾器不能影響原函數(shù)的功能,裝飾器是獨(dú)立出來(lái)的函數(shù)跪帝。誰(shuí)調(diào)用它今膊,誰(shuí)就可以使用它的功能。
2.舉個(gè)栗子
add的功能是計(jì)算x和y的值伞剑,我們稱(chēng)作功能函數(shù)斑唬。
logger的作業(yè)是在執(zhí)行add函數(shù)的同時(shí)再打印了其他的信息,這部分的作為add的功能增強(qiáng),我們稱(chēng)為裝飾恕刘。
在logger里我們可以加入其他類(lèi)似的功能函數(shù)缤谎,也能包裝它,可以進(jìn)行復(fù)用雪营。
1.引子
#功能函數(shù)
def add(x,y):
return x+y
#裝飾函數(shù)
def logger(fn):
print('frist')
x = fn(4,5)
print('second')
return x
print(logger(add))
#把函數(shù)add傳給logger 弓千,return x+y
#print('frist')
#print('secend')
# x = fn(4,5) ==> x = 4 y= 5 x= 4+5 = 9
#return 9
frist
second
9
2.提取參數(shù)
x,y的參數(shù)都放在logger函數(shù)內(nèi)部了,影響函數(shù)的靈活性献起,此處我們可以提取出來(lái)。
def add(x,y):
return x + y
def logger(fn,*args,**kwargs):
print('frist')
x = fn(*args,**kwargs)
print('second')
return x
print(logger(add,1,y=11))
frist
second
12
3.柯里化
def add(x,y):
return x + y
def logger(fn):
def wrapper(*args,**kwargs):
print('begin')
x = fn(*args,**kwargs)
print('end')
return x
return wrapper
print(logger(add)(5,y=11))
begin
end
16
懵逼ing
以下為個(gè)人理解镣陕,左邊為非柯里化函數(shù)谴餐,右邊是柯里化函數(shù)。
<img src="http://qiniu.mykernel.cn/klh.png" alt="柯里化函數(shù)"/>
前面說(shuō)過(guò)柯里化的定義呆抑,本來(lái)可以一次傳入兩個(gè)參數(shù)岂嗓,柯里化之后。只需要傳入一個(gè)函數(shù)了鹊碍。厌殉。
左邊傳入add 和 兩個(gè)參數(shù)。
右邊的logger(add)是一個(gè)函數(shù)侈咕,只需要傳入兩個(gè)參數(shù)公罕。logger(add)是個(gè)整體,結(jié)合成一個(gè)函數(shù)耀销。當(dāng)然這樣寫(xiě)楼眷,我們看函數(shù)主題的部分也是不一樣的。
函數(shù)的基礎(chǔ)中說(shuō)過(guò)熊尉,函數(shù)的傳參必須和函數(shù)參數(shù)的定義一致罐柳。重點(diǎn)分析右邊函數(shù)(柯里化)。
參數(shù)部分:參數(shù)傳入的方式狰住,logger函數(shù)需要傳入個(gè)fn张吉,fu的返回值是wrapper函數(shù),wrapper函數(shù)的參數(shù)是(*args,**kwargs)所以此次就需要分兩次傳入?yún)?shù)催植。
第一次傳入fn肮蛹,再次傳入wrapper函數(shù)需要的參數(shù)。所以就出現(xiàn)了最下邊的調(diào)用方式查邢。
print(logger(add)(5,y=50))蔗崎。
返回值部分:右側(cè)的logger函數(shù)是個(gè)嵌套函數(shù),logger的返回值是wrapper扰藕,內(nèi)層的wrapper函數(shù)返回值是x缓苛,x = fn(*args,**kwargs)。fn函數(shù)是最后調(diào)用時(shí)候傳入的add函數(shù)。
懵逼 X 2未桥。笔刹。。冬耿。
def add(x,y):
return x + y
def logger(fn,*args,**kwargs): def logger(fn): #參數(shù)剝離
def newfunction(*args,**kwargs): #新定義一個(gè)函數(shù)舌菜,logger函數(shù)返回也是這個(gè)函數(shù)名字
print('frist') print('frist')
x = fn(*args,**kwargs) == > x = fn(*args,**kwargs)
print('second') print('second')
return x return x
return newfunction
print(logger(add,1,y=11)) print(logger(add)(5,y=11)) #兩次傳入?yún)?shù)
效果如下:
def add(x,y):
return x + y
def logger(fn): #參數(shù)剝離
def newfunction(*args,**kwargs): #新定義一個(gè)函數(shù),logger函數(shù)返回也是這個(gè)函數(shù)名字
print('frist')
x = fn(*args,**kwargs)
print('second')
return x
return newfunction
print(logger(add)(5,y=11)) #兩次傳入?yún)?shù)
frist
second
16
繼續(xù)懵逼的話就這樣用吧亦镶。日月。。用多了就悟道了缤骨。爱咬。
4.裝飾器語(yǔ)法糖
#再次變形。绊起。精拟。
def add(x,y):
return x + y
def logger(fn):
def wrapper(*args,**kwargs):
print('begin')
x = fn(*args,**kwargs)
print('end')
return x
return wrapper
##調(diào)用方法1:
print(logger(add)(x=1111,y=1))
##調(diào)用方法2:
add = logger(add)
print(add(x=11,y=3))
##調(diào)用方法3: python給我們的語(yǔ)法糖
@logger # 說(shuō)明下邊的函數(shù),add 其實(shí)是 add = logger(add)
def add(x,y):
return x + y
print(add(45,40))
begin
end
1112
begin
end
14
begin
end
85
3.復(fù)雜的栗子
import datetime
import time
def logger(fn):
def warp(*arges,**kwarges):
print("arges={},kwarges={}".format(arges,kwarges)) #打印函數(shù)的兩個(gè)參數(shù)
start = datetime.datetime.now() #獲取函數(shù)運(yùn)行的開(kāi)始時(shí)間
ret = fn(*arges,**kwarges) #傳入兩個(gè)參數(shù)虱歪,調(diào)用add函數(shù) 此處有個(gè)return的值蜂绎,需要一層一層的返回出去
duratime = datetime.datetime.now() - start #獲得函數(shù)的運(yùn)行時(shí)間
print("function {} took {}s".format(fn.__name__,duratime.total_seconds())) #打印函數(shù)的運(yùn)行時(shí)間
return ret #返回fn的結(jié)果 ,fn = x+y ==> 返回x+y的值笋鄙。 x = 4 y= 11 ==> return 11
return warp #返回warp的 return ==> ret 的return ==> return 11 函數(shù)的最終結(jié)果為11
@logger
def add(x,y):
print("oooooook")
time.sleep(1.5)
return x+y
print(add(4,y=11))
#如果充分理解了每個(gè)小部件师枣,這個(gè)簡(jiǎn)單的完整版本也是很好理解的了。
#1局装,logger是個(gè)裝飾器坛吁,而且使用了柯里化技術(shù)
#2,add 傳參給logger的fn 形參铐尚,add(4,y=5)的兩個(gè)參數(shù)傳入給warp函數(shù)的兩個(gè)形參
#
#
arges=(4,),kwarges={'y': 11}
oooooook
function add took 1.5017s
15
再次翻譯
import datetime
import time
#####################################裝飾開(kāi)始############################################
def logger(fn): #拿到函數(shù)名稱(chēng)
def warp(*arges,**kwarges): #拿到函數(shù)帶過(guò)來(lái)的參數(shù)開(kāi)始裝飾
print("arges={},kwarges={}".format(arges,kwarges)) #來(lái)試試打印兩個(gè)參數(shù)
start = datetime.datetime.now() #
ret = fn(*arges,**kwarges) # 此處調(diào)用add函數(shù)拨脉。開(kāi)始執(zhí)行函數(shù),發(fā)現(xiàn)return語(yǔ)句宣增。玫膀。ret的結(jié)果就是return。
duratime = datetime.datetime.now() - start #
print("function {} took {}s".format(fn.__name__,duratime.total_seconds()))
return ret #加工完成開(kāi)始返回爹脾。warp的返回值是ret 帖旨,ret的返回值是 add函數(shù)的執(zhí)行結(jié)果(原函數(shù)的功能完整的保留了)
return warp # logger的返回結(jié)果是warp,warp的返回值是ret 灵妨,ret的返回值是 add函數(shù)的執(zhí)行結(jié)果(原函數(shù)的功能完整的保留了)
#####################################裝飾完成############################################
@logger #裝飾工廠
######add是需要被裝飾的函數(shù)解阅,當(dāng)你有這個(gè)想法的事情,其實(shí)事情已經(jīng)開(kāi)始發(fā)生了泌霍。
def add(x,y): # 此時(shí)add = logger(add) 此處前面的@logger標(biāo)記就是想要讓logger裝飾器像一個(gè)工廠一樣對(duì)add函數(shù)進(jìn)行加工货抄。
print("oooooook")
time.sleep(1.5)
return x+y
print(add(4,y=11))
arges=(4,),kwarges={'y': 11}
oooooook
function add took 1.501604s
15
4.帶參裝飾器
1. 文檔字符串
我們約定,在python函數(shù)的第一行需要對(duì)函數(shù)進(jìn)行說(shuō)明,使用三引號(hào)表示蟹地。
如果是英文說(shuō)明积暖,慣例首字母大寫(xiě),第一行寫(xiě)概述怪与,空一行夺刑,第三行寫(xiě)詳細(xì)描述。
如果函數(shù)中有文檔字符串分别,默認(rèn)會(huì)放在函數(shù)的doc屬性中遍愿,可以直接訪問(wèn)。
def add(x,y):
"""This is a function of addition"""
a = x+y
return x + y
print("function name is {}\nfunction doc = {}\n\n".format(add.__name__, add.__doc__))
print(help(add))
function name is add
function doc = This is a function of addition
Help on function add in module __main__:
add(x, y)
This is a function of addition
None
2. 前面裝飾器的副作用
前面裝飾器基本上已經(jīng)可以完成對(duì)函數(shù)進(jìn)行加強(qiáng)的功能了茎杂,但是還有些瑕疵错览。比如原來(lái)函數(shù)的原屬性已經(jīng)被替換為裝飾器的屬性了。如下:
def add(x,y):
return x + y
def logger(fn):
"This is logger doc"
def wrapper(*args,**kwargs):
"This is wrapper doc"
print('begin')
x = fn(*args,**kwargs)
print('end')
return x
return wrapper
@logger # add = logger(add)
def add(x,y):
"This is add doc "
print("name = {}\ndoc = {}".format(add.__name__,add.__doc__))
return x + y
print(add(45,40))
#可以看出來(lái)add被裝飾出來(lái)的函數(shù)(新的add)的屬性已經(jīng)全部改變了煌往。
begin
name = wrapper
doc = This is wrapper doc
end
85
3. 解決方案一
三個(gè)函數(shù):
- copy原函數(shù)的屬性 copy_properties
- 裝飾器 logger
- 功能函數(shù) add
def copy_properties(src, dst): # 把src的相關(guān)屬性賦值給dst (fn,wrap)
dst.__name__ = src.__name__
dst.__doc__ = src.__doc__
def logger(fn):
"""'This is a function of logger'"""
def wrap(*arges,**kwarges): #
"""'This is a function of wrap'"""
print('<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>')
x = fn(*arges,**kwarges)
#print("name={}\ndoc={}".format(add.__name__,add.__doc__))
print('<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>')
return x
copy_properties(fn,wrap) #思考1:為什么放在這個(gè)位置調(diào)用
return wrap
@logger
def add(x,y):
"""'This is a function of add'"""
print("name={}\ndoc={}".format(add.__name__,add.__doc__))
return x+y
print(add(4,6))
<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>
name=add
doc='This is a function of add'
<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>-<>
10
4. 解決方案二
但凡使用裝飾器都會(huì)出現(xiàn)屬性的這個(gè)問(wèn)題,為什么不把copy_properties也做成裝飾器呢轧邪?
三個(gè)函數(shù):
- copy原函數(shù)的裝飾器 copy_properties1
- 裝飾器 logger
- 功能函數(shù) add
def copy_properties(src, dst): # 把src的相關(guān)屬性賦值給dst (fn,wrap)
dst.__name__ = src.__name__
dst.__doc__ = src.__doc__
#利用前面的知識(shí)我們可以對(duì)copy_properties輕松進(jìn)行變形
def copy_properties1(src): # 把src的相關(guān)屬性賦值給dst (fn,wrap)
def _copy(dst):
dst.__name__ = src.__name__
dst.__doc__ = src.__doc__
return dst
return _copy
帶參裝飾器:
def logger(fn):
"""'This is a function of logger'"""
@copy_properties1(fn) #wrap = copy_properties(fn)(wrap)
#== > 柯里化 兩次傳入?yún)?shù) src = fn 刽脖, dst = wrap 新的wrap函數(shù)的屬性已經(jīng)替換為原函數(shù)的。
def wrap(*arges,**kwarges): #wrap = copy_properties(fn)(wrap)(*arges,**kwarges)
"""'This is a function of wrap'"""
print('>->->->->->->->->->->->->->->->->->->->->->->->->->')
x = fn(*arges,**kwarges)
print('<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<')
return x
return wrap
@logger #add =logger(add)
def add(x,y):
"""'This is a function of add'"""
print("name={}\ndoc={}".format(add.__name__,add.__doc__))
return x+y
print(add(4,11))
>->->->->->->->->->->->->->->->->->->->->->->->->->
name=add
doc='This is a function of add'
<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<-<
15
更多歡迎訪問(wèn):http://www.mykernel.cn/