python中的动态调度和继承

时间:2011-01-09 19:38:06

标签: python dynamic-dispatch

我正在尝试修改Guido的多方法(动态调度代码):

http://www.artima.com/weblogs/viewpost.jsp?thread=101605

处理继承和可能的乱序参数。

e.g。 (继承问题)

class A(object):
  pass

class B(A):
  pass

@multimethod(A,A)
def foo(arg1,arg2):
  print 'works'


foo(A(),A()) #works

foo(A(),B()) #fails

有没有比迭代检查每个项目的super()更好的方法直到找到一个?

e.g。 (论证排序问题) 从碰撞检测的角度来看,我正在考虑这个问题。

e.g。

foo(Car(),Truck()) and
foo(Truck(), Car()) and

应该触发

foo(Car,Truck) # Note: @multimethod(Truck,Car) will throw an exception if @multimethod(Car,Truck) was registered first?

我正在寻找一个'优雅'的解决方案。我知道我可以通过所有可能性来强行实施,但我试图避免这种情况。我只想在坐下来寻找解决方案之前得到一些意见和建议。

由于

2 个答案:

答案 0 :(得分:2)

super()返回一个代理对象,而不是父类(因为你可以有多个继承),所以这不起作用。使用isinstance()是最好的选择,尽管没有办法让它像使用type(arg)的字典查找一样优雅。

我不认为允许替代参数排序是一个好主意;它容易导致令人讨厌的惊喜,并使其与继承兼容也将是一个令人头痛的问题。但是,如果所有参数都是A类型,则使用“使用此函数”或者“如果所有参数都在类型{A,B,E}中则使用此函数”将非常简单。

答案 1 :(得分:2)

关于继承问题:这可以通过稍微更改MultiMethod来完成。 (通过self.typemap迭代并使用issubclass进行检查):

registry = {}

class MultiMethod(object):
    def __init__(self, name):
        self.name = name
        self.typemap = {}
    def __call__(self, *args):
        types = tuple(arg.__class__ for arg in args) # a generator expression!
        for typemap_types in self.typemap:
            if all(issubclass(arg_type,known_type)
                   for arg_type,known_type in zip(types,typemap_types)):
                function = self.typemap.get(typemap_types)
                return function(*args)
        raise TypeError("no match")
    def register(self, types, function):
        if types in self.typemap:
            raise TypeError("duplicate registration")
        self.typemap[types] = function

def multimethod(*types):
    def register(function):
        name = function.__name__
        mm = registry.get(name)
        if mm is None:
            mm = registry[name] = MultiMethod(name)
        mm.register(types, function)
        return mm
    return register

class A(object):
  pass

class B(A):
    pass

class C(object):
    pass

@multimethod(A,A)
def foo(arg1,arg2):
  print 'works'


foo(A(),A()) #works

foo(A(),B()) #works

foo(C(),B()) #raises TypeError

请注意self.typemap是一个字典,而且字符串是无序的。因此,如果使用@multimethod来注册两个函数,一个类型是另一个函数的子类,那么foo的行为可能是未定义的。也就是说,结果将取决于typemap_types在循环for typemap_types in self.typemap中首先出现的{{1}}。