由淺入深,走進(jìn)Python裝飾器-----第五篇:進(jìn)階--類裝飾類

上一篇:由淺入深,走進(jìn)Python裝飾器-----第四篇:進(jìn)階--函數(shù)裝飾類

類裝飾器  
@類
類

4.1 用類裝飾器來擴(kuò)展原類( 增加屬性和方法 )

# 用類裝飾器來擴(kuò)展原函數(shù),  通過對象函數(shù)化觸發(fā)__call__方法,進(jìn)行返回
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
            # 此處返回的是一個(gè)實(shí)例化對象
            return cls()
        return in_newfunc

@KuoZhan()        #1.  KuoZhan() ==> obj    2. @KuoZhan()==> @obj  ==>  obj( )   3.  @KuoZhan() == obj( MyClass), ,觸發(fā)__call__ ,得到 in_newfunc
class MyClass():
    def func(self):
        print("我是原類的方法")
# MyClass實(shí)際就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()
# 對象調(diào)用方式
obj = MyClass() 
obj.func()
obj.good()
print(obj.addpty)

>>>我是原類的方法
>>>新增的方法啦!
>>>新增的屬性

4.1.2 用帶參數(shù)的類裝飾器來擴(kuò)展原類( 增加屬性和方法 )

# 用帶參數(shù)類裝飾器擴(kuò)展原函數(shù),  通過對象函數(shù)化觸發(fā)__call__方法,進(jìn)行返回
class KuoZhan():
    # 帶參數(shù)的裝飾器,等于實(shí)例化對象,觸發(fā)__init__,傳遞參數(shù)
    def __init__(self,num):
        self.num = num
    def __call__(self,cls):
        if self.num == 1:
            return self.newfunc(cls)
        else:
            return self.newfunc2(cls)
    def good(self):
        print("新增的方法!")
    def newfunc(self,cls):
        def in_newfunc(): 
            cls.addpty = "新增的屬性"
            cls.good = KuoZhan.good
            # 此處返回的是一個(gè)實(shí)例化對象
            return cls()
        return in_newfunc
    def newfunc2(self,cls):
        pass

@KuoZhan(1)        #1.  KuoZhan() ==> obj    2. @KuoZhan()==> @obj  ==>  obj( )   3.  @KuoZhan() == obj( MyClass), ,觸發(fā)__call__ ,得到 in_newfunc
class MyClass():
    def func(self):
        print("我是原類的方法")
# MyClass實(shí)際就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()
# 對象調(diào)用方式
obj = MyClass() 
obj.func()
obj.good()
print(obj.addpty)

>>>我是原類的方法
>>>新增的方法啦!
>>>新增的屬性

4.2 用類裝飾器來擴(kuò)展原類( 增加屬性和方法 )

# 用類裝飾器來擴(kuò)展原函數(shù), 直接通過類方法修飾后,進(jìn)行返回
class KuoZhan():
    def good():
        print("新增的方法啦!")
    def newfunc(cls):
        def in_newfunc(): 
            cls.addpty = "新增的屬性"
            cls.good = KuoZhan.good
            # 此處返回的是一個(gè)類
            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 用類裝飾器來擴(kuò)展原類( 改變屬性和方法 )

# 用類裝飾器來擴(kuò)展原函數(shù),  通過對象函數(shù)化觸發(fā)__call__方法,進(jìn)行返回
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)
            # 此處返回的是一個(gè)實(shí)例化對象,使用的是綁定對象方法,所以上句代碼使用綁定對象方法
            return cls()
        return in_newfunc

@KuoZhan()        #1.  KuoZhan() ==> obj    2. @KuoZhan()==> @obj  ==>  obj( )   3.  @KuoZhan() == obj( MyClass), ,觸發(fā)__call__ ,得到 in_newfunc
class MyClass():
    addpty = "原有的屬性"
    def func(self):
        print("我是原類的方法")
    def func2(self):
        return "我是原類的方法2"
# MyClass實(shí)際就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()
obj = MyClass() 
obj.func()
print(obj.addpty)
print(obj.func2)

>>>新增的方法!
>>>新增的屬性
>>>我是原類的方法2

4.4 用類裝飾器來擴(kuò)展原類( 改變屬性和方法 )

# 用類裝飾器來擴(kuò)展原函數(shù),  通過直接調(diào)用類方法,進(jìn)行返回
class KuoZhan():
    def func():
        print("新增的方法!")
    def newfunc(cls):
        def in_newfunc(): 
            cls.addpty = "新增的屬性"
            cls.func = KuoZhan.func
            # 注意直接使用類方法,不用額外參數(shù)
            cls.func2 = cls.func2()
            # 此處返回的是一個(gè)類
            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實(shí)際就是in_newfunc, MyClass() == in_newfunc() , 所以 obj = cls()
obj = MyClass()
obj.func()
print(obj.addpty)
print(obj.func2)

>>>新增的方法!
>>>新增的屬性
>>>我是原類的方法2