从父类获取所有子模型-Django


问题内容

我如何从家长班获得所有孩子的模特儿?例如,我有:

class Device(PolymorphicModel):
   .....

class Mobile(Device): 
   .....

class Computer(Device):
   .....

因此,我想从Device模型中获取所有后代:移动设备,计算机作为类而不是实例。

谢谢。


问题答案:

您可以使用[Python-
doc]
方法获取
直接 子类:
class.__subclasses__()

>>> Device.__subclasses__()
[<class 'Mobile'>, <class 'Computer'>]

但是,这些也可能具有子类。我们可以开发一种算法,每次获得下一代算法,并一直这样做直到找到新的子类为止,例如:

def get_descendants(klass):
    gen = { klass }
    desc = set()
    while gen:
        gen = { skls for kls in gen for skls in kls.__subclasses__() }
        desc.update(gen)
    return desc

或具有可变数量的参数:

def get_descendants(*klass):
    gen = { *klass }
    desc = set()
    while gen:
        gen = { skls for kls in gen for skls in kls.__subclasses__() }
        desc.update(gen)
    return desc

这将返回一个set()包含所有后代(直接和间接)的。