快捷導(dǎo)航

如何正確理解python裝飾器

更新時間：2021年06月15日 15:11:14 作者：luyuze95

裝飾器(Decorators)是 Python 的一個重要部分。簡單地說：他們是修改其他函數(shù)的功能的函數(shù)。他們有助于讓我們的代碼更簡短

一、閉包

要想了解裝飾器，首先要了解一個概念，閉包。什么是閉包，一句話說就是，在函數(shù)中再嵌套一個函數(shù)，并且引用外部函數(shù)的變量，這就是一個閉包了。光說沒有概念，直接上一個例子。

def outer(x):
    def inner(y):
        return x + y
    return inner

print(outer(6)(5))
-----------------------------
>>>11

如代碼所示，在outer函數(shù)內(nèi)，又定義了一個inner函數(shù)，并且inner函數(shù)又引用了外部函數(shù)outer的變量x，這就是一個閉包了。在輸出時，outer(6)(5),第一個括號傳進去的值返回inner函數(shù)，其實就是返回6 + y，所以再傳第二個參數(shù)進去，就可以得到返回值，6 + 5。

二、裝飾器

接下來就講裝飾器，其實裝飾器就是一個閉包，裝飾器是閉包的一種應(yīng)用。什么是裝飾器呢，簡言之，python裝飾器就是用于拓展原來函數(shù)功能的一種函數(shù)，這個函數(shù)的特殊之處在于它的返回值也是一個函數(shù)，使用python裝飾器的好處就是在不用更改原函數(shù)的代碼前提下給函數(shù)增加新的功能。使用時，再需要的函數(shù)前加上@demo即可。

def debug(func):
    def wrapper():
        print("[DEBUG]: enter {}()".format(func.__name__))
        return func()
    return wrapper

@debug
def hello():
    print("hello")

hello()
-----------------------------
>>>[DEBUG]: enter hello()
>>>hello

例子中的裝飾器給函數(shù)加上一個進入函數(shù)的debug模式，不用修改原函數(shù)代碼就完成了這個功能，可以說是很方便了。

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

上面例子中的裝飾器是不是功能太簡單了，那么裝飾器可以加一些參數(shù)嗎，當(dāng)然是可以的，另外裝飾的函數(shù)當(dāng)然也是可以傳參數(shù)的。

def logging(level):
    def outwrapper(func):
        def wrapper(*args, **kwargs):
            print("[{0}]: enter {1}()".format(level, func.__name__))
            return func(*args, **kwargs)
        return wrapper
    return outwrapper

@logging(level="INFO")
def hello(a, b, c):
    print(a, b, c)

hello("hello,","good","morning")
-----------------------------
>>>[INFO]: enter hello()
>>>hello, good morning

如上，裝飾器中可以傳入?yún)?shù)，先形成一個完整的裝飾器，然后再來裝飾函數(shù)，當(dāng)然函數(shù)如果需要傳入?yún)?shù)也是可以的，用不定長參數(shù)符號就可以接收，例子中傳入了三個參數(shù)。

四、類裝飾器

裝飾器也不一定只能用函數(shù)來寫，也可以使用類裝飾器，用法與函數(shù)裝飾器并沒有太大區(qū)別，實質(zhì)是使用了類方法中的__call__魔法方法來實現(xiàn)類的直接調(diào)用。

class logging(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print("[DEBUG]: enter {}()".format(self.func.__name__))
        return self.func(*args, **kwargs)

@logging
def hello(a, b, c):
    print(a, b, c)

hello("hello,","good","morning")
-----------------------------
>>>[DEBUG]: enter hello()
>>>hello, good morning

類裝飾器也是可以帶參數(shù)的，如下實現(xiàn)

class logging(object):
    def __init__(self, level):
        self.level = level

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            print("[{0}]: enter {1}()".format(self.level, func.__name__))
            return func(*args, **kwargs)
        return wrapper
        
@logging(level="TEST")
def hello(a, b, c):
    print(a, b, c)

hello("hello,","good","morning")
-----------------------------
>>>[TEST]: enter hello()
>>>hello, good morning

好了，如上就是裝飾器的一些概念和大致的用法啦，想更深入的了解裝飾器還是需要自己在平時的練習(xí)和應(yīng)用中多體會

以上就是如何正確理解python裝飾器的詳細內(nèi)容，更多關(guān)于python裝飾器的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: