python3（十八）decorator

# -----------------------1--------------------------------------------
# 由於函數也是一個對象，而且函數對象可以被賦值給變量，所以，通過變量也能調用該函數。
def now():
    print('2019-3-25')


f = now
print(f())  # 2019-3-25
print(now.__name__)  # now
print(f.__name__)  # now


# ----------------------2---------------------------------------------
# 增強now()函數的功能，比如，在函數調用前后自動打印日志，但又不希望修改now()函數的定義，
# 這種在代碼運行期間動態增加功能的方式，稱之為“裝飾器”（Decorator）。
def log(func):
    def wrapper(*args, **kw):
        print('call %s():' % func.__name__)
        return func(*args, **kw)

    return wrapper


# 上面的log，因為它是一個decorator，所以接受一個函數作為參數，並返回一個函數。
# 我們要借助Python的@語法，把decorator置於函數的定義處
@log
def now():
    print('2019-3-26')


now()  # call now():  2019-3-26


# print(now.__name__) # wrapper

# 把@log放到now()函數的定義處,相當於 now = log(now)
# 由於log()是一個decorator，返回一個函數，所以，原來的now()函數仍然存在，只是現在同名的now變量指向了新的函數，於是調用now()將執行新函數，即在log()函數中返回的wrapper()函數。
# wrapper()函數的參數定義是(*args, **kw)，因此，wrapper()函數可以接受任意參數的調用。在wrapper()函數內，首先打印日志，再緊接着調用原始函數。

# ----------------3----------------------------------------------------
# 自定義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


# 如下@ 相當於now = log('execute')(now)
@log('execute')
def now():
    print('2015-3-27')


now()  # execute now(): 2015-3-27
# 上面的語句，首先執行log('execute')，返回的是decorator函數，
# 再調用返回的函數，參數是now函數，返回值最終是wrapper函數。

# ----------------------4----------------------------------------------
print(now.__name__)  # wrapper
# 因為返回的那個wrapper()函數名字就是'wrapper'，所以，需要把原始函數的__name__等屬性復制到wrapper()函數中，
# 否則，有些依賴函數簽名的代碼執行就會出錯。
# 不需要編寫wrapper.__name__ = func.__name__這樣的代碼，Python內置的functools.wraps就是干這個事的
# 修改log
import functools


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

    return wrapper


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

@log
def now():
    print('2019-3-28')


print(now.__name__)  # now