python裝飾器帶括號和不帶括號的語法和用法

裝飾器的寫法補充：

通常裝飾器的寫法是@func()，而有的時候為了減少出錯率，可能會寫成@func，沒有()括號，這時我們可以這樣定義，來減少括號。下面通過兩個例子還看。

一般裝飾器的寫法：

def log(func=None):
    
    def inner(*args, **kwargs):
        print('do something before')
        res = func(*args, **kwargs)
        print('do something after')
        return rees
    
    return inner

#使用裝飾器
@log()
def my_func():
    print('i am my_func')

#運行這個函數
my_func()

 

運行結果：

do something before
i am my_func
do something after

 

可以不帶括號的裝飾器寫法

需要增加判斷在函數內，用於判斷使用裝飾器的函數是否可以調用，以達到括號的自由寫法

判斷的裝飾器寫法

def log(func=None):
    
    #需要再次嵌套一層裝飾器，才可以供下面運行時使用
    def wrapper(fun)
        def inner(*args, **kwargs):
            print('do something before')
            res = fun(*args, **kwargs)
            print('do something after')
            return rees
        
        return inner
    
    #判斷func(參數)：
    if func is None：
        return wrapper
        
    #如果func是可以調用的函數    
    elif callable(func):
        return wrapper(func)
    

#使用裝飾器的時候，兩種寫法都可以運行

第一種不帶括號：
@log
def my_func():
    print('i am my_func')

第二種帶括號的：
@log()
def my_func():
    print('i am my_func')


#運行這個函數
my_func()

 

兩種方式的一樣，運行結果:

1 do something before
2 i am my_func 3 do something after

 

不帶括號的理解：

 

@log =>等同於 my_func = log(my_func)

帶括號的理解：

@log() =>等同於 my_func = log()(my_func)

 

 

結論： 內嵌之后，可以增加判斷條件，增加了裝飾器的靈活性。