最近需要用python写一个菜单,折腾了两三天才搞定,现在记录在此,需要的朋友可以借鉴一下。
环境
表结构
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) 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类型,否则返回的字符串中会带有“\”
查询结果
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持编程小技巧。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。