Linux中國

使用 singledispatch 在 Python 中追溯地添加方法

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

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

singledispatch

想像一下,你有一個有 Circle、Square 等類的「形狀」庫。

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

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

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

相反,functools 中的 singledispatch 函數可以幫助我們。

@singledispatch
def 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 get_area

@attr.s(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 代碼測試的工具。

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

via: https://opensource.com/article/19/5/python-singledispatch

作者:Moshe Zadka 選題:lujun9972 譯者:geekpi 校對:wxy

本文由 LCTT 原創編譯,Linux中國 榮譽推出


本文轉載來自 Linux 中國: https://github.com/Linux-CN/archive

對這篇文章感覺如何?

太棒了
0
不錯
0
愛死了
0
不太好
0
感覺很糟
0
雨落清風。心向陽

    You may also like

    Leave a reply

    您的電子郵箱地址不會被公開。 必填項已用 * 標註

    此站點使用Akismet來減少垃圾評論。了解我們如何處理您的評論數據

    More in:Linux中國