合理使用裝飾器可以簡(jiǎn)化開(kāi)發(fā)，并且使得代碼更加清晰。下面我們分別介紹兩種裝飾器，不帶參數(shù)的裝飾器和帶參數(shù)的裝飾器。

一、不帶參數(shù)的裝飾器

我們用一個(gè)實(shí)際的例子來(lái)引入裝飾器的概念，比如我們現(xiàn)在有一個(gè)方法A()，然后我們需要在方法A()執(zhí)行之前在終端打印"function is running"，這時(shí)候我們可以在方法A()的開(kāi)始部分加上下面的代碼：

print("function is running")

但是如果我們不想修改方法A()的代碼，也可以重新寫(xiě)一個(gè)方法decorator()，然后把A()作為入?yún)?，代碼如下：

def decorator(fun):
    @functools.wraps(fun)
    def wrapper(*args, **kwargs):
        print("Method is running")
        return fun()
    return wrapper

python中的方法可以作為參數(shù)傳入另一個(gè)方法，所以當(dāng)我們執(zhí)行decorator(A)的時(shí)候，返回的是一個(gè)方法，這個(gè)方法的本質(zhì)是先執(zhí)行print語(yǔ)句，然后再執(zhí)行A()。

但是這樣我們的調(diào)用代碼就需要進(jìn)行修改，也就是說(shuō)之前調(diào)用方法A()，現(xiàn)在的寫(xiě)法是調(diào)用decorator(A)()，為了不影響調(diào)用的代碼，我們使用python的語(yǔ)法糖改造方法A()的代碼。

@decorator
def A():
    # do something

實(shí)際上只是在A()的前面加上一行@decorator，這樣在我們的調(diào)用代碼就不需要改變了。

裝飾器中的@functools.wraps(fun)是為了保證返回的方法__name__屬性和入?yún)⒎椒╢un的__name__屬性相同。

二、帶入?yún)⒌难b飾器

在上面不帶參數(shù)的裝飾器中，我們實(shí)現(xiàn)了在執(zhí)行方法A()之前打印一行固定的字符串，如果我們想要打印不同的字符串就需要用帶參數(shù)的裝飾器。實(shí)際上就是在不帶參數(shù)的裝飾器上再封裝一層即可，代碼如下：

def decorator_par(name):
    def decorator(fun):
        @functools.wraps(fun)
        def wrapper(*args, **kwargs):
            print("Method {} is running".format(name))
            return fun()
        return wrapper
    return decorator

這樣我們就可以在使用裝飾器的時(shí)候設(shè)置參數(shù)name，實(shí)現(xiàn)打印不同的信息。完整的代碼如下：

import functools
 
def decorator_par(name):
    def decorator(fun):
        @functools.wraps(fun)
        def wrapper(*args, **kwargs):
            print("Method {} is running".format(name))
            return fun()
        return wrapper
    return decorator
 
@decorator_par("A")
def A():
    print("I am A")
 
@decorator_par("B")
def B():
    print("I am B")
 
A()
B()

執(zhí)行輸出為：

Method A is running
I am A
Method B is running
I am B

可以看到我們?cè)诓桓淖兎椒ㄕ{(diào)用代碼的情況下，實(shí)現(xiàn)了在該方法之前打印特定的信息。

以上就是裝飾器的兩種實(shí)現(xiàn)示例，實(shí)際開(kāi)發(fā)中也可以將多個(gè)裝飾器進(jìn)行嵌套，可以實(shí)現(xiàn)更加復(fù)雜的需求。

內(nèi)容擴(kuò)展：

python函數(shù)式編程之裝飾器

1.開(kāi)放封閉原則

簡(jiǎn)單來(lái)說(shuō)，就是對(duì)擴(kuò)展開(kāi)放，對(duì)修改封閉。

在面向?qū)ο蟮木幊谭绞街?，?jīng)常會(huì)定義各種函數(shù)。一個(gè)函數(shù)的使用分為定義階段和使用階段，一個(gè)函數(shù)定義完成以后，可能會(huì)在很多位置被調(diào)用。這意味著如果函數(shù)的定義階段代碼被修改，受到影響的地方就會(huì)有很多，此時(shí)很容易因?yàn)橐粋€(gè)小地方的修改而影響整套系統(tǒng)的崩潰，所以對(duì)于現(xiàn)代程序開(kāi)發(fā)行業(yè)來(lái)說(shuō)，一套系統(tǒng)一旦上線，系統(tǒng)的源代碼就一定不能夠再改動(dòng)了。然而一套系統(tǒng)上線以后，隨著用戶數(shù)量的不斷增加，一定會(huì)為一套系統(tǒng)擴(kuò)展添加新的功能。

此時(shí)，又不能修改原有系統(tǒng)的源代碼，又要為原有系統(tǒng)開(kāi)發(fā)增加新功能，這就是程序開(kāi)發(fā)行業(yè)的開(kāi)放封閉原則，這時(shí)就要用到裝飾器了。

2.什么是裝飾器

裝飾器，顧名思義，就是裝飾，修飾別的對(duì)象的一種工具。

所以裝飾器可以是任意可調(diào)用的對(duì)象，被裝飾的對(duì)象也可以是任意可調(diào)用對(duì)象。

3.裝飾器的作用

在不修改被裝飾對(duì)象的源代碼以及調(diào)用方式的前提下為被裝飾對(duì)象添加新功能。

以上就是如何真正的了解python裝飾器的詳細(xì)內(nèi)容，更多關(guān)于你真的了解python裝飾器么的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論