python 裝飾器

參考網址：https://www.zhihu.com/question/26930016/answer/99243411

裝飾器本質上是Python函數，作用是使其他函數在不需要做任何代碼變動的前提下增加額外功能，裝飾器的返回值也是一個函數對象

• 簡單裝飾器：

def use_logging(func):
    def wrapper(*args, **kwargs):
        logging.warn("%s is running" % func.__name__)
        return func(*args, **kwargs)
    return wrapper

def bar():
    print('i am bar')

bar = use_logging(bar)
bar()

@符號是裝飾器的語法糖，在定義函數的時候使用，避免再一次賦值操作

def use_logging(func):
    def wrapper(*args, **kwargs):
        logging.warn("%s is running" % func.__name__)
        return func(*args, **kwargs)
    return wrapper

@use_logging
def bar():
    print("i am bar")

bar()

由此可見，@等效於bar = use_logging(bar)，這裏等號左邊的bar應該是指use_logging中定義的wrapper函數

舉個例子：

def funA(f):
    print(f())

@funA
def funB():
    print('B')
    return 'i am b'

直接運行，output:

B
i am b

由前可知，funB = funA(funB), 先執行等式右邊，funB送入funA,執行funB，輸出'B'，再打印funB的返回值‘i am b’,注意，這是直接運行的結果，若此時最後像上面那樣調用funB()，會報錯，因爲funA沒有返回函數

• 帶參數的裝飾器

def use_logging(level):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if level == "warn":
                logging.warn("%s is running" % func.__name__)
            return func(*args)
        return wrapper
    return decorator

@use_logging(level="warn")
def foo(name='foo'):
    print("i am %s" % name)

foo()

上面的use_logging是允許帶參數的裝飾器。它實際上是對原有裝飾器的一個函數封裝，並返回一個裝飾器。

我的理解是，這裏是 foo = decorator(foo)