裝飾器的作用就是用一個新函數封裝舊函數(是舊函數代碼不變的情況下增加功能)然后會返回一個新函數,新函數就叫做裝飾器,一般為了簡化裝飾器會用語法糖@新函數來簡化
例子:
這是一段代碼,但功能太少,要對這個進行增強,但又不能改變代碼。
1 def hello(): 2 return "hello world!"
現在我們的需求是要增強hello()函數的功能,希望給返回加上HTML標簽,比如<i>hello world</i>,但要求我們不得改變hello()函數原來的定義。
1 def makeitalic(fun):#makitalic傳了一個新函數 2 def wrapped():#內部函數 3 return "<i>"+fun()+"</i>"#要加的新功能 4 return wrapped#返回的是wrapped函數功能 5 6 def hello():#對這個功能進行增強 7 return "hello world!" 8 #makeitalic里面傳入了hello函數,然后內部函數fun()函數也就相當於hello函數了 9 hello_2=makeitalic(hello) 10 #打印新函數,返回的就是<i>hello world!</i> 11 print(hello_2())
為了增強原函數hello的功能,定義了一個函數,它接收原函數作為參數,並返回一個新的函數,在這個返回的函數中,執行了原函數,並對原函數的功能進行了增強。
事實上,makeitalic就是一個裝飾器(decorator),它封裝了原函數hello,並返回了一個新函數,用於增強原函數的功能,並將其賦值給hello。
一般情況下,我們使用裝飾器提供的@語法糖(Syntactic Sugar),來簡化上面的操作。
####使用@語法糖 def makeitalic(fun): def wrapped(): return "<i>" + fun() + "</i>" return wrapped @makeitalic#使用了裝飾器可以直接調用,不需要賦值了 def hello(): return "hello world" print(hello())#使用了裝飾器可以直接調用,不需要賦值了
像上面的情況,可以動態的修改函數(或類的)功能的函數就是裝飾器。本質上,它是一個高階函數,以被裝飾的函數(比如上面的hello)為參數,並返回一個包裝后的函數(比如上面的wrapped)給被修飾函數(hello)。
當調用hello()函數時,hello函數的執行流程如下分析:
1.把hello函數作為參數傳給@符號后面的裝飾器函數。
2.然后開始執行裝飾器函數,並返回一個包裝了的函數,同時,改變原函數的指向,現在原函數指向了這個包裝函數。
3.執行原函數,其實此時執行的是包裝了的函數,所以說,裝飾器增強了一個現有函數的功能,但不會改變現有函數的定義。
普通裝飾器的使用形式:
1 @decorator 2 def fun(): 3 pass 4 5 #格式就如同下面的: 6 7 def fun(): 8 pass 9 fun = decorator(fun)#不使用語法糖要進行賦值
裝飾器可以定義多個,離函數定義最近的裝飾器最先被調用,比如:
1 @decotator_one 2 @decorator_two 3 def fun(): 4 pass 5 6 #格式如同下面的: 7 8 def fun(): 9 pass 10 fun = decorator_one(decorator_two(fun))
裝飾器還可以帶參數,比如:
1 @decorator(arg1, arg2) 2 def fun(): 3 pass 4 5 #格式如同下面的: 6 7 def fun(): 8 pass 9 fun = decorator(arg1, arg2)(fun)