Django-MPTT叶子节点有多种类型

时间:2012-06-19 19:59:59

标签: django hierarchical-data django-mptt

我正在开发一个hirerchical应用程序,其中叶节点可以是不同模型的实例。我无法弄清楚如何使它与django-mptt应用程序一起工作。这个应用程序甚至可以实现吗?如果是的话,我做错了什么?如果没有,那么我有什么要做的吗?

以下是模型的基本结构:

class FolderItemBase(MPTTModel):
    order = models.PositiveIntegerField()

    class Meta:
        abstract = True

    class MPTTMeta:
        parent_attr = 'folder'
        order_insertion_by = ['order']

class Folder(FolderItemBase):
    folder = TreeForeignKey('Folder', related_name='folders', blank=True, null=True)
    ...

class Image(FolderItemBase):
    folder = TreeForeignKey('Gallery', related_name='images') # cannot be null since leaf has to be inside of a folder
    ...

当我尝试执行以下操作时,我只能获取“文件夹”子项,而无法获取任何图像。当我试图获得图像的祖先时,同样的事情

>>> folder1 = Folder.objects.create(title='Folder 1', order=0)
>>> folder2 = Folder(title='Folder 2', order=0)
>>> folder2.insert_at(folder1, save=True)
>>> image = Image(...)
>>> image.insert_at(folder1, save=True)
>>> folder1.get_children()
[<Folder: Folder 2>]
>>> image.get_ancestores()
[]

这就是在所有这些之后事物存储在数据库中的方式:

Folder table
----------------
+----+-------+-----+------+---------+-------+-----------+----------+
| ID | order | lft | rght | tree_id | level | folder_id | title    |
+----+-------+-----+------+---------+-------+-----------+----------+
| 1  | 0     | 1   | 4    | 1       | 0     |           | Folder 1 |
+----+-------+-----+------+---------+-------+-----------+----------+
| 2  | 0     | 2   | 3    | 1       | 1     | 1         | Folder 2 |
+----+-------+-----+------+---------+-------+-----------+----------+

Images Table
------------
+----+-------+-----+------+---------+-------+-----------+
| ID | order | lft | rght | tree_id | level | folder_id |
+----+-------+-----+------+---------+-------+-----------+
| 1  | 1     | 2   | 3    | 1       | 1     | 1         |
+----+-------+-----+------+---------+-------+-----------+

正如你所看到的,它会显示图像的级别编号和正确的(至少似乎是)左右数字,但是它不会更新文件夹表中的任何内容,因此当您尝试进行查询,没有任何内容被选中。

任何指针都表示赞赏。谢谢。

1 个答案:

答案 0 :(得分:6)

AFAIK,这是不可能的; django-mptt捎带着Django的QuerySet,它只能用于某种类型的东西。您可以使用contenttypes框架将“真实”项与FolderItem之类的内容相关联,这些内容仅用于层次结构,例如。

from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes import generic

class FolderItem(MPTTModel):
    folder = TreeForeignKey('Folder', related_name='folders', blank=True, null=True
    order = models.PositiveIntegerField()
    content_type = models.ForeignKey(ContentType)
    object_id = models.PositiveIntegerField()
    content_object = generic.GenericForeignKey('content_type', 'object_id')

    class Meta:
        abstract = True

    class MPTTMeta:
        parent_attr = 'folder'
        order_insertion_by = ['order']

然后,当您使用django-mptt管理器方法等时,您将获得FolderItem s的查询集,并且您可以访问Folder / Image当你通过通用外键遍历集合时,每个人都可以使用。

但是,请注意,这在数据库查询方面可能会很昂贵,因为每次访问通用外键时,都必须发出新的查询。