通過 Python 裝飾器實現DRY(不重複程式碼)原則

發表於2016-01-17

Python裝飾器是一個消除冗餘的強大工具。隨著將功能模組化為大小合適的方法,即使是最複雜的工作流,裝飾器也能使它變成簡潔的功能。

例如讓我們看看Django web框架,該框架處理請求的方法接收一個方法物件,返回一個響應物件:

我最近遇到一個案例,需要編寫幾個滿足下述條件的api方法:

  • 返回json響應
  • 如果是GET請求,那麼返回錯誤碼

做為一個註冊api端點例子,我將會像這樣編寫:

然而這樣我將會在每個api方法中編寫json響應和錯誤返回的程式碼。這將會導致大量的邏輯重複。所以讓我們嘗試用裝飾器實現DRY原則吧。

裝飾器簡介

如果你不熟悉裝飾器,我可以簡單解釋一下,實際上裝飾器就是有效的函式包裝器,python直譯器載入函式的時候就會執行包裝器,包裝器可以修改函式的接收引數和返回值。舉例來說,如果我想要總是返回比實際返回值大一的整數結果,我可以這樣寫裝飾器:

現在我們就可以用@符號和這個裝飾器去裝飾另外一個函式了:

裝飾器修改了存在的函式,將裝飾器返回的結果賦值給了變數。在這個例子中,’plus’的結果實際指向increment(plus)的結果。

對於非post請求返回錯誤

現在讓我們在一些更有用的場景下應用裝飾器。如果在django中接收的不是POST請求,我們用裝飾器返回一個錯誤響應。

現在我們可以在上述註冊api中應用這個裝飾器:

現在我們就有了一個可以在每個api方法中重用的裝飾器。

傳送json響應

為了傳送json響應(同時處理500狀態碼),我們可以新建另外一個裝飾器:

現在我們就可以在原方法中去除json相關的程式碼,新增一個裝飾器做為代替:

現在,如果我需要編寫新的方法,那麼我就可以使用裝飾器做冗餘的工作。如果我要寫登入方法,我只需要寫真正相關的程式碼:

BONUS: 引數化你的請求方法

我曾經使用過Tubogears框架,其中請求引數直接解釋轉遞給方法這一點我很喜歡。所以要怎樣在Django中模仿這一特性呢?嗯,裝飾器就是一種解決方案!

例如:

注意這是一個引數化裝飾器的例子。在這個例子中,函式的結果是實際的裝飾器。

現在我就可以用引數化裝飾器編寫方法了!我甚至可以選擇是否允許GET和POST,或者僅僅一種請求引數型別。

現在我們有了一個簡潔的、易於理解的api。

BONUS #2: 使用functools.wraps儲存docstrings和函式名

很不幸,使用裝飾器的一個副作用是沒有儲存方法名(__name__)和docstring(__doc__)值:

這將對使用反射的應用造成麻煩,比如Sphinx,一個 自動生成文件的應用

為了解決這個問題,我們可以使用’wraps’裝飾器附加上名字和docstring:

BONUS #3: 使用’decorator’裝飾器

如果仔細看看上述使用裝飾器的方式,在包裝器宣告和返回的地方也有不少重複。

你可以安裝python egg ‘decorator’,其中包含一個提供裝飾器模板的’decorator’裝飾器!

使用easy_install:

或者Pip:

然後你可以簡單的編寫:

這個裝飾器更牛逼的一點是儲存了__name__和__doc__的返回值,也就是它封裝了
functools.wraps的功能!

相關文章