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

发布于 2021-01-29 17:18:06

使用模型继承时,我试图找到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

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

关注者
0
被浏览
50
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    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()方法将需要对每个项目进行一次查询。

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



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看