从`django-mptt`使用`get_ancestors`函数时出现错误的结果

时间:2012-09-29 16:14:00

标签: python django tree hierarchy django-mptt

我正在开发一个使用django-mptt的项目,但是当我使用get_ancestors函数时,我得到了奇怪的结果。这是一个例子 我创建了一个简单的模型,继承自MPTTModel:

class Classifier(MPTTModel):
    title = models.CharField(max_length=255)
    parent = TreeForeignKey('self', null = True, blank = True, 
                            related_name = 'children')

    def __unicode__(self):
        return self.title

以下是适用于此模型的功能:

def test_mptt(self):
    # Erase all data from table
    Classifier.objects.all().delete()

    # Create a tree root
    root, created = Classifier.objects.get_or_create(title=u'root', parent=None)

    # Create 'a' and 'b' nodes whose parent is 'root'
    a = Classifier(title = "a")
    a.insert_at(root, save = True)
    b = Classifier(title = "b")
    b.insert_at(root, save = True)

    # Create 'aa' and 'bb' nodes whose parents are
    # 'a' and 'b' respectively
    aa = Classifier(title = "aa")
    aa.insert_at(a, save = True)
    bb = Classifier(title = "bb")
    bb.insert_at(b, save = True)

    # Create two more nodes whose parents are 'aa' and 'bb' respectively
    aaa = Classifier(title = "aaa")
    aaa.insert_at(aa, save = True)
    bba = Classifier(title = "bbb")
    bba.insert_at(bb, save = True)

    # Select from table just created nodes
    first = Classifier.objects.get(title = "aaa")
    second = Classifier.objects.get(title = "bbb")

    # Print lists of selected nodes' ancestors:
    print first.get_ancestors(ascending=True, include_self=True)
    print second.get_ancestors(ascending=True, include_self=True)

我希望在输出中看到下一个值:

[<Classifier: aaa>, <Classifier: aa>, <Classifier: a>, <Classifier: root>]
[<Classifier: bbb>, <Classifier: bb>, <Classifier: b>, <Classifier: root>]

但是我看到了:

[<Classifier: aaa>, <Classifier: bb>, <Classifier: b>, <Classifier: root>]
[<Classifier: bbb>, <Classifier: bb>, <Classifier: b>, <Classifier: root>]

因此,当您看到此函数为bbb节点打印正确的祖先列表时,会为aaa节点打错错误的祖先。你能解释一下为什么会这样吗?这是django-mptt中的错误还是我的代码不正确?

提前致谢。

2 个答案:

答案 0 :(得分:4)

将节点插入树中会导致整个树发生更改。因此,当您插入b节点时,您的aroot节点会在数据库中更改,但您的变量不会更新并保持包含旧的左/右值,这些值用于建立正确的树结构。

在您的情况下,当行aa.insert_at(a, save = True)位于proccess中时,您的a变量包含lft = 2且rght = 3的旧实例,而在数据库{ {1}}节点包含a = 4和lft = 5。

在插入新项目之前,您需要获取父项的新实例。最简单的方法是运行rght

refresh_from_db

答案 1 :(得分:0)

同意谢尔盖的回答。我用来重建树的命令是:

ModelName.objects.rebuild()   

(在'python manage.py shell'中执行)