如何在 Python 中裝飾一個(gè)類

第如何在Python中裝飾一個(gè)類？它使用額外的Python語句修改舊對(duì)象并返回相同的引用。

例如，考慮下面的類，它有兩個(gè)方法：__init__和display。__init__方法在顯示輸出名稱時(shí)初始化名稱變量：

classStudent:

def__init__(self,name):

=name

defdisplay(self):

print(Name:,)

要在Python中裝飾這個(gè)類，我們可以向該類添加新方法或修改現(xiàn)有方法，或者兩者都做。

此外，在Python中有兩種方法可以做到這一點(diǎn)，要么使用函數(shù)裝飾器，要么使用類裝飾器。

讓我們一個(gè)一個(gè)看下例子。

使用函數(shù)裝飾器裝飾類

要使用函數(shù)裝飾器來裝飾類，接受類作為參數(shù)，修改其代碼并在最后返回類。

defmydecorator(student):

#defineanewdisplaymethod

defnewdisplay(self):

print(Name:,)

print(Subject:Programming)

#replacethedisplaywithnewdisplay

student.display=newdisplay

#returnthemodifiedstudent

returnstudent

@mydecorator

classStudent:

def__init__(self,name):

=name

defdisplay(self):

print(Name:,)

obj=Student(PencilProgrammer)

obj.display()

Name:PencilProgrammer

Subject:Programming

如果類中不存在display方法，則newdisplay將作為display方法添加到類中。

由于類的引用在裝飾器函數(shù)中是可用的，我們除了修改現(xiàn)有方法外，還可以為類添加新的屬性和方法

使用類裝飾器裝飾類

要使用類裝飾器裝飾類，接受類的引用作為參數(shù)（在裝飾器的__init__方法中），在__call__方法中修改其代碼，最后返回修改后的類的實(shí)例。

classMydecorator:

#accepttheclassasargument

def__init__(self,student):

self.student=student

#accepttheclasss__init__methodarguments

def__call__(self,name):

#defineanewdisplaymethod

defnewdisplay(self):

print(Name:,)

print(Subject:Python)

#replacedisplaywithnewdisplay

self.student.display=newdisplay

#returntheinstanceoftheclass

obj=self.student(name)

returnobj

@Mydecorator

classStudent:

def__init__(self,name):

=name

defdisplay(self):

print(Name:,)

obj=Student(PencilProgrammer)

obj.di