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 方法