删除某些元素时模型继承中的问题

时间:2009-04-04 12:06:38

标签: django inheritance model

我在http://www.djangosnippets.org/snippets/1034/中使用了一个片段作为我的模型继承。它在第一次工作正常。但是,在删除数据库中的某些元素后,代码运行错误。 在调试时,我发现问题出现在方法中:as_leaf_class。 在以下代码中:

if (model == Meal):
    return self
return model.objects.get(id=self.id)

删除元素时,最后一行会引发异常。 任何人都可以为此提供解决方案吗?

具有内容类型和继承感知管理器的模型继承

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.db.models.query import QuerySet

class SubclassingQuerySet(QuerySet):
    def __getitem__(self, k):
        result = super(SubclassingQuerySet, self).__getitem__(k)
        if isinstance(result, models.Model) :
            return result.as_leaf_class()
        else :
            return result
    def __iter__(self):
        for item in super(SubclassingQuerySet, self).__iter__():
            yield item.as_leaf_class()

class MealManager(models.Manager):
    def get_query_set(self):
        return SubclassingQuerySet(self.model)

class Meal (models.Model) :
    name = models.TextField(max_length=100)
    content_type = models.ForeignKey(ContentType,editable=False,null=True)
    objects = MealManager()

    def save(self, *args, **kwargs):
        if(not self.content_type):
            self.content_type = ContentType.objects.get_for_model(self.__class__)
            super(Meal, self).save(*args, **kwargs)

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

class Salad (Meal) :
    too_leafy = models.BooleanField(default=False)
    objects = MealManager()

2 个答案:

答案 0 :(得分:0)

我不知道您现在可以使用abstract base classes.

,该代码段是否仍然相关

这允许您声明一个不是db表但其他模型可以继承的模型。

答案 1 :(得分:0)

第一个答案:为什么要尝试在已删除的对象上调用as_leaf_class?如果你这样做会伤害,不要这样做。

第二个答案是,您可以使用try ...包装失败的行...除了Meal.DoesNotExist,并返回None或self或者其他内容。