如何找到Django模型基类的“具体类”

小编典典

如何找到Django模型基类的“具体类”

python

使用模型继承时，我试图找到Django模型对象的实际类。

一些代码来描述问题：

class Base(models.model):
    def basemethod(self):
        ...

class Child_1(Base):
    pass

class Child_2(Base):
    pass

如果我创建两个Child类的各种对象，并创建一个包含所有对象的查询集：

Child_1().save()
Child_2().save()
(o1, o2) = Base.objects.all()

我想确定对象在基本方法中是类型为Child_1还是Child_2，我可以通过o1.child_1和o2.child_2到达子对象，但这可以重新获得有关基类中子类的知识。

我想出了以下代码：

def concrete_instance(self):
    instance = None
    for subclass in self._meta.get_all_related_objects():
        acc_name = subclass.get_accessor_name()
        try:
            instance = self.__getattribute__(acc_name)
            return instance
        except Exception, e:
            pass

但这感觉很脆弱，我不确定如果继承更多级别会发生什么。

阅读 214

2020-12-20

共1个答案

小编典典

Django通过父模型表与子模型表之间的OneToOneField实现模型继承。当您这样做时Base.object.all()，Django仅查询基本表，因此无法知道子表是什么。因此，不幸的是，没有其他查询就不可能直接进入子模型实例。

此代码段显示了将ContentType字段添加到基本模型的常用方法：

from django.contrib.contenttypes.models import ContentType

class Base(models.Model):
    content_type = models.ForeignKey(ContentType,editable=False,null=True)

    def save(self):
        if(not self.content_type):
            self.content_type = ContentType.objects.get_for_model(self.__class__)
        self.save_base()

    def as_leaf_class(self):
        content_type = self.content_type
        model = content_type.model_class()
        if(model == Base):
            return self
        return model.objects.get(id=self.id)

然后可以说if Base.content_type.model_class()确定类型。

这是另一个将自定义管理器添加到混合中的代码段。

如您所见，这两种解决方案都可能非常昂贵。如果您有大量实例，则使用as_leaf_class（）方法将需要对每个项目进行一次查询。

相反，如果您拥有一组已知的子模型，只需简单地分别查询每个模型并将实例聚合到一个列表中。

2020-12-20