如何找到Django模型基类的“具体类”
使用模型继承时,我试图找到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
但这感觉很脆弱,我不确定如果继承更多级别会发生什么。
-
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()方法将需要对每个项目进行一次查询。
相反,如果您拥有一组已知的子模型,只需简单地分别查询每个模型并将实例聚合到一个列表中。