python装饰器由浅入深_由浅入深,走进Python装饰器-----第五篇:进阶--类装饰类

**类装饰器**

@类

类

4.1 用类装饰器来扩展原类( 增加属性和方法 )

# 用类装饰器来扩展原函数, 通过对象函数化触发__call__方法,进行返回

class KuoZhan():

def __call__(self,cls):

return self.newfunc(cls)

def good(self):

print("新增的方法!")

def newfunc(self,cls):

def in_newfunc():

cls.addpty = "新增的属性"

cls.good = KuoZhan.good

# 此处返回的是一个实例化对象

return cls()

return in_newfunc

@KuoZhan() #1. KuoZhan() ==> obj 2. @KuoZhan()==> @obj ==> obj( ) 3. @KuoZhan() == obj( MyClass), ,触发__call__ ,得到 in_newfunc

class MyClass():

def func(self):

print("我是原类的方法")

# MyClass实际就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()

# 对象调用方式

obj = MyClass()

obj.func()

obj.good()

print(obj.addpty)

>>>我是原类的方法

>>>新增的方法啦!

>>>新增的属性

4.2 用类装饰器来扩展原类( 增加属性和方法 )

# 用类装饰器来扩展原函数, 直接通过类方法修饰后,进行返回

class KuoZhan():

def good():

print("新增的方法啦!")

def newfunc(cls):

def in_newfunc():

cls.addpty = "新增的属性"

cls.good = KuoZhan.good

# 此处返回的是一个类

return cls

return in_newfunc

# 类.方法 的方式

@KuoZhan.newfunc #1. MyClass = KuoZhan.newfunc(MyClass) ==> MyClass = in_newfunc

class MyClass():

def func():

print("我是原类的方法")

obj = MyClass()

obj.func()

obj.good()

print(obj.addpty)

>>>我是原类的方法

>>>新增的方法啦!

>>>新增的属性

4.3 用类装饰器来扩展原类( 改变属性和方法 )

# 用类装饰器来扩展原函数, 通过对象函数化触发__call__方法,进行返回

class KuoZhan():

def __call__(self,cls):

return self.newfunc(cls)

def func(self):

print("新增的方法!")

def newfunc(self,cls):

def in_newfunc():

cls.addpty = "新增的属性"

cls.func = KuoZhan.func

# 此处将原方法变成属性

cls.func2 = cls.func2(self)

# 此处返回的是一个实例化对象,使用的是绑定对象方法,所以上句代码使用绑定对象方法

return cls()

return in_newfunc

@KuoZhan() #1. KuoZhan() ==> obj 2. @KuoZhan()==> @obj ==> obj( ) 3. @KuoZhan() == obj( MyClass), ,触发__call__ ,得到 in_newfunc

class MyClass():

addpty = "原有的属性"

def func(self):

print("我是原类的方法")

def func2(self):

return "我是原类的方法2"

# MyClass实际就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()

obj = MyClass()

obj.func()

print(obj.addpty)

print(obj.func2)

>>>新增的方法!

>>>新增的属性

>>>我是原类的方法2

4.4 用类装饰器来扩展原类( 改变属性和方法 )

# 用类装饰器来扩展原函数, 通过直接调用类方法,进行返回

class KuoZhan():

def func():

print("新增的方法!")

def newfunc(cls):

def in_newfunc():

cls.addpty = "新增的属性"

cls.func = KuoZhan.func

# 注意直接使用类方法,不用额外参数

cls.func2 = cls.func2()

# 此处返回的是一个类

return cls

return in_newfunc

# 类.方法的方式

@KuoZhan.newfunc #1. MyClass = KuoZhan.newfunc(MyClass) ==> MyClass = in_newfunc

class MyClass():

addpty = "原有的属性"

def func():

print("我是原类的方法")

def func2():

return "我是原类的方法2"

# MyClass实际就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()

obj = MyClass()

obj.func()

print(obj.addpty)

print(obj.func2)

>>>新增的方法!

>>>新增的属性

>>>我是原类的方法2