python裝飾器詳解

目錄

• python 裝飾器詳解
  • 一、閉包
  • 二、裝飾器
  • 三、帶參數的裝飾器
  • 四、類裝飾器

python 裝飾器詳解

一、閉包

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

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

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

如代碼所示，在outer函數內，又定義了一個inner函數，而且inner函數又引用了外部函數outer的變量x，這就是一個閉包了。在輸出時，outer(6)(5),第一個括號傳進去的值返回inner函數，其實就是返回6 + y，因此再傳第二個參數進去，就能夠獲得返回值，6 + 5。

二、裝飾器

接下來就講裝飾器，其實裝飾器就是一個閉包，裝飾器是閉包的一種應用。什麼是裝飾器呢，簡言之，python裝飾器就是用於拓展原來函數功能的一種函數，這個函數的特殊之處在於它的返回值也是一個函數，使用python裝飾器的好處就是在不用更改原函數的代碼前提下給函數增長新的功能。使用時，再須要的函數前加上@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

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

三、帶參數的裝飾器

上面例子中的裝飾器是否是功能太簡單了，那麼裝飾器能夠加一些參數嗎，固然是能夠的，另外裝飾的函數固然也是能夠傳參數的。

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

如上，裝飾器中能夠傳入參數，先造成一個完整的裝飾器，而後再來裝飾函數，固然函數若是須要傳入參數也是能夠的，用不定長參數符號就能夠接收，例子中傳入了三個參數。

四、類裝飾器

裝飾器也不必定只能用函數來寫，也可使用類裝飾器，用法與函數裝飾器並無太大區別，實質是使用了類方法中的__call__魔法方法來實現類的直接調用。

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

類裝飾器也是能夠帶參數的，以下實現

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

好了，如上就是裝飾器的一些概念和大體的用法啦，想更深刻的瞭解裝飾器仍是須要本身在平時的練習和應用中多體會，本篇只是給出一個概念，謝謝~