Python中是否有类似dict的多态类型?这就是多态性的意思:考虑一个具有Animal
基类和一些派生类(Cat
,Snake
等)的基本类层次结构,让我们拥有一个神秘的映射类型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
;这使您可以通过添加接口来完全做到这一点。它在很大程度上优化了查找映射,以将原理扩展到更大的对象映射注册表。