首先需要设计表结构,包含父子关联关系,例如
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