裝飾器 - Python函數(shù)使用基礎(chǔ)教程

現(xiàn)在我們有上面兩個函數(shù)，分別用來求加法和乘法，但是現(xiàn)在我們覺得功能不夠，想在返回結(jié)果前添加一些輸出語句，一般來說我們要重構(gòu)兩個函數(shù)，就向下面這樣。
>>> def add(x, y):
... print("input:", x, y)
... return x + y
...
>>> def multiply(x, y):
... print("input:", x, y)
... return x * y
...

如果使用裝飾器我們可以像下面這樣做，雖然現(xiàn)在我們這種情形看起來使用裝飾器并沒有什么優(yōu)勢，但是如果我們要添加的不止一條打印功能，以及除了add/multiply我們還有minus/divide等函數(shù)，這個時候裝飾器的威力就體現(xiàn)出來了，我們只用修改一處代碼即可，這樣不僅提高了程序的可讀性而且也為以后我們重構(gòu)代碼省去了很多的工作量。
def decorator(F):
def new_function(x, y):
print("input:", x, y)
return F(x, y)
return new_function
@decorator
def add(x, y):
return x + y
@decorator
def multiply(x, y):
return x * y