在我們覆蓋 7 個 PyPI 庫的系列文章中了解更多解決 Python 問題的資訊。
Python 是當今使用最多流行的程式語言之一,因為:它是開源的,它具有廣泛的用途(例如 Web 程式設計、業務應用、遊戲、科學程式設計等等),它有一個充滿活力和專注的社群支援它。這個社群是我們在 Python Package Index(PyPI)中提供如此龐大、多樣化的軟體包的原因,用以擴充套件和改進 Python。並解決不可避免的問題。
在本系列中,我們將介紹七個可以幫助你解決常見 Python 問題的 PyPI 庫。今天,我們將研究 singledispatch,這是一個能讓你追溯地向 Python 庫新增方法的庫。
想象一下,你有一個有 Circle、Square 等類的“形狀”庫。
Circle 類有半徑、Square 有邊、Rectangle 有高和寬。我們的庫已經存在,我們不想改變它。
然而,我們想給庫新增一個面積計算。如果我們不會和其他人共用這個庫,我們只需新增 area
方法,這樣我們就能呼叫 shape.area()
而無需關心是什麼形狀。
雖然可以進入類並新增一個方法,但這是一個壞主意:沒有人希望他們的類會被新增新的方法,程式會因奇怪的方式出錯。
相反,functools 中的 singledispatch
函數可以幫助我們。
@singledispatchdef get_area(shape): raise NotImplementedError("cannot calculate area for unknown shape", shape)
get_area
函數的“基礎類別”實現會報錯。這保證了如果我們出現一個新的形狀時,我們會明確地報錯而不是返回一個無意義的結果。
@get_area.register(Square)def _get_area_square(shape): return shape.side ** 2@get_area.register(Circle)def _get_area_circle(shape): return math.pi * (shape.radius ** 2)
這種方式的好處是如果某人寫了一個匹配我們程式碼的新形狀,它們可以自己實現 get_area
。
from area_calculator import [email protected](auto_attribs=True, frozen=True)class Ellipse: horizontal_axis: float vertical_axis: float@get_area.register(Ellipse)def _get_area_ellipse(shape): return math.pi * shape.horizontal_axis * shape.vertical_axis
呼叫 get_area
很直接。
print(get_area(shape))
這意味著我們可以將大量的 if isintance()
/elif isinstance()
的程式碼以這種方式修改,而無需修改介面。下一次你要修改 if isinstance,你試試 `singledispatch!
在本系列的下一篇文章中,我們將介紹 tox,一個用於自動化 Python 程式碼測試的工具。