由淺入深,走進Python裝飾器-----第四篇:進階--函數裝飾類

上一篇:由淺入深,走進Python裝飾器-----第三篇:進階--類裝飾函數

函數裝飾器

@函數
類

本篇只是用一個簡單的例子,驗證函數可以對類進行修飾, 但是由于返回的值為函數,并不是類, 把原有類的實例化調用方式變為了函數調用.
#### 要點:讀者還應注意返回值為類或者對象時候的相應參數使用區別

3.1 用函數裝飾器來擴展原類

def KuoZhan(cls):
    def new_func():
        print("類里新增的方法new_func")
    def new_class():
        cls.say = "類里新增的屬性"
        cls.new_func = new_func
        # 此處返回類,注意和3.2返回的對象例子做參數對比, 
        return cls
    return new_class
# 通過函數裝飾器給類增加屬性或者方法 
@KuoZhan        #1.  @KuoZhan ==> KuoZhan( )                                                                       
class MyClass():
    def func():
        print("我是原類里的方法func")
# 此處改變了原類的調用方式, 實際上MyClass已經是一個函數名了
# type(MyClass) = <class 'function'>
obj = MyClass()
# obj 拿到的是函數new_class的返回值cls
obj.func()
obj.new_func()
print(obj.say)

>>>我是原類里的方法func
>>>類里新增的方法new_func
>>>類里新增的屬性

3.2 用函數裝飾器來擴展原類

def KuoZhan(cls):
    # 增加參數self
    def new_func(self):
        print("類里新增的方法new_func")
    def new_class():
        cls.say = "類里新增的屬性"
        cls.new_func = new_func
        # 此處返回對象
        return cls()
    return new_class
# 通過函數裝飾器給類增加屬性或者方法 
@KuoZhan        #1.  @KuoZhan ==> KuoZhan( )                                                                       
class MyClass():
    # 增加參數self
    def func(self):
        print("我是原類里的方法func")
# 此處改變了原類的調用方式, 實際上MyClass已經是一個函數名了
# type(MyClass) = <class 'function'>
obj = MyClass()
# obj 拿到的是函數new_class的返回值cls
obj.func()
obj.new_func()
print(obj.say)

>>>我是原類里的方法func
>>>類里新增的方法new_func
>>>類里新增的屬性

3.3 用帶參數的函數裝飾器來擴展原類

def outer(n):
    def KuoZhan1(cls):
        def new_func():
            print("類里新增的方法new_func")
        def new_class():
            cls.say = "類里新增的屬性"
            cls.new_func = new_func
            # 此處返回類,注意和3.2返回的對象例子做參數對比, 
            return cls
        return new_class
    def KuoZhan2(cls):
        pass
    if n == 1:
        return KuoZhan1
    else:
        return KuoZhan2

# 通過函數裝飾器給類增加屬性或者方法 
@outer(1)        #1. outer(1) ==> KuoZhan1   2.@out(1) == @KuoZhan1                                                                        
class MyClass():
    def func():
        print("我是原類里的方法func")
# 此處改變了原類的調用方式, 實際上MyClass已經是一個函數名了
# type(MyClass) = <class 'function'>
obj = MyClass()
# obj 拿到的是函數new_class的返回值cls
obj.func()
obj.new_func()
print(obj.say)

>>>我是原類里的方法func
>>>類里新增的方法new_func
>>>類里新增的屬性