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

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

環(huán)境

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

python：3.6

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

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 '創(chuàng)建時間',

`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):

"""

根據(jù)傳遞過來的父菜單id，遞歸設置各層次父菜單的子菜單列表

:param id: 父級id

:param menus: 子菜單列表

:return: 如果這個菜單沒有子菜單，返回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方法，遞歸進行子菜單列表的下級菜單設置；

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

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

轉(zhuǎn)Json

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

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

def set_subMenuDtos(id, menuDtos):

"""

根據(jù)傳遞過來的父菜單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)

# 把子菜單的子菜單再循環(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é)點

sub.__delattr__('subMenus')

# 子菜單列表不為空

if len(subMenuDtos):

return subMenuDtos

else: # 沒有子菜單了

return None

備注：

當一個菜單沒有子菜單時，刪除掉“subMenus”屬性，否則轉(zhuǎn)Json時會出現(xiàn)空值

model_list_2_dto_list 方法可以把Menu列表轉(zhuǎn)成MenuDto列表

to_pinyin 是把漢字轉(zhuǎ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='')")

# 設置各級子菜單

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)字典，否則返回的字符串會帶有“\”

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方法轉(zhuǎn)成dict類型，否則返回的字符串中會帶有“\”

查詢結(jié)果

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

總結(jié)

以上是生活随笔為你收集整理的python list转换成树形结构_python递归查询菜单并转换成json实例的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。