限制 Django 模型中的递归外键深度

Question

我有一个 Django 模型，它带有一个可以为“自我”的外键。我有兴趣在递归深度上引入一个硬限制（比如 10）。哪里是检查这个的合适的地方，我应该在那里抛出什么样的异常？

我的意思的伪示例：

def limit_recursion_depth(self):
    depth = 0
    model = self
    while model.parent is not None:
        depth += 1
        model = model.parent
    if depth > 10:
        pass # Complain here and stop processing
    else:
        pass # Return the model, save, whatever is normal

我对覆盖模型的 save() 方法之类的解决方案更感兴趣，而不是仅适用于管理员的任何类型的验证。 （例如，我希望能够验证 Model.objects.create()。）

更新

尽管这个问题主要是学术性的，并且与我不再从事的项目有关，但我想更新这个问题，以防有人通过 Google 发现这个问题。以下来自django-mptt documentation：

要获取所有这些页面标题，您可以执行以下操作：

titles = []
while page:
    titles.append(page.title)
    page = page.parent

每个页面都有一个数据库查询...

即使对于小型项目，Django-mptt 的效率也要高得多，即使在这种情况下，我也应该使用它。

Answer 1

也许是这样的？

def save(self, *args, **kwargs):
    if Foo.objects.filter(foo=self.foo).count() > 10:
        raise Exception("not more than 10")
    else:
        super(Foo, self).save(*args, **kwargs)

更新：

对于带有django-mptt 的self 引用字段，您可以执行以下操作：

def save(self, *args, **kwargs):
    if Foo.objects.filter(foo=self.foo).get_descendant_count() > 10:
        raise Exception("not more than 10")
    else:
        super(Foo, self).save(*args, **kwargs)

Answer 2

无循环版本：

    def clean_parent(self):
        parent = self.cleaned_data['parent']
        if parent and parent.level >= 10:
            raise ValidationError('Recursion Depth Exceeded')
        return parent

注意：levels 从 0 开始。

Answer 3

不幸的是，正如this answer 中很好解释的那样，在 Django 中确实没有一个很好的方法来做到这一点。例如，每次最终用户点击更改表单上的保存按钮时，上述方法只会让管理员崩溃，并显示 500 错误——几乎没有帮助。唯一正确的方法是define clean_{fieldname} and throw a ValidationError。例如，它不会验证来自manage.py shell 的model.save()，但也不会覆盖该验证，例如queryset.update()。

因此，我现在采用的解决方案如下所示：

def clean_parent(self):
    depth = 0
    model = self.instance
    while model.parent is not None:
        depth += 1
        model = model.parent
    if depth > 10:
        raise ValidationError('Recursion Depth Exceeded')

请注意，这属于模型的管理表单。