最近幾周，陸續收到幾位讀者關于裝飾器使用的提問，今天統一回復。

[[391225]]

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 

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