python递归查询菜单并转换成json实例

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

最近需要用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(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时会出现空值 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,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,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)fullResponsefrom flask import Jsonifydef 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实例全部内容，希望文章能够帮你解决python递归查询菜单并转换成json实例所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。