python中的裝飾器原理和作用

裝飾器的作用就是用一個新函數封裝舊函數(是舊函數代碼不變的情況下增加功能)然後會返回一個新函數,新函數就叫做裝飾器,一般爲了簡化裝飾器會用語法糖@新函數來簡化

例子:

這是一段代碼,但功能太少,要對這個進行增強,但又不能改變代碼。

def hello():
    return "hello world!"

現在我們的需求是要增強hello()函數的功能,希望給返回加上HTML標籤,比如hello world,但要求我們不得改變hello()函數原來的定義。

def makeitalic(fun):#makitalic傳了一個新函數
    def wrapped():#內部函數
        return "<i>"+fun()+"</i>"#要加的新功能
    return wrapped#返回的是wrapped函數功能

def hello():#對這個功能進行增強
    return "hello world!"
#makeitalic裏面傳入了hello函數,然後內部函數fun()函數也就相當於hello函數了
hello_2=makeitalic(hello)
#打印新函數,返回的就是<i>hello world!</i>
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.執行原函數,其實此時執行的是包裝了的函數,所以說,裝飾器增強了一個現有函數的功能,但不會改變現有函數的定義。

普通裝飾器的使用形式:

@decorator
def fun():
 pass

#格式就如同下面的:
#Python小白交流學習羣:711312441
def fun():
 pass
fun = decorator(fun)#不使用語法糖要進行賦值

裝飾器可以定義多個,離函數定義最近的裝飾器最先被調用,比如:

@decotator_one
@decorator_two
def fun():
 pass

#格式如同下面的:

def fun():
 pass
fun = decorator_one(decorator_two(fun))

裝飾器還可以帶參數,比如:

@decorator(arg1, arg2)
def fun():
 pass

#格式如同下面的:

def fun():
 pass
fun = decorator(arg1, arg2)(fun)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章