django中查询表数据组成树形数据返回

Question

Django中自定义的一个表，表的字段是：id、name、parentid
想要通过某个ID查询出这个ID及以下的树形数据，并组成下面格式返回

[{
          name: '一级 1',
          id：1，
          children: [{
            name: '二级 1-1', 
            id：3，
            children: [{
              name: '三级 1-1-1'，
               id：5，
            }]
          }]
        }, {
          name: '一级 2', 
          id：2，
          children: [{
            name: '二级 2-1', 
            id：4，
            children: [{
              name: '三级 2-1-1'，
               id：6，
            }]
          }, {
            name: '二级 2-2',
            id：7，
            children: [{
              name: '三级 2-2-1'，
              id：8，
            }]
          }]
  }
]

不知道怎么查询处理可以得到这个数据，想到递归或django-treebeard，但不会用，请哪位能帮忙给讲一下。

Answer 1

一种方法是使用递归，在查询每个节点时调用自身函数，并将父节点 ID 作为参数传入。另一种方法是使用 django-treebeard 库，该库提供了一些工具来方便地处理树形数据。

递归示例代码如下：

def get_tree_data(parent_id):
    children = MyModel.objects.filter(parentid=parent_id)
    data = []
    for child in children:
        data.append({
            'name': child.name,
            'id': child.id,
            'children': get_tree_data(child.id)
        })
    return data

使用 django-treebeard 库的示例代码如下：

from treebeard.mp_tree import MP_Node

class MyModel(MP_Node):
    name = models.CharField(max_length=255)
    parentid = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True)

    node_order_by = ['name']

# 获取子树
subtree = MyModel.objects.get(id=parent_id).get_descendants()

这里给出的是示例代码，实际使用中需要根据你的具体需求进行修改。