python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python singledispatch单分派

python之singledispatch单分派问题

作者:Audreybiubiu

这篇文章主要介绍了python之singledispatch单分派问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

python singledispatch单分派

singledispathch 是Pyhton 在functools里的方法,用作装饰器,它可以把整体方案拆成多个模块,甚至可以为你无法修改的类提供专门的函数,使用@singledispatch装饰的函数会变成泛函数。

好处

类似于java的重载机制,可以在一个类中为同一个方法定义多个重载变体,比在一个函数中使用一长串的if/elif好

使用方法

1、使用@singledispatch标记处理object类型的基函数(base_function)

2、各个专门函数使用@<<base_function>>.register(<<type>>)装饰

3、专门函数的名称无关紧要,可以直接选择用_作为函数名

4、为每个需要处理的类型注册一个函数

5、可以叠放多个register装饰器,让同一个函数支持不同类型

例子

@singledispatch
def inner_as_dict() -> object:
    """
    singledispatch方法,对于不同类型的键,注册不同的序列化为字典的方法
    """
    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

以上代码能实现,在运行inner_as_dict的时候,能够根据传入的参数类型进行判断,应该运行实际的哪个专门函数,从而实现返回不一样的结果。

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装饰器将其转换为单分派泛型函数。

我们为该函数提供了两个实现:

如果参数是整数或浮点数,则将调用默认实现;如果参数是字符串,则将调用特定于字符串类型的实现。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文