Django-mptt中的动态排序

Question

我正在使用django-mptt包来开发我的评论应用程序，以下是我的模型：

class Comment(MPTTModel):
    content = models.TextField(verbose_name='Treść')
    author = models.ForeignKey(AUTH_USER_MODEL, verbose_name='Autor', blank=False, null=True)
    is_deleted = models.BooleanField(verbose_name='Komentarz usunięty', default=False,
                                     help_text='Zaznacz, aby usunąć komentarz')

    ip = models.GenericIPAddressField(default=0, verbose_name='Adres IP')

    content_type = models.ForeignKey(ContentType, verbose_name='Typ obiektu')
    object_id = models.PositiveIntegerField(verbose_name='ID obiektu')
    content_object = GenericForeignKey('content_type', 'object_id')
    parent = TreeForeignKey('self', null=True, blank=True, related_name='children', db_index=True)
    hotness = models.FloatField(default=0)

    created_at = models.DateTimeField(auto_now_add=False, verbose_name='Data dodania')

    updated_at = models.DateTimeField(auto_now=True, verbose_name='Aktualizacja')

    class MPTTMeta:
        order_insertion_by = ('-hotness', '-created_at')

    class Meta:
        verbose_name = 'Komentarz'
        verbose_name_plural = 'Komentarze'

    def __unicode__(self):
        if len(self.content) > 50:
            return self.content[:50] + '...'
        else:
            return self.content

我希望为用户提供按热度或创建日期排序评论树的可能性。是否可以从视图编辑order_insertion_by字段以生成两种类型的排序（按日期、按热度）？感谢您的帮助。

- Peterek

---

你找到关于那个的解决方案了吗？ - Murat Çorlu

我用下面的代码示例（Python3）进行了几次测试，看起来它可以工作 - 但如果有其他人也试一试就好了。 - biodiv

Answer 1

Modified Preorder Tree Traversal（MPTT）是一种使用左侧编号（在mptt中为lft）和右侧编号（rgt）检索树结构的方法，如此处所示：http://sitepointstatic.com/graphics/sitepoint_numbering.gif。定义多个order_insertion_by将执行以下操作（根据mptts注释）：

"""
    Creates a filter which matches suitable right siblings for ``node``,
    where insertion should maintain ordering according to the list of
    fields in ``order_insertion_by``.

    For example, given an ``order_insertion_by`` of
    ``['field1', 'field2', 'field3']``, the resulting filter should
    correspond to the following SQL::

       field1 > %s
       OR (field1 = %s AND field2 > %s)
       OR (field1 = %s AND field2 = %s AND field3 > %s)

"""

如果我理解正确，order_insertion_by指定了同级元素的顺序，这些元素代表父元素的子节点（而不是后代节点）。如果您想要两种不同的顺序，lft和rgt也必须改变，从而形成第二棵树。这在mptt中不包括。

您仍然可以这样做：

Comment.objects.all().order_by('-hotness')

但是你会失去树形结构。一般情况下，无法保持原有的树形结构并按其他方式（例如热门程度）对整个树进行排序。想象一下你有以下内容：

Comment1 (hotness 0)
    Comment2 (hotness 2, child of Comment1)
Comment3 (hotness 1)

这将导致...

Comment2
Comment3
Comment1

这里有一个排序的问题，但是Comment2并没有被绑定到Comment1上。 如果你想使用一些与order_insertion_by定义不同的方式进行同级基础的排序，请尝试以下方法：

Comment3
Comment1
    Comment2

有可能可以编写一个新的模板标签，例如{% recursetree objects -hotness %}，它可以遍历和重新排序children元素并返回新树。它仍然只需要一次数据库查询-但我无法估计性能影响。

您需要fork mptt并编辑mptt_tags.py如下：

class RecurseTreeNode(template.Node):
    def __init__(self, template_nodes, queryset_var, order_var=None):
        self.template_nodes = template_nodes
        self.queryset_var = queryset_var
        self.order_var = order_var

    def _render_node(self, context, node):
        bits = []
        context.push()

        children = node.get_children()

        if children and self.order_var is not None:
            children = children.order_by(self.order_var)            

        for child in children:
            bits.append(self._render_node(context, child))
        context['node'] = node
        context['children'] = mark_safe(''.join(bits))
        rendered = self.template_nodes.render(context)
        context.pop()
        return rendered

    def render(self, context):
        queryset = self.queryset_var.resolve(context)
        roots = cache_tree_children(queryset)
        bits = [self._render_node(context, node) for node in roots]
        return ''.join(bits)


@register.tag
def recursetree(parser, token):
    bits = token.contents.split()
    if len(bits) < 2:
        raise template.TemplateSyntaxError(_('%s tag requires a queryset') % bits[0])

    queryset_var = template.Variable(bits[1])

    if len(bits) == 3:
        order_var = bits[2]
    else:
        order_var = None

    template_nodes = parser.parse(('endrecursetree',))
    parser.delete_first_token()

    return RecurseTreeNode(template_nodes, queryset_var, order_var)

- biodiv