python生成树形结构 菜单 tree

首先需要设计表结构，包含父子关联关系，例如

class Menu(models.Model):
    """
    权限
    """
    title = models.CharField(max_length=128, unique=True, verbose_name="标题")
    name = models.CharField(max_length=128, unique=True, verbose_name="编码")
    path = models.CharField(max_length=256, blank=True, null=True, verbose_name='含正则的URL')
    parent_id = models.IntegerField(verbose_name='父菜单id')

编写生成tree方法，具体用法看说明，调用方式 get_trees(Menu.objects.values().all())

# 生成树结构
def get_trees(data,
              key_column='id',
              parent_column='parent_id',
              child_column='children',
              current_column=None,
              current_path=None):
    """
    :param data: 数据列表
    :param key_column: 主键字段，默认id
    :param parent_column: 父ID字段名，父ID默认从0开始
    :param child_column: 子列表字典名称
    :param current_column: 当前展开值字段名，若找到展开值增加['open'] = '1'
    :param current_path: 当前展开值
    :return: 树结构
    """
    data_dic = {}
    for d in data:
        data_dic[d.get(key_column)] = d  # 以自己的权限主键为键,以新构建的字典为值,构造新的字典

    data_tree_list = []  # 整个数据大列表
    for d_id, d_dic in data_dic.items():
        pid = d_dic.get(parent_column)  # 取每一个字典中的父id
        if not pid:  # 父id=0，就直接加入数据大列表
            data_tree_list.append(d_dic)
        else:  # 父id>0 就加入父id队对应的那个的节点列表
            try:  # 判断异常代表有子节点，增加子节点列表=[]
                data_dic[pid][child_column].append(d_dic)
            except KeyError:
                data_dic[pid][child_column] = []
                data_dic[pid][child_column].append(d_dic)

        # 展开节点
        if current_path:
            if current_path == d_dic.get(current_column):
                d_dic['open'] = '1'
                while pid:
                    data_dic[pid]['open'] = '1'
                    pid = data_dic[pid][parent_column]
    return data_tree_list