本篇文章主要介紹了python遞歸查詢選單並轉換成json實例，具有一定的參考價值，有興趣的小夥伴們可以參考一下。
最近需要用python寫一個選單，折騰了兩三天才搞定，現在記錄在此，需要的朋友可以藉鏡一下。
備註：文章引用非可執行完整程式碼，僅僅摘錄了關鍵部分的程式碼
f環境
#表格結構##
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(id=%s,menu_name=%s,menu_url=%s,type=%s,parent=%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(id='', 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時會出現空值-
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(id='', 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(id='', 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轉換的實例的詳細內容。更多資訊請關注PHP中文網其他相關文章！
