構建健壯 Python 包的 5 個簡單規則

艾凌風發表於2015-06-08

建立一個軟體包(package)似乎已經足夠簡單了,也就是在檔案目錄下蒐集一些模組,再加上一個__init__.py檔案,對吧?我們很容易看出來,隨著時間的推移,通過對軟體包的越來越多的修改,一個設計很差的軟體包可能會出現迴圈依賴問題,或是可能變得不可移植和不可靠。

1. __init__.py 僅為匯入服務

對於一個簡單的軟體包,你可能會忍不住把工具方法,工廠方法和異常處理都丟進__init__.py,千萬別這樣!

一個結構良好的__init__.py檔案,僅為一個非常重要的目的來服務:從子模組匯入。你的__init__.py應該看起來像這個樣子:

2.使用__init__.py來限制匯入順序

  1. 把方法和類置於軟體包的作用域中,這樣使用者就不需要深入軟體包的內部結構,使你的軟包變得易用。
  2. 作為調和匯入順序的唯一地方。

使用得當的話,__init__.py 可以為你提供重新組織內部軟體包結構的靈活性,而不需要擔心由內部匯入子模組或是每個模組匯入順序所帶來的副作用。因為你是以一個特定的順序匯入子模組,你的__init__.py 對於他程式設計師來講應該簡單易懂,並且能夠明顯的表示該軟體包所能提供的全部功能。

文件字串,以及在軟體包層面對__all__屬性的賦值應當是__init__.py中唯一與匯入無關的程式碼:

3.使用一個模組來定義所有的異常

你也許已經注意到了,__init__.py中的第一個匯入語句從exceptions.py子模組中匯入了全部的異常。從這裡出發,你將看到,在大多數的軟體包中,異常被定義在引起它們的程式碼附近。儘管這樣可以為一個模組提供高度的完整性,一個足夠複雜的軟體包會通過如下兩種方式,使得這一模式出現問題。

  1. 通常一個模組/程式需要從一個子模組匯入一個函式, 利用它匯入程式碼並丟擲異常。為了捕獲異常並保持一定的粒度,你需要匯入你需要的模組,以及定義了異常的模組(或者更糟,你要匯入一系列的異常)。這一系列衍生出來的匯入需求,是在你的軟體包中編織一張錯綜複雜的匯入之網的始作俑者。你使用這種方式的次數越多,你的軟體包內部就變的越相互依賴,也更加容易出錯。
  2. 隨著異常數量的不斷增長,找到一個軟體包可能引發的全部異常變的越來越難。把所有的異常定義在一個單獨的模組中,提供了一個方便的地方,在這裡,程式設計師可以審查並確定你的軟體包所能引發全部潛在錯誤狀態。

你應該為你的軟體包的異常定義一個基類:

然後確保你的軟體包在任何錯誤狀態下,只會引發這個基類異常的子類異常,這樣如果你需要的話,你就可以阻止全部的異常:

對於一般的錯誤狀態,這裡有一些重要的異常處理已經被包括在標準庫中了(例如,TypeError, ValueError等)

靈活地定義異常處理並保持足夠的粒度:

在你的異常處理中保持更大的粒度,有利於讓程式設計師們在一個try/except中包含越來越大的,互相不干涉的程式碼段。

在異常定義時保持高度的粒度,會減少錯綜複雜的錯誤處理,並且允許你把正常執行指令和錯誤處理指令分別開來,使你的程式碼更加易懂和更易維護。

4. 在軟體包內部只進行相對匯入

在子模組中你時常見到的一個簡單錯誤,就是使用軟體包的名字來匯入軟體包。

這樣做會導致兩個不好的結果:

  1. 子模組只有當軟體包被安裝在 PYTHONPATH 內才能正確執行。
  2. 子模組只有當這個軟體包的名字是 a_package 時才能正確執行。

儘管第一條看上去並不是什麼大問題,但是考慮一下,如果你在 PYTHONPATH 下的兩個目錄中,有兩個同名的軟體包。你的子模組可能最終匯入了另一個軟體包,你將無意間使得某個或某些對此毫無戒備的程式設計師(或是你自己)debug 到深夜。

5. 讓模組保持較小的規模

你的模組應當比較小。記住,那個使用你軟體包的程式設計師會在軟體包作用域進行匯入,同時你會使用你的 __init__.py 檔案來作為一個組織工具,來暴露一個完整的介面。

好的做法是一個模組只定義一個類,伴隨一些幫助方法和工廠方法來協助建立這個模組。

如果你的模組暴露了一些方法,把一些相互依賴的方法分為一組放進一個模組,並且把不相互依賴的方法移動到單獨的模組中:

上面的例子是 fsq/enqueue.py,它暴露了一系列的方法來為同一個功能提供不同的介面(就像 simplejson 中的l oad/loads)。儘管這個例子足夠直觀,讓你的模組保持較小規模需要一些判斷,但是一個好的原則是:

當你有疑問的時候,就去建立一個新的子模組吧。

打賞支援我翻譯更多好文章,謝謝!

打賞譯者

打賞支援我翻譯更多好文章,謝謝!

任選一種支付方式

構建健壯 Python 包的 5 個簡單規則 構建健壯 Python 包的 5 個簡單規則

相關文章