python之multimethod包多分派解读
作者:Audreybiubiu
这篇文章主要介绍了python之multimethod包多分派问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
python multimethod包多分派
multimethod (又名multiple dispatch、generic function)是python提供的一个功能包,它可以实现 singledispatch 类似的功能,而对于相同类型的返回不用多次注册
使用方法
1、使用 @multimethod 标记处理object类型的基函数(base_function)
2、各个专门函数使用 @<<base_function>>.register 装饰
3、专门函数的名称无关紧要,可以直接选择用_作为函数名
4、为每个需要处理的类型注册一个函数
例子
@multimethod
def inner_as_dict() -> object:
"""
multimethod方法,对于不同类型的键,注册不同的序列化为字典的方法
"""
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基本一致,区别在于,multimethod在注册函数的时候不需要写明需要处理的类型,并且针对针对多种类型是相同的处理,可以不用写多个注册函数,更加简洁明了。
Python单分派泛型函数
概念
它允许您为一个函数提供多个实现,这些实现基于参数的类型。
实现
下面是一个使用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在这个例子中,我们定义了一个名为add的函数,并使用@singledispatch装饰器将其转换为单分派泛型函数。
我们为该函数提供了两个实现:
一个默认实现和两个特定于整数和字符串类型的实现。
当我们调用add函数时,它将根据参数的类型选择适当的实现。
如果参数是整数或浮点数,则将调用默认实现;如果参数是字符串,则将调用特定于字符串类型的实现。
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
