python遞歸查詢菜單并轉(zhuǎn)換成json實(shí)例

更新時(shí)間：2017年03月27日 11:07:24 作者：快樂檸檬他哥

本篇文章主要介紹了python遞歸查詢菜單并轉(zhuǎn)換成json實(shí)例，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下。

最近需要用python寫一個(gè)菜單，折騰了兩三天才搞定，現(xiàn)在記錄在此，需要的朋友可以借鑒一下。

備注：文章引用非可執(zhí)行完整代碼，僅僅摘錄了關(guān)鍵部分的代碼

環(huán)境

數(shù)據(jù)庫：mysql
python：3.6

表結(jié)構(gòu)

CREATE TABLE `tb_menu` (
 `id` varchar(32) NOT NULL COMMENT '唯一標(biāo)識',
 `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 '刪除標(biāo)志 0：不刪除 1：已刪除',
 `create_time` datetime DEFAULT CURRENT_TIMESTAMP COMMENT '創(chuàng)建時(shí)間',
 `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新時(shí)間',
 PRIMARY KEY (`id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='菜單表';

Python代碼

Menu對象中，有一個(gè)子菜單列表的引用“subMenus”，類型為list

核心代碼

def set_subMenus(id, menus):
  """
  根據(jù)傳遞過來的父菜單id，遞歸設(shè)置各層次父菜單的子菜單列表

  :param id: 父級id
  :param menus: 子菜單列表
  :return: 如果這個(gè)菜單沒有子菜單，返回None;如果有子菜單，返回子菜單列表
  """
  # 記錄子菜單列表
  subMenus = []
  # 遍歷子菜單
  for m in menus:
    if m.parent == id:
      subMenus.append(m)

  # 把子菜單的子菜單再循環(huán)一遍
  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方法，遞歸進(jìn)行子菜單列表的下級菜單設(shè)置；

支持傳遞菜單Id，查詢該菜單下面的所有子菜單。傳遞空字符，則從根目錄開始查詢

在“rootMenus ”對象中，可以看到完整的菜單樹形結(jié)構(gòu)

轉(zhuǎn)Json

我采用的ORM框架是：sqlalchemy，直接從數(shù)據(jù)庫中查詢出來的Menu對象，轉(zhuǎn)Json時(shí)會(huì)報(bào)錯(cuò)。需要重新定義一個(gè)DTO類，來把Menu對象轉(zhuǎn)成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__

于是，重新定義了遞歸設(shè)置子菜單的方法

def set_subMenuDtos(id, menuDtos):
  """
  根據(jù)傳遞過來的父菜單id，遞歸設(shè)置各層次父菜單的子菜單列表

  :param id: 父級id
  :param menuDtos: 子菜單列表
  :return: 如果這個(gè)菜單沒有子菜單，返回None;如果有子菜單，返回子菜單列表
  """
  # 記錄子菜單列表
  subMenuDtos = []
  # 遍歷子菜單
  for m in menuDtos:
    m.name = to_pinyin(m.menu_name)
    if m.parent == id:
      subMenuDtos.append(m)

  # 把子菜單的子菜單再循環(huán)一遍
  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: # 沒有子菜單，刪除該節(jié)點(diǎn)
        sub.__delattr__('subMenus')

  # 子菜單列表不為空
  if len(subMenuDtos):
    return subMenuDtos
  else: # 沒有子菜單了
    return None

備注：

當(dāng)一個(gè)菜單沒有子菜單時(shí)，刪除掉“subMenus”屬性，否則轉(zhuǎn)Json時(shí)會(huì)出現(xiàn)空值
model_list_2_dto_list 方法可以把Menu列表轉(zhuǎn)成MenuDto列表
to_pinyin 是把漢字轉(zhuǎn)成拼音的方法，在這里不用關(guān)注

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='')")
    # 設(shè)置各級子菜單
    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)
    # 需要轉(zhuǎn)字典，否則返回的字符串會(huì)帶有“\”
    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返回給頁面時(shí)，需要先使用json_dict方法轉(zhuǎn)成dict類型，否則返回的字符串中會(huì)帶有“\”

查詢結(jié)果