講 Python 裝飾器前,我想先舉個例子,雖有點污,但跟裝飾器這個話題很貼切。
每個人都有的內褲主要功能是用來遮羞,但是到了冬天它沒法為我們防風御寒,咋辦?我們想到的一個辦法就是把內褲改造一下,讓它變得更厚更長,這樣一來,它不僅有遮羞功能,還能提供保暖,不過有個問題,這個內褲被我們改造成了長褲后,雖然還有遮羞功能,但本質上它不再是一條真正的內褲了。于是聰明的人們發明長褲,在不影響內褲的前提下,直接把長褲套在了內褲外面,這樣內褲還是內褲,有了長褲后寶寶再也不冷了。裝飾器就像我們這里說的長褲,在不影響內褲作用的前提下,給我們的身子提供了保暖的功效。
談裝飾器前,還要先要明白一件事,Python 中的函數和 Java、C++不太一樣,Python 中的函數可以像普通變量一樣當做參數傳遞給另外一個函數,例如:
def foo(): print("foo")def bar(func): func()bar(foo)
正式回到我們的主題。裝飾器本質上是一個 Python 函數或類,它可以讓其他函數或類在不需要做任何代碼修改的前提下增加額外功能,裝飾器的返回值也是一個函數/類對象。它經常用于有切面需求的場景,比如:插入日志、性能測試、事務處理、緩存、權限校驗等場景,裝飾器是解決這類問題的絕佳設計。有了裝飾器,我們就可以抽離出大量與函數功能本身無關的雷同代碼到裝飾器中并繼續重用。概括的講,裝飾器的作用就是為已經存在的對象添加額外的功能。
先來看一個簡單例子,雖然實際代碼可能比這復雜很多:
def foo(): print('i am foo')
現在有一個新的需求,希望可以記錄下函數的執行日志,于是在代碼中添加日志代碼:
def foo(): print('i am foo') logging.info("foo is running")
如果函數 bar()、bar2() 也有類似的需求,怎么做?再寫一個 logging 在 bar 函數里?這樣就造成大量雷同的代碼,為了減少重復寫代碼,我們可以這樣做,重新定義一個新的函數:專門處理日志 ,日志處理完之后再執行真正的業務代碼
def use_logging(func): logging.warn("%s is running" % func.__name__) func()def foo(): print('i am foo')use_logging(foo)
這樣做邏輯上是沒問題的,功能是實現了,但是我們調用的時候不再是調用真正的業務邏輯 foo 函數,而是換成了 use_logging 函數,這就破壞了原有的代碼結構, 現在我們不得不每次都要把原來的那個 foo 函數作為參數傳遞給 use_logging 函數,那么有沒有更好的方式的呢?當然有,答案就是裝飾器。
簡單裝飾器
def use_logging(func):def wrapper(): logging.warn("%s is running" % func.__name__)return func() # 把 foo 當做參數傳遞進來時,執行func()就相當于執行foo()return wrapperdef foo(): print('i am foo')foo = use_logging(foo) # 因為裝飾器 use_logging(foo) 返回的時函數對象 wrapper,這條語句相當于 foo = wrapperfoo() # 執行foo()就相當于執行 wrapper()
新聞熱點
疑難解答