【每天學點AI】一個例子帶你了解Python裝飾器到底在干嘛！

今天我們來聊聊一種能給你的代碼變得“加料”的神器——Python裝飾器。

就像一杯咖啡，原本它是苦的，為了讓它符合我的口味，我給它添加了糖，添加之后就完美的符合了我的口味。

那么，裝飾器又是如何給代碼進行“加料”呢？

Python裝飾器提供了一個更為優(yōu)雅的方式來增強現(xiàn)有函數(shù)的行為，并且不需要修改現(xiàn)有的函數(shù)代碼及調(diào)用方式。

接下來通過一個案例來介紹一下裝飾器的原理和實際的行為。

這里有一個函數(shù)，他就像一個勤勞的快遞員，每次運行都送一次快遞。

現(xiàn)在我們想知道它每次送快遞需要多長時間，這時候裝飾器就派上用場了，我們可以使用裝飾器來給它添加計時的功能。

這是使用語法糖給函數(shù)添加裝飾器的用法。

下面的部分稱為原函數(shù)，上面的部分稱裝飾器函數(shù)，此時去調(diào)用原函數(shù)時，它除了會執(zhí)行原本的功能外，還會執(zhí)行裝飾器給它添加的計時的功能。

這是裝飾器最常見的寫法，但它是怎么給函數(shù)增加新功能的呢？我們來看去除掉語法糖后裝飾器的本來面目。

如果不使用語法糖，它本質(zhì)上就是裝飾器函數(shù)的一次調(diào)用，特殊的是它會將原函數(shù)作為參數(shù)傳遞給裝飾器函數(shù)。而在裝飾器函數(shù)中，它接收一個函數(shù)作為參數(shù)，并且在它的內(nèi)部定義了一個新的函數(shù)，最后將新函數(shù)進行返回。在這里我們稱timer decorator為外函數(shù)，稱wrapper為內(nèi)函數(shù)，而這個func就是原函數(shù)。

外函數(shù)將內(nèi)函數(shù)名稱進行返回并賦值為原函數(shù)的函數(shù)名，下面在調(diào)用原函數(shù)的函數(shù)名時，實際上調(diào)用的是內(nèi)函數(shù)。

內(nèi)函數(shù)被執(zhí)行后，首先會記錄當前時間，然后調(diào)用func，func是什么？是外函數(shù)的形參，這里可能有人會疑惑，外函數(shù)已經(jīng)執(zhí)行完了，為什么func還會存在呢？

這其實就是裝飾器函數(shù)的立身之本：內(nèi)函數(shù)所使用的外函數(shù)數(shù)據(jù)不會隨著外函數(shù)的調(diào)用而銷毀，所以此時func就指向了原函數(shù)，在調(diào)用內(nèi)函數(shù)時，這里的func()可以去執(zhí)行原函數(shù)，接著記錄一下時間，最后通過差值得到執(zhí)行函數(shù)所需要的時間。至此，通過裝飾器我們不僅為原函數(shù)添加了新的功能，并且還沒有修改其調(diào)用方式。

而語法糖的存在就是為了省略這一行代碼，由此就形成了我們常見的裝飾器了。

完整視頻鏈接：

https://www.bilibili.com/video/BV1fGHDepExY?t=1.2