快捷導(dǎo)航

python之singledispatch單分派問題

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

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

python singledispatch單分派

singledispathch 是Pyhton 在functools里的方法，用作裝飾器，它可以把整體方案拆成多個(gè)模塊，甚至可以為你無法修改的類提供專門的函數(shù)，使用@singledispatch裝飾的函數(shù)會(huì)變成泛函數(shù)。

好處

類似于java的重載機(jī)制，可以在一個(gè)類中為同一個(gè)方法定義多個(gè)重載變體，比在一個(gè)函數(shù)中使用一長串的if/elif好

使用方法

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

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

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

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

5、可以疊放多個(gè)register裝飾器，讓同一個(gè)函數(shù)支持不同類型

例子

@singledispatch
def inner_as_dict() -> object:
? ? """
? ? singledispatch方法，對(duì)于不同類型的鍵，注冊不同的序列化為字典的方法
? ? """
? ? return None
@inner_as_dict.register
def _(element) -> None
?? ?return None
@inner_as_dict.register(TestData1)
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(TestData2)
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(TestData3)
def _(element: TestData3) -> Tuple[int, str]:
? ? verify_result = 0 if element.result else -1
? ? verify_message = element.error.msg if element.error else "-"
? ? return verify_result, verify_message
@inner_as_dict.register(int)
@inner_as_dict.register(float)
@inner_as_dict.register(bool)
@inner_as_dict.register(str)
@inner_as_dict.register(list)
@inner_as_dict.register(dict)
@inner_as_dict.register(tuple)
def _(element: Union[int, float, bool, str, List, Dict, Tuple]) -> Union[int, float, bool, str, List, Dict, Tuple]:
? ? return element

以上代碼能實(shí)現(xiàn)，在運(yùn)行inner_as_dict的時(shí)候，能夠根據(jù)傳入的參數(shù)類型進(jìn)行判斷，應(yīng)該運(yùn)行實(shí)際的哪個(gè)專門函數(shù)，從而實(shí)現(xiàn)返回不一樣的結(jié)果。

python實(shí)現(xiàn)單分派泛型函數(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)：