Python---裝飾器

# 裝飾器
# 由於函數也是一個對象，而且函數對象可以被賦值給變量，所以，通過變量也能調用該函數


def now():
    print('2017-08-07')

f = now
f()

# 函數對象有一個__name__屬性，可以拿到函數的名字

print(now.__name__)
print(f.__name__)

# 假設我們要增強now()函數的功能
# 比如，在函數調用前後自動打印日誌，但又不希望修改now()函數的定義，這種代碼運行期間動態增加功能的方式，稱之爲"裝飾器"（Decorator）

# 本質上，decorator就是一個返回函數的高階函數
# 定義一個能打印日誌的decorator


def log(func):
    def wrapper(*args, **kw):
        print('call %s():' % func.__name__)
        return func(*args, **kw)
    return wrapper

# decorator接受一個函數作爲參數，並返回一個函數
# 我們要藉助Python的語法，把decorator置於函數的定義處


@log
def now():
    print('2017-08-07')

now()

# 把@log放到now()函數的定義處，相當於執行了語句 now = log(now)
# 由於log()是一個decorator，返回一個函數，所以，原來的now()函數依然存在，只是現在同名的now變量指向了新的函數
# 於是調用now()將執行新函數，即在log()函數中返回的wrapper()函數

# wrapper()函數的參數定義是(*args, **kw)，因此，wrapper()函數可以接受任意參數的調用
# 在wrapper()函數內，首先打印日誌，再緊接着調用原始函數

# 如果decorator本身需要傳入參數，那就需要編寫一個返回decorator的高階函數，寫出來會更復雜
# 自定義log的文本


def log(text):
    def decorator(func):
        def wrapper(*args, **kw):
            print('%s %s():' % (text, func.__name__))
            return func(*args, **kw)
        return wrapper
    return decorator


@log('execute')
def now2():
    print('2017-08-07')

now2()

# 和兩層嵌套的decorator相比，3層嵌套的效果是  now = log('execute')(now)
# 上面的語句，首先執行log('execute')，返回的是decorator函數，再調用返回的函數，參數是now函數，返回值最終是wrapper函數