快捷導(dǎo)航

一文帶你探尋Python中的裝飾器

更新時(shí)間：2023年04月12日 09:56:34 作者：真的不能告訴你我的名字

這篇文章就來(lái)和大家詳細(xì)講一講Python中裝飾器的相關(guān)知識(shí)，文中的示例代碼講解詳細(xì)，對(duì)我們深入了解Python有一定的幫助，感興趣的可以了解一下

上述是弟弟前段時(shí)間去面試運(yùn)維開(kāi)發(fā)，遇到的問(wèn)題，emmm，運(yùn)維是一個(gè)很雜的職業(yè)，在小公司，總結(jié)一句話就是寬而淺，痛定思痛，決定來(lái)了解一下python特性，于是乎，就有了這篇文章。

這邊文章，我們將介紹python生成器，使用環(huán)境為: Python 3.6.8

什么是裝飾器

要理解裝飾器之前，我們需要了解什么是閉包函數(shù)。

閉包函數(shù)

我們簡(jiǎn)單寫(xiě)個(gè)demo，再解釋一下什么是閉包函數(shù)。

def exterFunc(x):
  def innerFunc(y):
    return x * y
  
  return innerFunc

def main() -> None:
  f = exterFunc(6)
  result = f(5)

  print(result)

if __name__ == '__main__':
  main()

可以看到，上述代碼所示，所謂的閉包函數(shù)是指: 閉包函數(shù)是指在函數(shù)中再定義函數(shù)，內(nèi)部函數(shù)可以訪問(wèn)外部的變量，在外部函數(shù)中，將內(nèi)部函數(shù)作為返回值返回。

可以看到上述例子中，我們定義了exterFunc的函數(shù)，它將接收一個(gè)形參x，在exterFunc函數(shù)中又中定義了innerFunc，它也接收一個(gè)形參y，在innerFunc函數(shù)中，返回x * y，沒(méi)錯(cuò)，內(nèi)部函數(shù)可以訪問(wèn)外部函數(shù)傳入的變量，最后將exterFunc作為返回值返回，這就是閉包函數(shù)。

最簡(jiǎn)單的裝飾器

裝飾器是一種很特殊的函數(shù)，可以接收函數(shù)作為形參，且返回一個(gè)新的函數(shù)，在我們上一篇介紹生成器的時(shí)候，還記得我們使用memory_profiler庫(kù)來(lái)打印函數(shù)的內(nèi)存運(yùn)行情況么？這就是用的裝飾器。

我們可以寫(xiě)個(gè)最簡(jiǎn)單的例子，來(lái)闡述一下python裝飾器，即：

def foo(func):
    def wrapper():
        print("裝飾器開(kāi)始運(yùn)行了")
        func()
        print("裝飾器結(jié)束運(yùn)行了")

    return wrapper

@foo
def sayHello():
    print("hello pdudo in juejin")

def main() -> None:
  sayHello()

if __name__ == '__main__':
  main()

上面代碼，我們定義了一個(gè)裝飾器foo，foo需要傳入一個(gè)函數(shù)， foo內(nèi)部有一個(gè)函數(shù)wrapper。這樣的函數(shù)中包函數(shù)，我們將其稱之為閉包函數(shù)，后面會(huì)介紹閉包函數(shù)。言歸正傳，在wrapper函數(shù)中，我們可以在運(yùn)行func函數(shù)的時(shí)候，再其執(zhí)行前后語(yǔ)句。

需要調(diào)用裝飾器的時(shí)候，只需要@加上函數(shù)名稱即可。

為什么需要裝飾器

要解釋這個(gè)問(wèn)題，我們可以看來(lái)了解下，裝飾器解決了一些什么問(wèn)題：

解決代碼重復(fù)性，對(duì)于經(jīng)常需要實(shí)現(xiàn)類似的功能而言，可以將該功能抽離出來(lái)，作為裝飾器來(lái)調(diào)用，從而避免代碼重復(fù)。
增強(qiáng)代碼可讀性，在不修改原始代碼的前提下，可以利用裝飾器在函數(shù)前后增加代碼，例如處理異常、記錄日志等等，可以利用裝飾器將附加功能和函數(shù)主要功能分開(kāi)，增加代碼可讀性。

說(shuō)了那么多，我們來(lái)列舉一個(gè)最簡(jiǎn)單的例子，利用裝飾器打印一下函數(shù)的運(yùn)行時(shí)間。

import time

def getExecTimers(func):
  def wrapper():
    startTimes = time.time()
    func()
    endTimes = time.time()
    print("函數(shù)運(yùn)行時(shí)間: " , endTimes - startTimes ,"s")
  return wrapper

@getExecTimers
def testFunc():
  print("開(kāi)始執(zhí)行函數(shù)")
  time.sleep(5)
  print("函數(shù)執(zhí)行結(jié)束")

def main() -> None:
  testFunc()
  
if __name__ == '__main__':
  main()

這個(gè)裝飾器，會(huì)記錄函數(shù)的運(yùn)行時(shí)間?？梢钥吹?，我們?yōu)檫@個(gè)函數(shù)增加了一個(gè)附屬功能，但是又沒(méi)有修改到原始函數(shù)。

上述案例，應(yīng)該可以證明為什么需要使用裝飾器了。

裝飾器用法

上述我們討論了最簡(jiǎn)單的裝飾器寫(xiě)法，并且寫(xiě)了一個(gè)小功能，即打印函數(shù)的運(yùn)行時(shí)間。接下來(lái)，我們要看下裝飾器的其他寫(xiě)法。

不是用語(yǔ)法糖調(diào)用

還記得上面我們調(diào)用裝飾器，是使用的@+裝飾器名稱么？其實(shí)這是python的語(yǔ)法糖，如果不是用語(yǔ)法糖的話，應(yīng)該是這樣來(lái)使用的：

def foo(func):
    def wrapper():
        print("裝飾器開(kāi)始運(yùn)行了")
        func()
        print("裝飾器結(jié)束運(yùn)行了")

    return wrapper


def sayHello():
    print("hello pdudo in juejin")

def main() -> None:
  f1 = sayHello
  f2 = foo(f1)

  f2()

if __name__ == '__main__':
  main()

完整的寫(xiě)法應(yīng)該如下代碼所示，這是一個(gè)完整的閉包調(diào)用邏輯。

f1 = sayHello
f2 = foo(f1)

f2()

而在函數(shù)前加上@+裝飾器名稱，是一種python的語(yǔ)法糖

帶參數(shù)的裝飾器

這里要做一個(gè)鋪墊，在python中，有2個(gè)特殊的變量，分別為*args和**kwargs，都是用來(lái)處理不定量參數(shù)的，分別代表的含義為:

*args: 將會(huì)將參數(shù)打包為元組
**kwargs: 將會(huì)打包字典傳遞給函數(shù)

def foo(func):
    def wrapper(*args,**kwargs):
        print("裝飾器開(kāi)始運(yùn)行了")
        print("裝飾器捕獲到的參數(shù): " ,args,**kwargs)
        func(*args,**kwargs)
        print("裝飾器結(jié)束運(yùn)行了")

    return wrapper

@foo
def sayHello(a,b,c,dicts):
    print("傳入的參數(shù): " , a,b,c)
    print("傳入的參數(shù): " , dicts)

def main() -> None:
  sayHello(1,2,3,{"name":"juejin"})

if __name__ == '__main__':
  main()

在裝飾器中，若我們要給函數(shù)傳遞參數(shù)，是需要先將參數(shù)傳遞給裝飾器，而在裝飾器中接收后再進(jìn)行傳遞的，所以代碼才會(huì)是這樣的：

def foo(func):
    def wrapper(*args,**kwargs):
        print("裝飾器開(kāi)始運(yùn)行了")
        print("裝飾器捕獲到的參數(shù): " ,args,**kwargs)
        func(*args,**kwargs)
        print("裝飾器結(jié)束運(yùn)行了")

首先，我們?cè)谧鰝鬟f調(diào)用的時(shí)候，wrapper應(yīng)該調(diào)用形參來(lái)接收，接收后，再進(jìn)行傳遞給函數(shù)func。