Django-mptt 排序

Question

在我的项目中，我使用了django-mptt来处理分类。

我的模型是：

class Category(models.model):
    name = models.CharField()
    parent = models.ForeignKey("self", blank=True, null=True,
                           related_name="sub_category")
    nav_order = models.IntegerField(null=False, blank=False, default=0)
    # unsure need nav_order column in DB

    class Meta:
        verbose_name_plural = 'Categories'
mptt.register(Category)

我需要能够获取当前分类的顺序，像这样：

Category                Navigation order(one column)

CatA                      0
|-subcat11                 0
  |-sub11a                    0
  |-sub11b                    1
  \-sub11c                    2
\-subcat12                 1
CatB                      1  
|-subcat21                 0
|-subcat22                 1
\-subcat23                 2
  \-sub23a                    0
CatC                      2

我该如何快速填充或重新计算在创建或移动元素时的顺序列呢？或者通过分类的方法来计算，比如 Category.objects.get(name='sub11b').get_order() 应该返回1。

Answer 1

在最近的 mptt 版本中（比如 0.8.7），你应该使用 TreeForeignKey 这个字段：

from mptt.models import MPTTModel
from mptt.fields import TreeForeignkey

class Category(MPTTModel):
    name = models.CharField()
    parent = TreeForeignKey("self", 
                           blank=True, 
                           null=True, 
                           related_name="sub_category")

    class MPTTMeta:
        order_insertion_by = ['name']

Answer 2

在定义模型的时候，你可以用“order_insertion_by”来指定排序方式。

比如可以这样写：

class Category(MPTTModel):
    name = models.CharField()
    parent = models.ForeignKey("self", blank=True, null=True, 
             related_name="sub_category")

    class MPTTMeta:
        order_insertion_by = ['name']

然后你可以通过 Category.tree.rebuild() 来重建你的数据库，这样就会按照你指定的排序来处理。