示例

Python中的裝飾器既可以是函數(shù)也可以是類。在前幾節(jié)中，我們使用的是函數(shù)裝飾器。現(xiàn)在，我們將學(xué)習(xí)如何定義類裝飾器。

我們將定義充當(dāng)裝飾器的自定義類。當(dāng)用類裝飾/修飾函數(shù)時(shí)，該函數(shù)就成為該類的實(shí)例。讓我們來一探究竟:

如上所示，我們創(chuàng)建了一個(gè)簡單的類裝飾器。任何類要成為裝飾器，都需要實(shí)現(xiàn)__call__()方法。__call__()方法的作用與函數(shù)裝飾器中的包裝器函數(shù)相同。

現(xiàn)在讓我們用這個(gè)類來裝飾一個(gè)函數(shù):

運(yùn)行程序輸出結(jié)果如下：

帶*args和**kwargs參數(shù)的類裝飾器

為了讓類裝飾器使用參數(shù)用*args和**kwargs,我們需要使用這些參數(shù)來實(shí)現(xiàn)__call__()方法,并將其傳遞給被修飾的函數(shù)。通用的表示方法如下：

上面的代碼中，類裝飾器的__call__()方法接受*args和**kwargs參數(shù)。在代碼行中，call方法將它們傳遞給所裝飾的函數(shù)：self.func(*args,**kwargs)。

現(xiàn)在我們用這個(gè)類裝飾器來裝飾一個(gè)函數(shù):

類裝飾裝飾函數(shù)

運(yùn)行程序所輸出結(jié)果類似如下：

Hello Solo Cui

帶return語句的類裝飾器

還記得在函數(shù)裝飾器的wrapper函數(shù)中，我們可使用return關(guān)鍵字返回裝飾過的函數(shù)。我們會在這里做同樣的事情，但是在__call__方法里面。

返回結(jié)果的類裝飾器

運(yùn)行程序輸出結(jié)果如下：

TRACY CUI

如上所示，我們是在__call__方法中將修飾過的函數(shù)返回:

return self.func(*args,**args)

至此，關(guān)于Python的裝飾器就介紹完了。