python裝飾器代碼深入講解

Posted on by

python裝飾器就是用於擴展原函數功能的一種函數，這個函數特殊的地方就是它的返回值也是一個函數，使用Python裝飾器的一個好處就是：在不需要修改原函數代碼的情況下，給函數增加新的功能。

先來看個例子：

def say():
 print('Nice day')

say()
# 這個函數的輸出為：
Nice day

現在，我想在輸出Nice day的前面再打印一行****************，類似下面的效果：

**************** 

Nice day

一般情況下，我可以修改上面的代碼：

def say():
 print('****************')
 print('Nice day')

say()

可是，如果我忽然發現自己看錯瞭需求，這時候又要把代碼修改到原來的樣子，慶幸的是我隻是在原來函數的基礎上增加瞭一行代碼，想要回到原來的狀態並不難，可如果我是修改瞭復雜的邏輯，代碼有一百行呢，難道我還要一步步撤銷嗎？顯然做不到，不過沒關系，肯定還有別的辦法：

def say():
 print('Nice day')

def outer(): # 重新定義一個新函數
 print('****************') # 處理新的邏輯
 say() # 再調用原來的函數

outer()
# 現在的輸出為：
'''
****************
Nice day
'''

怎麼樣，看上去已經滿足要求瞭吧，不過仔細一看，就能發現新的問題，如果不僅僅是say()函數需要打印****************，新來的talk()函數也需要呢，這時候我又要再寫一個outer()函數嗎？這會累壞丹丹的，所以得再想個辦法：

def say():
 print('Nice day')

def talk():
 print('I am talk')

def outer(func): # 接收一個函數
 print('****************') # 處理新的邏輯
 func() # 調用傳遞過來的函數

outer(talk) # 把talk函數作為參數傳遞過去
# 輸出結果如下：
'''
****************
I am talk
'''

這時，不管有幾個函數需要打印****************，我直接把函數名傳給outer()就可以啦，是不是方便很多^-^ 但是勤勞的丹丹會止步於此嗎？肯定不會，於是又把代碼做瞭如下修改：

def say():
 print('Nice day')

def outer(func):
 def inner():
 print('****************')
 func() # 相當於 say()
 return inner

s = outer(say) # 相當於 s = inner
s() # 相當於 inner()

猜猜這次的是輸出是什麼~當然還是和上面一樣啦！其實這裡隻是把處理邏輯的部分封裝在瞭一個函數裡面，調用outer(say)的時候，把say傳給outer，獲得返回值inner給s，此時的s就相當於inner，s()也就相當於inner()，所以會輸出：

**************

Nice day

這就是一個最簡單的裝飾器啦，是不是很簡單~ 但是我們每次在使用的時候還需要先賦值給一個變量(這裡的s)，然後再經由s調用，未免違反瞭丹丹“多一行代碼都是累贅”的原則，所以我們再修改一下代碼：

def outer(func):
 def inner():
 print('****************')
 func()
 return inner

@outer # 用outer裝飾say
def say():
 print('Nice day')

say() # 調用say函數

我把outer和say調換瞭一下位置，先定義瞭outer函數，@outer表示用outer裝飾say，這樣直接用say()就能實現我想先打印一行****************的功能瞭，如果不調換兩個函數的位置，是會報NameError: name 'outer' is not defined的錯誤的噢（作用域的原因，outer未定義），這個應該算是復雜一點的裝飾器瞭吧，哈哈
這時候很多細心同學肯定就會問瞭，你寫的都是無參的呀，那如果我的函數有參數怎麼辦呢，參數還是不固定的又該怎麼辦呢？萬能的python+聰明的丹丹當然可以解決：

# 帶參數的裝飾器
def outer(func):
 def inner(name):
 func(name)
 return inner

@outer
def say(name):
 print('name is %s.' % (name))

say('dandan')
# name is dandan.

不過這個參數個數是固定的，萬一我又突發奇想，想多傳一個hobby或者age怎麼辦呢？

# 帶不定參數的裝飾器
def outer(func):
 def inner(*args, **kwargs):
 func(*args, **kwargs)
 return inner

@outer
def say(name, age):
 print('name is %s, age is %d.' % (name, age))

@outer
def talk(name, age, hobby):
 print('name is %s, age is %d, hobby is %s.' % (name, age, hobby))

say('dandan', 18)
talk('dandan', 18, 'Coding')
'''
name is dandan, age is 18.
name is dandan, age is 18, hobby is Coding.
'''

如果我要新增的功能有很多，一個裝飾器搞不定，怎麼辦呢？我可以同時使用多個裝飾器嗎？當然可以：

# 多個裝飾器
def outer(func):
 def inner(*args, **kwargs):
 print('****************')
 func(*args, **kwargs)
 return inner

def outer2(func):
 def inner2(*args, **kwargs):
 print('這裡有1w+新功能')
 func(*args, **kwargs)
 return inner2

@outer
@outer2
def say(name, age):
 print('name is %s, age is %d.' % (name, age))

@outer
@outer2
def talk(name, age, hobby):
 print('name is %s, age is %d, hobby is %s.' % (name, age, hobby))

say('dandan', 18)
talk('dandan', 18, 'Coding')
'''
****************
這裡有1w+新功能
name is dandan, age is 18.
****************
這裡有1w+新功能
name is dandan, age is 18, hobby is Coding.
'''

要註意的是，多個裝飾器的執行順序是從第一個裝飾器開始，執行到最後一個裝飾器，再執行函數本身。

到此這篇關於python裝飾器代碼深入講解的文章就介紹到這瞭,更多相關python裝飾器內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：