由淺入深,走進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.1.2 用帶參數的類裝飾器來擴展原類( 增加屬性和方法 )

# 用帶參數類裝飾器擴展原函數,  通過對象函數化觸發__call__方法,進行返回
class KuoZhan():
    # 帶參數的裝飾器,等于實例化對象,觸發__init__,傳遞參數
    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
            # 此處返回的是一個實例化對象
            return cls()
        return in_newfunc
    def newfunc2(self,cls):
        pass

@KuoZhan(1)        #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