使用 singledispatch 在 Python 中追溯地新增方法 | Linux 中國

Python 是當今使用最多流行的程式語言之一，因為：它是開源的，它具有廣泛的用途（例如 Web 程式設計、業務應用、遊戲、科學程式設計等等），它有一個充滿活力和專註的社群支援它。這個社群是我們在 Python Package Index（PyPI）中提供如此龐大、多樣化的軟體包的原因，用以擴充套件和改進 Python。並解決不可避免的問題。

在本系列中，我們將介紹七個可以幫助你解決常見 Python 問題的 PyPI 庫。今天，我們將研究 singledispatch，這是一個能讓你追溯地向 Python 庫新增方法的庫。

singledispatch

想象一下，你有一個有 Circle、Square 等類的“形狀”庫。

Circle 類有半徑、Square 有邊、Rectangle 有高和寬。我們的庫已經存在，我們不想改變它。

然而，我們想給庫新增一個面積計算。如果我們不會和其他人共享這個庫，我們只需新增 area方法，這樣我們就能呼叫 shape.area() 而無需關心是什麼形狀。

雖然可以進入類並新增一個方法，但這是一個壞主意：沒有人希望他們的類會被新增新的方法，程式會因奇怪的方式出錯。

相反，functools 中的 singledispatch 函式可以幫助我們。

1. @singledispatch
2. def get_area(shape):
3. raise NotImplementedError("cannot calculate area for unknown shape",
4. shape)

get_area 函式的“基類”實現會報錯。這保證瞭如果我們出現一個新的形狀時，我們會明確地報錯而不是傳回一個無意義的結果。

1. @get_area.register(Square)
2. def _get_area_square(shape):
3. return shape.side ** 2
4. @get_area.register(Circle)
5. def _get_area_circle(shape):
6. return math.pi * (shape.radius ** 2)

這種方式的好處是如果某人寫了一個匹配我們程式碼的新形狀，它們可以自己實現 get_area。

1. from area_calculator import get_area
3. @attr.s(auto_attribs=True, frozen=True)
4. class Ellipse:
5. horizontal_axis: float
6. vertical_axis: float
8. @get_area.register(Ellipse)
9. def _get_area_ellipse(shape):
10. return math.pi * shape.horizontal_axis * shape.vertical_axis

呼叫 get_area 很直接。

1. print(get_area(shape))

這意味著我們可以將大量的 if isintance()/elif isinstance() 的程式碼以這種方式修改，而無需修改介面。下一次你要修改 if isinstance，你試試 `singledispatch！

在本系列的下一篇文章中，我們將介紹 tox，一個用於自動化 Python 程式碼測試的工具。

回顧本系列的前幾篇文章：