最近需要用python寫一個(gè)菜單，折騰了兩三天才搞定，現(xiàn)在記錄在此，需要的朋友可以借鑒一下。
備注：文章引用非可執(zhí)行完整代碼，僅僅摘錄了關(guān)鍵部分的代碼
環(huán)境
- 數(shù)據(jù)庫：mysql
- python：3.6
表結(jié)構(gòu)
|
1
2
3
4
5
6
7
8
9
10
11
|
CREATE TABLE `tb_menu` ( `id` varchar(32) NOT NULL COMMENT '唯一標(biāo)識(shí)', `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
核心代碼
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
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 |
測試方法
|
1
2
3
4
5
6
|
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
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
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è)置子菜單的方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
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的方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
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)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返回給頁面時(shí)，需要先使用json_dict方法轉(zhuǎn)成dict類型，否則返回的字符串中會(huì)帶有“\”
查詢結(jié)果
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持服務(wù)器之家。
原文鏈接：http://www.jianshu.com/p/4e63bc531da5
