如果我有Python代码
class A():
pass
class B():
pass
class C(A, B):
pass
我有课C
,有没有办法迭代它的超级分类(A
和B
)?类似伪代码的东西:
>>> magicGetSuperClasses(C)
(<type 'A'>, <type 'B'>)
一个解决方案似乎是inspect module和getclasstree
功能。
def magicGetSuperClasses(cls):
return [o[0] for o in inspect.getclasstree([cls]) if type(o[0]) == type]
但这是实现目标的“蟒蛇”方式吗?
答案 0 :(得分:36)
C.__bases__
是一个超类的数组,所以你可以像这样实现你的假设函数:
def magicGetSuperClasses(cls):
return cls.__bases__
但我想在大多数情况下直接引用cls.__bases__
会更容易。
答案 1 :(得分:11)
@John:您的代码段不起作用 - 您将返回基类的类(也称为元类)。你真的只想要cls.__bases__
:
class A: pass
class B: pass
class C(A, B): pass
c = C() # Instance
assert C.__bases__ == (A, B) # Works
assert c.__class__.__bases__ == (A, B) # Works
def magicGetSuperClasses(clz):
return tuple([base.__class__ for base in clz.__bases__])
assert magicGetSuperClasses(C) == (A, B) # Fails
此外,如果您使用的是Python 2.4+,则可以使用generator expressions而不是创建列表(通过[]),然后将其转换为元组(通过tuple
)。例如:
def get_base_metaclasses(cls):
"""Returns the metaclass of all the base classes of cls."""
return tuple(base.__class__ for base in clz.__bases__)
这是一个有点令人困惑的例子,但是genexps通常很简单而且很酷。 :)
答案 2 :(得分:4)
检查模块是一个良好的开端,使用getmro功能:
以方法解析顺序返回类cls基类的元组,包括cls。在这个元组中没有类出现过多次。 ...
>>> class A: pass
>>> class B: pass
>>> class C(A, B): pass
>>> import inspect
>>> inspect.getmro(C)[1:]
(<class __main__.A at 0x8c59f2c>, <class __main__.B at 0x8c59f5c>)
返回元组的第一个元素是C
,你可以忽略它。
答案 3 :(得分:2)
如果你需要知道订购哪个super()会调用这些类,你可以使用C.__mro__
而不需要inspect
。