Python 超类反射

Question

如果我有一段Python代码

class A():
    pass
class B():
    pass
class C(A, B):
    pass

而且我有一个类 C，有没有办法遍历它的父类（A 和 B）呢？就像伪代码那样：

>>> magicGetSuperClasses(C)
(<type 'A'>, <type 'B'>)

一个解决方案似乎是使用 inspect模块 和 getclasstree 函数。

def magicGetSuperClasses(cls):
    return [o[0] for o in inspect.getclasstree([cls]) if type(o[0]) == type]

但是这样做算不算是“Python风格”的方法呢？

Answer 1

inspect模块是个不错的起点，可以使用getmro这个函数：

这个函数会返回一个元组，里面包含了类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，你可以直接忽略它。

Answer 2

@John：你的代码片段不太对——你返回的是基类的类（这些也叫做 metaclasses）。你其实只需要用 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 及以上版本，你可以使用 生成器表达式，这样就不用先创建一个列表（用 []），再把它转成元组（用 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__)

这个例子有点让人困惑，但生成器表达式通常很简单，也很酷。:)

Answer 3

C.__bases__ 是一个包含父类的数组，所以你可以像这样实现你想要的功能：

def magicGetSuperClasses(cls):
  return cls.__bases__

不过我觉得在大多数情况下，直接使用 cls.__bases__ 会更简单。