Decorator 裝飾器

理解decorator(裝飾器)的關(guān)鍵, 在于理解在python中函數(shù)是第一公民, 并且python中萬物皆對象. 函數(shù)本身可以作為返回對象, 可以作為操作對象.
裝飾器decorator, 就是以@這個特殊操作符, 將它所修飾的函數(shù)A以它所指定的函數(shù)B來處理, 形成另一個函數(shù)C的方式, 其本質(zhì)上跟Java等語言的注解處理沒有兩樣.
@符號, 在python中可以翻譯成這樣一句話: 將下面這個函數(shù)作為參數(shù), 當(dāng)它被調(diào)用時, 以@符號后面的函數(shù)代替它, 并將被裝飾函數(shù)作為參數(shù)傳入給裝飾器. 相當(dāng)于重新賦值了被裝飾的函數(shù).

舉個例子:

@anno1 # 用anno1代替anno2
@anno2 # 用anno2代替anno3
@anno3 # 用anno3代替swap
def swap():
    pass
    
# 以上修飾, 相當(dāng)于調(diào)用了這一句 swap = anno3(anno2(anno1(swap)))
# 因此函數(shù)的裝飾器都需要以函數(shù)作為返回值
# 注意有調(diào)用順序

裝飾器當(dāng)然可以帶參數(shù), 帶上參數(shù)后實際是自動加了一層帶參數(shù)的調(diào)用

@decorator(arg1, arg2)
def func():
    pass
# 相當(dāng)于 func = decorator(arg1, arg2)(func)
# 因此要求decorator(arg1, arg2)返回的是一個可調(diào)用的對象

# 例如定義一個允許任意參數(shù)的裝飾器

# 接受任意參數(shù)的裝飾器, 定義方式與一般函數(shù)相同
def makeHtmlTag(tag, *args, **kwds):
    def real_decorator(fn):
        css_class = " class='{0}'".format(kwds["css_class"]) if "css_class" in kwds else ""
        def wrapped(*args, **kwds): # 定義一個接受參數(shù)的函數(shù), 這個函數(shù)用來替代待裝飾的函數(shù)
            return "<"+tag+css_class+">" + fn(*args, **kwds) + "</"+tag+">"
        return wrapped
    return real_decorator # 返回一個可調(diào)用的對象, 以被裝飾的函數(shù)作為參數(shù)
  
@makeHtmlTag(tag="b", css_class="bold_css")
@makeHtmlTag(tag="i", css_class="italic_css")
def hello():
    return "hello world"

Reference

python中的decorator