如何正確用裝飾器

這篇文章主要講解了“如何正確用裝飾器”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“如何正確用裝飾器”吧！

1. 問題

大概問題是這樣，想要自定義一個Python裝飾器，問我這樣寫裝飾器行不行?如果不行，那又是為什么?

import datetime import time  def print_time(g):     def f():         print('開始執行時間')         print(datetime.datetime.today())                  g()                  print('結束時間')         print(datetime.datetime.today())     f()

下面使用 print_time裝飾函數 foo：

@print_time def foo():     time.sleep(2)     print('hello world')

當調用 foo函數時，拋出如下異常：

foo()  --------------------------------------------------------------------------- TypeError                                 Traceback (most recent call last) <ipython-input-27-c19b6d9633cf> in <module> ----> 1 foo()  TypeError: 'NoneType' object is not callable

所以，按照如上定義 print_time裝飾器，肯定是不行的。

2. 為什么不行

要想明白為啥不行，首先要知道裝飾器這個語法的本質。其實很簡單，@print_time裝飾foo函數等于：

foo = print_time(foo)

就是這一行代碼，再也沒有其他。

因為上面的 print_time 無返回值，所以賦值給 foo 函數后，foo 函數變為 None，所以當調用 foo() 時拋出 'NoneType'  object is not callable

這也就不足為奇了。

3. 應該怎么寫

print_time 需要返回一個函數，這樣賦值給 foo函數后，正確寫法如下所示：

import datetime import time  def print_time(g):     def f():         print('開始執行時間')         print(datetime.datetime.today())                  g()                  print('結束時間')         print(datetime.datetime.today())     return f

裝飾 foo：

@print_time def foo():     time.sleep(2)     print('hello world')

調用 foo ，運行結果如下：

foo()  開始執行時間 2021-04-02 22:32:49.114124 hello world 結束時間 2021-04-02 22:32:51.119506

一切正常

4. 裝飾器好處

上面自定義print_time裝飾器，除了能裝飾foo函數外，還能裝飾任意其他函數和類內方法。

裝飾任意一個函數 foo2：

@print_time def foo2():   print('this is foo2')

裝飾類內方法 foo3，需要稍微修改原來的print_time：

def print_time(g):     def f(*args, **kargs):         print('開始執行時間')         print(datetime.datetime.today())              g(*args, **kargs)              print('結束時間')         print(datetime.datetime.today())     return f

為類MyClass中foo3方法增加print_time裝飾：

class MyClass(object):     @print_time     def foo3(self):         print('this is a method of class')

執行結果如下：

MyClass().foo3()  開始執行時間 2021-04-02 23:16:32.094025 this is a method of class 結束時間 2021-04-02 23:16:32.094078

以上就是裝飾器的通俗解釋，平時可以多用用，讓我們的代碼更加精煉、可讀。

感謝各位的閱讀，以上就是“如何正確用裝飾器”的內容了，經過本文的學習后，相信大家對如何正確用裝飾器這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！