如何在django-treenode中限制评论树获取深度并实现‘更多’功能？

可以实现，只需要给序列化器添加深度控制逻辑，通过递归时传递深度参数限制层级，同时在达到最大深度时返回“有更多内容”的标识，供前端触发加载。

修改序列化器代码 #

在现有序列化器基础上，新增深度参数的处理逻辑：

class CommentSerializer(serializers.ModelSerializer):
    leaf_nodes = serializers.SerializerMethodField()

    class Meta:
        model = Comment
        fields = ['id', 'body', 'leaf_nodes']

    def __init__(self, *args, **kwargs):
        # 从context读取最大深度，默认设为3层
        self.max_depth = self.context.get('depth', 3)
        super().__init__(*args, **kwargs)

    def get_leaf_nodes(self, obj):
        children = obj.get_children()
        if self.max_depth > 0:
            # 递归序列化子节点，深度逐层递减
            return CommentSerializer(
                children,
                many=True,
                context={'depth': self.max_depth - 1}
            ).data
        else:
            # 达到最大深度，返回是否存在子节点的标识
            return [{'has_more': True, 'parent_id': obj.id}] if children.exists() else []

在视图中使用序列化器 #

获取根评论时，通过context传递预设的最大深度：

# 假设根评论是parent为None的记录
root_comments = Comment.objects.filter(parent__isnull=True)
serializer = CommentSerializer(root_comments, many=True, context={'depth': 3})
return Response(serializer.data)

实现“更多”加载功能 #

前端识别到leaf_nodes中的has_more: True标识时，渲染“更多”按钮。点击按钮时，向API发送请求（比如GET /api/comments/{parent_id}?depth=3），视图中处理该请求：

def load_more_comments(request, parent_id):
    parent_comment = Comment.objects.get(id=parent_id)
    depth = int(request.query_params.get('depth', 3))
    serializer = CommentSerializer(
        parent_comment.get_children(),
        many=True,
        context={'depth': depth}
    )
    return Response(serializer.data)

前端拿到返回的子评论数据后，替换掉原有的has_more条目，即可实现分层加载。

内容的提问来源于stack exchange，提问作者Shay