多态映射类型 - python

Python中是否有类似dict的多态类型?这就是多态性的意思:考虑一个具有Animal基类和一些派生类(CatSnake等)的基本类层次结构,让我们拥有一个神秘的映射类型mystery_dict

mapping = mystery_dict({
    Animal : 'foo',
    Cat    : 'bar',
    Snake  : 'baz',
    Python : 'eggs',
    Boa    : 'spam'
})

现在,我要使以下几行正确:

mapping[Animal]   == 'foo'
mapping[Cat]      == 'bar'
mapping[Dog]      == 'foo' # No Dog in mapping, take the base class Animal
mapping[Snake]    == 'baz'
mapping[Boa]      == 'spam'
mapping[Anaconda] == 'baz' # No Anaconda in mapping, take the base class Snake

我知道自Python 3.4起,我可以在isinstance中使用一堆functools.singledispatch或“重载集”,但是在某些情况下,多态字典类型将很容易减少样板。这样的类型是否存在于野外还是我必须创建一个?当然,如果您有更好的选择,我将很高兴听到它。

注意:以防万一出现问题,我的需求很简单,因此不必处理多个继承。

python大神给出的解决方案

这足以让你自己写:

from collections.abc import MutableMapping


class PolymorphicDict(MutableMapping):
    def __init__(self, *args, **kwargs):
        self._mapping = dict(*args, **kwargs)

    def __getitem__(self, key):
        for cls in key.__mro__:
            if cls in self._mapping:
                return self._mapping[cls]
        raise KeyError(key)

    def __delitem__(self, key):
        del self._mapping[key]

    def __setitem__(self, key, value):
        self._mapping[key] = value

    def __iter__(self):
        return iter(self._mapping)

    def __len__(self):
        return len(self._mapping)

它使用class.__mro__ attribute以方法查找顺序(“方法解析顺序”)列出当前对象的类层次结构。此序列包括当前类,并列出直到object的所有类。

演示:

>>> class Animal: pass
... 
>>> class Cat(Animal): pass
... 
>>> class Dog(Animal): pass
... 
>>> class Snake(Animal): pass
... 
>>> class Python(Snake): pass
... 
>>> class Boa(Snake): pass
... 
>>> class Anaconda(Snake): pass
... 
>>> Anaconda.__mro__
(<class '__main__.Anaconda'>, <class '__main__.Snake'>, <class '__main__.Animal'>, <class 'object'>)
>>> mapping = PolymorphicDict({
...     Animal : 'foo',
...     Cat    : 'bar',
...     Snake  : 'baz',
...     Python : 'eggs',
...     Boa    : 'spam'
... })
>>> mapping[Animal]
'foo'
>>> mapping[Cat]
'bar'
>>> mapping[Dog]
'foo'
>>> mapping[Snake]
'baz'
>>> mapping[Boa]
'spam'
>>> mapping[Anaconda]
'baz'

您可能想看看zope.component;这使您可以通过添加接口来完全做到这一点。它在很大程度上优化了查找映射,以将原理扩展到更大的对象映射注册表。