python遞迴查詢菜單並轉換成json執行個體，pythonjson

python遞迴查詢菜單並轉換成json執行個體，pythonjson

最近需要用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，查詢該菜單下面的所有子功能表。傳遞Null 字元，則從根目錄開始查詢

在“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

備忘：

1. 當一個菜單沒有子功能表時，刪除掉“subMenus”屬性，否則轉Json時會出現空值
2. model_list_2_dto_list 方法可以把Menu列錶轉成MenuDto列表
3. 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)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類型，否則返回的字串中會帶有“\”

查詢結果

以上就是本文的全部內容，希望對大家的學習有所協助，也希望大家多多支援幫客之家。