快捷導(dǎo)航

python之multimethod包多分派解讀

更新時(shí)間：2023年08月14日 09:39:08 作者：Audreybiubiu

這篇文章主要介紹了python之multimethod包多分派問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

python multimethod包多分派

multimethod （又名multiple dispatch、generic function）是python提供的一個(gè)功能包，它可以實(shí)現(xiàn) singledispatch 類似的功能，而對(duì)于相同類型的返回不用多次注冊(cè)

使用方法

1、使用 @multimethod 標(biāo)記處理object類型的基函數(shù)(base_function)

2、各個(gè)專門函數(shù)使用 @<<base_function>>.register 裝飾

3、專門函數(shù)的名稱無(wú)關(guān)緊要，可以直接選擇用_作為函數(shù)名

4、為每個(gè)需要處理的類型注冊(cè)一個(gè)函數(shù)

例子

@multimethod
def inner_as_dict() -> object:
    """
    multimethod方法，對(duì)于不同類型的鍵，注冊(cè)不同的序列化為字典的方法
    """
    return None
@inner_as_dict.register
def _(element: List[TestData1]) -> [Dict]:
    result = []
    for i in element:
        tmp = asdict(i)
        tmp_list = [i.msg for i in tmp["code"]]
        tmp["msg"] = tmp_list if tmp_list else ["-"]
        result.append(tmp)
    return result
@inner_as_dict.register
def _(element: TestData2) -> List[Dict]:
    result: List[Dict] = []
    if not element.data:
        for message in element.message.splitlines():
            single_fail_res = {"item": message.split(",")[0], "result": False}
            result.append(single_fail_res)
    return result
@inner_as_dict.register
def _(element: TestData3) -> Tuple[int, str]:
    verify_result = 0 if element.result else -1
    verify_message = element.error.msg if element.error else "-"
    return dom_result, dom_message
@inner_as_dict.register
def _(element: Union[int, float, bool, str, List, Dict, Tuple]) -> Union[int, float, bool, str, List, Dict, Tuple]:
    return element

可以看到multimethod的使用方法與singledispatch基本一致，區(qū)別在于，multimethod在注冊(cè)函數(shù)的時(shí)候不需要寫明需要處理的類型，并且針對(duì)針對(duì)多種類型是相同的處理，可以不用寫多個(gè)注冊(cè)函數(shù)，更加簡(jiǎn)潔明了。

Python單分派泛型函數(shù)

概念

它允許您為一個(gè)函數(shù)提供多個(gè)實(shí)現(xiàn)，這些實(shí)現(xiàn)基于參數(shù)的類型。

實(shí)現(xiàn)

下面是一個(gè)使用functools.singledispatch的例子

from functools import singledispatch
@singledispatch
def add(x, y):
? ? print("Default implementation for integers")
? ? return x + y
@add.register(int)
def _(x, y):
? ? print("Implementation for integers")
? ? return x + y
@add.register(str)
def _(x, y):
? ? print("Implementation for strings")
? ? return x + y
print(add(1, 2)) # Output: Default implementation for integers -> 3
print(add("hello", "world")) # Output: Implementation for strings -> helloworld
print(add(1.5, 2.5)) # Output: Default implementation for integers -> 4
print(add("foo", "bar")) # Output: Default implementation for integers -> foobar

在這個(gè)例子中，我們定義了一個(gè)名為add的函數(shù)，并使用@singledispatch裝飾器將其轉(zhuǎn)換為單分派泛型函數(shù)。

我們?yōu)樵摵瘮?shù)提供了兩個(gè)實(shí)現(xiàn)：

一個(gè)默認(rèn)實(shí)現(xiàn)和兩個(gè)特定于整數(shù)和字符串類型的實(shí)現(xiàn)。

當(dāng)我們調(diào)用add函數(shù)時(shí)，它將根據(jù)參數(shù)的類型選擇適當(dāng)?shù)膶?shí)現(xiàn)。

如果參數(shù)是整數(shù)或浮點(diǎn)數(shù)，則將調(diào)用默認(rèn)實(shí)現(xiàn)；如果參數(shù)是字符串，則將調(diào)用特定于字符串類型的實(shí)現(xiàn)。