我正在尝试使用模型继承时找到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()
我想确定对象在basemethod中是否为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只查询Base表,因此无法知道子表是什么.因此,遗憾的是,如果没有其他查询,则无法直接转到子模型实例.
此代码段显示了将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()方法将需要对每个项目进行一次查询.
相反,如果您有一组已知的子模型,只需单独查询每个模型并将实例聚合到一个列表中.
看看django-model-utils中的InheritanceManager - 将它附加到模型会给你具体的子类(至少在第一级):
from model_utils.managers import InheritanceManager class Base(models.Model): objects = InheritanceManager() # ... Base.objects.all().select_subclasses() # returns instances of child classes
model-utils需要Django 1.2或更高版本.