裝飾器其實一直是我的一個"老大難"。這個知識點就放在那，但是拖延症。。。 其實在平常寫寫指令碼的過程中，這個知識點你可能用到不多 但在面試的時候，這可是一個高頻問題。 ### 一、什麼是裝飾器 所謂的裝飾器，其實就是通過裝飾器函式，來修改原函式的一些功能，使得原函式不需要修改。 這一句話理解起來可能沒那麼輕鬆，那先來看一個"傻瓜"函式。 放心，絕對不是"Hello World"！ ``` def hello(): print("你好，裝飾器") ``` 腫麼樣，木騙你吧？ 哈哈，這個函式不用執行相信大家都知道輸出結果：``"你好，裝飾器"``。 那如果我想讓``hello()``函式再實現個其他功能，比如多列印一句話。 那麼，可以這樣"增強"一下： ``` def my_decorator(func): def wrapper(): print("這是裝飾後具有的新輸出") func() return wrapper def hello(): print("你好，裝飾器") hello = my_decorator(hello) hello() ``` 執行結果： ``` 這是裝飾後具有的新輸出 你好，裝飾器 [Finished in 0.1s] ``` 很顯然，這個"增強"沒啥作用，但是可以幫助理解裝飾器。 當執行最後的``hello()``函式時，呼叫過程是這樣的： 1. ``hello = my_decorator(hello)``中，變數hello指向的是``my_decorator()`` 2. ``my_decorator(func)``中傳參是``hello``，返回的``wrapper``,因此又會呼叫到原函式``hello()`` 3. 於是乎，先打印出了``wrapper()``函式裡的，然後才打印出``hello()``函式裡的 那上述程式碼裡的``my_decorator()``就是一個裝飾器。 它改變了``hello()``的行為，但是並沒有去真正的改變``hello()函式``的內部實現。 但是，python一直以"優雅"被人追捧，而上述的程式碼顯然不夠優雅。 ### 二、優雅的裝飾器 所以，想讓上述裝飾器變得優雅，可以這樣寫： ``` def my_decorator(func): def wrapper(): print("這是裝飾後具有的新輸出") func() return wrapper @my_decorator def hello(): print("你好，裝飾器") hello() ``` 這裡的``@my_decorator``就相當於舊程式碼的``hello = my_decorator(hello)``，``@``符號稱為語法糖。 那如果還有其他函式也需要加上類似的裝飾，直接在函式的上方加上``@my_decorator``就可以，大大提高函式 的重複利用與可讀性。 ``` def my_decorator(func): def wrapper(): print("這是裝飾後具有的新輸出") func() return wrapper @my_decorator def hello(): print("你好，裝飾器") @my_decorator def hello2(): print("你好，裝飾器2") hello2() ``` 輸出： ``` 這是裝飾後具有的新輸出 你好，裝飾器2 [Finished in 0.1s] ``` ### 三、帶引數的裝飾器 #### 1. 單個引數 上面的只是一個非常簡單的裝飾器，但是實際場景中，很多函式都是要帶有引數的，比如hello(people_name)。 其實也很簡單，要什麼我們就給什麼唄，直接在對應裝飾器的``wrapper()``上，加上對應的引數： ``` def my_decorator(func): def wrapper(people_name): print("這是裝飾後具有的新輸出") func(people_name) return wrapper @my_decorator def hello(people_name): print("你好，{}".format(people_name)) hello("張三") ``` 輸出： ``` 這是裝飾後具有的新輸出 你好，張三 [Finished in 0.1s] ``` #### 2. 多個引數 但是還沒完，這樣雖然簡單，但是隨之而來另一個問題：因為並不是所有函式引數都是一樣的， 當其他要使用裝飾器的函式引數不止這個一個腫麼辦？比如： ``` @my_decorator def hello3(speaker, listener): print("{}對{}說你好！".format(speaker, listener)) ``` 沒關係，在python裡，``*args``和``**kwargs``表示接受任意數量和型別的引數，所以我們可以這樣 寫裝飾器裡的``wrapper()``函式： ``` def my_decorator(func): def wrapper(*args, **kwargs): print("這是裝飾後具有的新輸出") func(*args, **kwargs) return wrapper @my_decorator def hello(people_name): print("你好，{}".format(people_name)) @my_decorator def hello3(speaker, listener): print("{}對{}說你好！".format(speaker, listener)) hello("老王") print("------------------------") hello3("張三", "李四") ``` 同時執行下``hello("老王")``，和``hello3("張三", "李四")``，看結果： ``` 這是裝飾後具有的新輸出 你好，老王 ------------------------ 這是裝飾後具有的新輸出 張三對李四說你好！ [Finished in 0.1s] ``` #### 3. 自定義引數 上面2種，裝飾器都是接收外來的引數，其實裝飾器還可以接收自己的引數。 比如，我加個引數來控制下裝飾器中列印資訊的次數： ``` def count(num): def my_decorator(func): def wrapper(*args, **kwargs): for i in range(num): print("這是裝飾後具有的新輸出") func(*args, **kwargs) return wrapper return my_decorator @count(3) def hello(people_name): print("你好，{}".format(people_name)) hello("老王") ``` 注意，這裡``count``裝飾函式中的2個``return``. 執行下，應該會出現3次： ``` 這是裝飾後具有的新輸出 你好，老王 這是裝飾後具有的新輸出 你好，老王 這是裝飾後具有的新輸出 你好，老王 [Finished in 0.1s] ``` #### 4. 內建裝飾器``@functools.wrap`` 現在多做一步探索，我們來列印下下面例子中的hello()函式的元資訊： ``` def my_decorator(func): def wrapper(*args, **kwargs): print("這是裝飾後具有的新輸出") func(*args, **kwargs) return wrapper @my_decorator def hello(people_name): print("你好，{}".format(people_name)) print(hello.__name__) #看下hello函式的元資訊 ``` 輸出： ``` wrapper ``` 這說明了，它不再是以前的那個 ``hello()`` 函式，而是被 ``wrapper()`` 函式取代了。 如果我們需要用到元函式資訊，那怎麼保留它呢？這時候可以用內建裝飾器``@functools.wrap``。 ``` import functools def my_decorator(func): @functools.wraps(func) def wrapper(*args, **kwargs): print("這是裝飾後具有的新輸出") func(*args, **kwargs) return wrapper @my_decorator def hello(people_name): print("你好，{}".format(people_name)) print(hello.__name__) ``` 執行下： ``` hello [Finished in 0.1s] ``` 好記性不如爛筆頭，寫一下理解一下會好很多。 下面還分享類的裝飾器，以及裝飾器所用場景。