python singledispatch 使用简单说明

singledispatch 可以实现类似方法的范型能力，以下是使用的简单说明

方法

• 参考代码

from functools import singledispatch

@singledispatch

def add(a,b):

    return f"default---{a}-{b}"

@add.register

def _(a:int,b:int)->int:

    return a + b

@add.register

def _(a:str,b:str)->str:

    return f"{a} {b}---str"

@add.register(float)

def _(a:float,b:float)->float:

    return a + b + 100

result = add(1,3)

result2 = add("dalong","demoapap")

result3 = add(0.1,0.2)

(a,b) = ((1,3),(3,4))

result4 = add(a,b)

print(result)

print(result2)

print(result3)

print(result4)

类

对于类的使用，注意需要python 3.8 以及以上

• 参考代码

from functools import singledispatchmethod

class A:

    @singledispatchmethod

    def add(self,a,b):

        return f"default---{a}-{b}"

    @add.register

    def _(self,a:int,b:int)->int:

        return a + b + 100 

    @add.register

    def _(self,a:str,b:str)->str:

        return f"{a} {b}---str"

a = A()

result = a.add(1,3)

result2 = a.add("dalong","demoapap")

result3 = a.add(0.1,0.2)

print(result)

print(result2)

print(result3)

• 低版本的处理
  提供一个新的包装方法

from functools import singledispatch, wraps

def singledispatchmethod(func):

    dispatcher = singledispatch(func)

    @wraps(func)

    def wrapper(*args, **kwargs):

        return dispatcher.dispatch(args[1].__class__)(*args, **kwargs)

    wrapper.register = dispatcher.register

    return wrapper

class A:

    @singledispatchmethod

    def add(self,a,b):

        return f"default---{a}-{b}"

    @add.register

    def _(self,a:int,b:int)->int:

        return a + b + 100 

    @add.register

    def _(self,a:str,b:str)->str:

        return f"{a} {b}---str"

a = A()

result = a.add(1,3)

result2 = a.add("dalong","demoapap")

result3 = a.add(0.1,0.2)

print(result)

print(result2)

print(result3)

说明

python 的singledispatch 比较有意思，对于开发是一个很不错的小技巧,比如dbt 中就使用了dispatch 的概念处理macro（没有使用python的singledispatch）实现比较方便的处理调用不同adapter 自己的macro 方法

参考资料

https://peps.python.org/pep-0443/