為什么要使用?

使用裝飾器使代碼變得整潔，并且能解決硬編碼問題，使用起來也很方便，但是理解起來相對沒那么容易，因此寫下這篇文章加深理解。下面以記錄日志的功能作為裝飾器的講解：

原始硬編碼版

下面的logit作為裝飾器，但是它只能裝飾有兩個參數(shù)的函數(shù)，因此是硬編碼的。

def logit(func):
    def decorator(a, b):
        print("func %s is calling " % func.__name__)
        return func(a, b)
    return decorator

@logit()
def my_add(a, b):
    return a+b
  
my_add(1, 49)

可變參數(shù)版

使用*args, **kwargs將上述的硬編碼去除，使得裝飾器能裝飾不同的函數(shù)

def logit(func):
    def decorator(*args, **kwargs):
        print("func %s is calling " % func.__name__)
        return func(*args)
    return decorator

wraps通用版

輸出my_add.name發(fā)現(xiàn)函數(shù)的名稱改變了，不是我們想要的my_add，因此使用wraps改造

print(my_add.__name__)

from functools import wraps
def logit(func):
    @wraps(func)
    def decorator(*args, **kwargs):
        print("func %s is calling " % func.__name__)
        return func(*args)
    return decorator

自定義參數(shù)版

有時候需要向裝飾器來傳遞參數(shù)，例如不同的函數(shù)需要不同的日志文件時：

def logit(log_name="add.log"):
    def decorator(func)
        @wraps(func)
        def wrapper(*args, **kwargs):
            log_str = "func %s is calling " % func.__name__
            print(log_str)
            with open(log_name, "a") as log_file:
              log_file.write(log_str)
            return func(*args)
        return wrapper
    return decorator