如何在Python中对类列表进行排序,以确保任何子类在列表中的任何父类之前?
我问,因为我想知道一个对象属于类型列表中的哪个类型,但如果它属于多个类型,则查找最具体的类型
按len(cls.mro())
简单排序
如果C2
是C1
的子类,则必须保持len(C1.mro()) < len(C2.mro())
(因为C1.mro()
中的每个类也必须出现在C2.mro()
中)。因此,您可以简单地按mro列表的长度排序:
class A(object): pass
class X(object): pass
class B(A, X): pass
class C(B): pass
class D(C): pass
sorted([B, C, A, D, X], key = lambda cls: len(cls.mro()))
=> [__main__.A, __main__.X, __main__.B, __main__.C, __main__.D]
若要支持旧式类,可以将cls.mro()
替换为inspect.getmro(cls)
。
更好的方法是使用一个集合,然后沿着目标类的继承树移动。
def find_parent(target, class_set):
if target in class_set:
return [target]
else:
bases = []
for parent in target.__bases__:
result = find_parent(parent, class_set)
if result is not None:
bases.extend(result)
return bases
的例子:
class A(object):
pass
class B(str):
pass
class C(A):
pass
class D(A, dict):
pass
class_set = {dict, A}
print find_parent(A, class_set)
print find_parent(B, class_set)
print find_parent(C, class_set)
print find_parent(D, class_set)
给出如下结果:
[<class '__main__.A'>]
[]
[<class '__main__.A'>]
[<class '__main__.A'>, <type 'dict'>]