首頁 > 文章 > 後端開發 > 詳解python遞歸查詢選單並轉換成json實例代碼

詳解python遞歸查詢選單並轉換成json實例代碼

高洛峰原創: 2017-03-28 09:16:252334瀏覽

本篇文章主要介紹了python遞歸查詢選單並轉換成json實例，具有一定的參考價值，有興趣的小夥伴們可以參考一下。

最近需要用python寫一個選單，折騰了兩三天才搞定，現在記錄在此，需要的朋友可以藉鏡一下。

備註：文章引用非可執行完整程式碼，僅僅摘錄了關鍵部分的程式碼

f環境

資料庫： mysql
python：3.6

#表結構

CREATE TABLE `tb_menu` (
 `id` varchar(32) NOT NULL COMMENT &#39;唯一标识&#39;,
 `menu_name` varchar(40) DEFAULT NULL COMMENT &#39;菜单名称&#39;,
 `menu_url` varchar(100) DEFAULT NULL COMMENT &#39;菜单链接&#39;,
 `type` varchar(1) DEFAULT NULL COMMENT &#39;类型&#39;,
 `parent` varchar(32) DEFAULT NULL COMMENT &#39;父级目录id&#39;,
 `del_flag` varchar(1) NOT NULL DEFAULT &#39;0&#39; COMMENT &#39;删除标志 0：不删除 1：已删除&#39;,
 `create_time` datetime DEFAULT CURRENT_TIMESTAMP COMMENT &#39;创建时间&#39;,
 `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT &#39;更新时间&#39;,
 PRIMARY KEY (`id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT=&#39;菜单表&#39;;

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(&#39;&#39;)
    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 &#39;%s(id=%s,menu_name=%s,menu_url=%s,type=%s,parent=%s)&#39; % (
      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=&#39;&#39;, menu_name=&#39;&#39;, menu_url=&#39;&#39;, type=&#39;&#39;, parent=&#39;&#39;, subMenus=&#39;&#39;)")
    # 还有子菜单
    if len(menuDtos):
      if len(menusDto2):
        sub.subMenus = set_subMenuDtos(sub.id, menusDto2)
      else: # 没有子菜单，删除该节点
        sub.__delattr__(&#39;subMenus&#39;)

  # 子菜单列表不为空
  if len(subMenuDtos):
    return subMenuDtos
  else: # 没有子菜单了
    return None

備註：

View層回傳Json的方法

  def get(self):
    param = request.args
    id = param[&#39;id&#39;]
    # 如果id为空，查询的是从根目录开始的各级菜单
    rootMenus = queryByParent(id)
    rootMenuDtos = model_list_2_dto_list(rootMenus,
                       "MenuDto(id=&#39;&#39;, menu_name=&#39;&#39;, menu_url=&#39;&#39;, type=&#39;&#39;, parent=&#39;&#39;, subMenus=&#39;&#39;)")
    # 设置各级子菜单
    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=&#39;&#39;, menu_name=&#39;&#39;, menu_url=&#39;&#39;, type=&#39;&#39;, parent=&#39;&#39;, subMenus=&#39;&#39;)")
        menu.subMenus = set_subMenuDtos(menu.id, subMenuDtos)
      else:
        menu.__delattr__(&#39;subMenus&#39;)

    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=&#39;&#39;, msg=&#39;&#39;, code=0):
  if msg == &#39;&#39;:
    return jsonify({&#39;code&#39;: code, &#39;data&#39;: data})
  elif data == &#39;&#39;:
    return jsonify({&#39;code&#39;: code, &#39;msg&#39;: msg})
  else:
    return jsonify({&#39;code&#39;: code, &#39;msg&#39;: msg, &#39;data&#39;: data})

備註：python中json和字典的意思類似，在最後json回傳給頁時，需要先使用json_dict方法轉換成dict類型，否則傳回的字串中會帶有「\」

#查詢結果

以上是詳解python遞歸查詢選單並轉換成json實例代碼的詳細內容。更多資訊請關注PHP中文網其他相關文章！

陳述：

本文內容由網友自願投稿，版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容，請聯絡admin@php.cn

上一篇：詳解Python利用Beautiful Soup模組修改內容範例程式碼下一篇：詳解Python利用Beautiful Soup模組修改內容範例程式碼

看更多

詳解python遞歸查詢選單並轉換成json實例代​​碼

相關文章

詳解python遞歸查詢選單並轉換成json實例代碼