如何解决Python中multimethod匹配自定义类参数失效问题

问题重现 #

定义类：

class A:
    pass

class B(A):
    pass

class C(A):
    pass

需求：实现click方法，第一个参数x始终为字符串，第二个参数y为A/B/C的实例，根据y的类型执行不同逻辑。

尝试的多方法代码（触发异常）：

@multimethod
def click(x: str, y: A=None):
    do_A_thing()

@multimethod
def click(x: str, y: B=None):
    do_B_thing()

@multimethod
def click(x: str, y: object=None):
    raise Exception

调用时，即使y是A/B类型，也总是触发最后一个抛出异常的分支。同时尝试过singledispatch（无法针对第二个参数分发）、multipledispatch（效果类似）都没解决。

问题原因 #

给参数设置默认值None后，类型注解会被解析为Union[指定类型, NoneType]，而object是所有类型的父类（包括NoneType），导致分发逻辑优先匹配最宽泛的object分支，忽略了更具体的A/B类型。

解决方案 #

方案1：使用multipledispatch（修正写法） #

移除参数默认值，明确指定类型，让分发逻辑按继承关系匹配最具体的类型：

from multipledispatch import dispatch

class A:
    pass

class B(A):
    pass

class C(A):
    pass

def do_A_thing():
    print("执行A相关操作")

def do_B_thing():
    print("执行B相关操作")

@dispatch(str, A)
def click(x, y):
    do_A_thing()

@dispatch(str, B)
def click(x, y):
    do_B_thing()

@dispatch(str, object)
def click(x, y):
    raise ValueError("不支持的y类型")

# 测试
click("测试", A())  # 输出：执行A相关操作
click("测试", B())  # 输出：执行B相关操作
click("测试", C())  # 匹配A的分支，输出：执行A相关操作（若需单独处理C，新增@dispatch(str, C)分支即可）

方案2：基于singledispatch包装实现 #

利用包装函数调换参数顺序，适配singledispatch只针对第一个参数分发的特性：

from functools import singledispatch

class A:
    pass

class B(A):
    pass

class C(A):
    pass

def do_A_thing():
    print("执行A相关操作")

def do_B_thing():
    print("执行B相关操作")

# 对外暴露的方法，保持参数顺序
def click(x: str, y):
    _click(y, x)

# 内部分发函数，以y作为第一个参数
@singledispatch
def _click(y, x):
    raise ValueError("不支持的y类型")

@_click.register(A)
def _(y, x):
    do_A_thing()

@_click.register(B)
def _(y, x):
    do_B_thing()

# 测试
click("测试", A())  # 输出：执行A相关操作
click("测试", B())  # 输出：执行B相关操作

内容的提问来源于stack exchange，提问作者KSS