详解python递归查询菜单并转换成json实例代码

本篇文章主要介绍了python递归查询菜单并转换成json实例,具有一定的参考价值,感兴趣的小伙伴们可以参考一下。

最近需要用python写一个菜单,折腾了两三天才搞定,现在记录在此,需要的朋友可以借鉴一下。

备注:文章引用非可执行完整代码,仅仅摘录了关键部分的代码

环境

数据库:mysql

python:3.6

表结构

create table `tb_menu` (
`id` varchar(32) not null comment ‘唯一标识’,
`menu_name` varchar(40) default null comment ‘菜单名称’,
`menu_url` varchar(100) default null comment ‘菜单链接’,
`type` varchar(1) default null comment ‘类型’,
`parent` varchar(32) default null comment ‘父级目录id’,
`del_flag` varchar(1) not null default ‘0’ comment ‘删除标志 0:不删除 1:已删除’,
`create_time` datetime default current_timestamp comment ‘创建时间’,
`update_time` timestamp not null default current_timestamp on update current_timestamp comment ‘更新时间’,
primary key (`id`) using btree
) engine=innodb default charset=utf8 comment=’菜单表’;

python代码

menu对象中,有一个子菜单列表的引用“submenus”,类型为list

核心代码

def set_submenus(id, menus):
“””
根据传递过来的父菜单id,递归设置各层次父菜单的子菜单列表
:param id: 父级id
:param menus: 子菜单列表
:return: 如果这个菜单没有子菜单,返回none;如果有子菜单,返回子菜单列表
“””
# 记录子菜单列表
submenus = []
# 遍历子菜单
for m in menus:
if m.parent == id:
submenus.append(m)
# 把子菜单的子菜单再循环一遍
for sub in submenus:
menus2 = querybyparent(sub.id)
# 还有子菜单
if len(menus):
sub.submenus = set_submenus(sub.id, menus2)
# 子菜单列表不为空
if len(submenus):
return submenus
else: # 没有子菜单了
return none

测试方法

def test_set_submenus(self):
# 一级菜单
rootmenus = querybyparent(”)
for menu in rootmenus:
submenus = querybyparent(menu.id)
menu.submenus = set_submenus(menu.id, submenus)

备注:基本流程是:先查询一级菜单,然后分别把该级菜单的id、和这级菜单的子菜单列表传入set_submenus方法,递归进行子菜单列表的下级菜单设置;

支持传递菜单id,查询该菜单下面的所有子菜单。传递空字符,则从根目录开始查询

在“rootmenus ”对象中,可以看到完整的菜单树形结构

转json

我采用的orm框架是:sqlalchemy,直接从数据库中查询出来的menu对象,转json时会报错。需要重新定义一个dto类,来把menu对象转成dto对象。

menudto

class menudto():
def __init__(self, id, menu_name, menu_url, type, parent, submenus):
super().__init__()
self.id = id
self.menu_name = menu_name
self.menu_url = menu_url
self.type = type
self.parent = parent
self.submenus = submenus
def __str__(self):
return ‘%s( % (
self.__class__.__name__, self.id, self.menu_name, self.menu_url, self.type, self.parent)
__repr = __str__

于是,重新定义了递归设置子菜单的方法

def set_submenudtos(id, menudtos):
“””
根据传递过来的父菜单id,递归设置各层次父菜单的子菜单列表
:param id: 父级id
:param menudtos: 子菜单列表
:return: 如果这个菜单没有子菜单,返回none;如果有子菜单,返回子菜单列表
“””
# 记录子菜单列表
submenudtos = []
# 遍历子菜单
for m in menudtos:
m.name = to_pinyin(m.menu_name)
if m.parent == id:
submenudtos.append(m)
# 把子菜单的子菜单再循环一遍
for sub in submenudtos:
menus2 = querybyparent(sub.id)
menusdto2 = model_list_2_dto_list(menus2,
“menudto(, menu_name=”, menu_url=”, type=”, parent=”, submenus=”)”)
# 还有子菜单
if len(menudtos):
if len(menusdto2):
sub.submenus = set_submenudtos(sub.id, menusdto2)
else: # 没有子菜单,删除该节点
sub.__delattr__(‘submenus’)
# 子菜单列表不为空
if len(submenudtos):
return submenudtos
else: # 没有子菜单了
return none

备注:

当一个菜单没有子菜单时,删除掉“submenus”属性,否则转json时会出现空值

model_list_2_dto_list 方法可以把menu列表转成menudto列表

to_pinyin 是把汉字转成拼音的方法,在这里不用关注

view层返回json的方法

def get(self):
param = request.args
id = param[‘id’]
# 如果id为空,查询的是从根目录开始的各级菜单
rootmenus = querybyparent(id)
rootmenudtos = model_list_2_dto_list(rootmenus,
“menudto(, menu_name=”, menu_url=”, type=”, parent=”, submenus=”)”)
# 设置各级子菜单
for menu in rootmenudtos:
menu.name = to_pinyin(menu.menu_name)
submenus = querybyparent(menu.id)
if len(submenus):
submenudtos = model_list_2_dto_list(submenus,
“menudto(, menu_name=”, menu_url=”, type=”, parent=”, submenus=”)”)
menu.submenus = set_submenudtos(menu.id, submenudtos)
else:
menu.__delattr__(‘submenus’)
menus_json = json.dumps(rootmenudtos, default=lambda o: o.__dict__, sort_keys=true, allow_nan=false,
skipkeys=true)
# 需要转字典,否则返回的字符串会带有“\”
menus_dict = json_dict(menus_json)
return fullresponse(menus_dict)
fullresponse
from flask import jsonify
def fullresponse(data=”, msg=”, code=0):
if msg == ”:
return jsonify({‘code’: code, ‘data’: data})
elif data == ”:
return jsonify({‘code’: code, ‘msg’: msg})
else:
return jsonify({‘code’: code, ‘msg’: msg, ‘data’: data})

备注:python中json和字典的含义类似,在最后json返回给页面时,需要先使用json_dict方法转成dict类型,否则返回的字符串中会带有“\”

查询结果

以上就是详解python递归查询菜单并转换成json实例代码的详细内容,更多请关注 第一php社区 其它相关文章!

Posted in 未分类