0
点赞
收藏
分享

微信扫一扫

python——singledispatch 单分派

鱼满舱 2022-03-26 阅读 83

介绍:
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方法,对于TaskResult内的不同类型的键,注册不同的序列化为字典的方法
    """
    return None


@inner_as_dict.register
def _(element) -> None
	return None


@inner_as_dict.register(ClickInfo)
def _(element: List[ClickInfo]) -> [Dict]:
    result = []
    for i in element:
        tmp = asdict(i)
        tmp_error_list = [i.msg for i in tmp["error"]]
        tmp["error"] = tmp_error_list if tmp_error_list else ["-"]
        result.append(tmp)
    return result


@inner_as_dict.register(CheckResult)
def _(element: CheckResult) -> List[Dict]:
    result: List[Dict] = []
    if not element.data:
        for message in element.message.splitlines():
            single_fail_res = {"itemType": message.split(",")[0], "result": False, "errorMessage": message}
            result.append(single_fail_res)
    return result


@inner_as_dict.register(DOMResult)
def _(element: DOMResult) -> Tuple[int, str]:
    dom_verify_result = 0 if element.result else -1
    dom_verify_message = element.error.msg if element.error else "-"
    return dom_verify_result, dom_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的时候,能够根据传入的参数类型进行判断,应该运行实际的哪个专门函数,从而实现返回不一样的结果。

举报

相关推荐

0 条评论