帶返回值的python多級選單
阿新 • • 發佈：2019-01-02
介紹：
利用函式遞迴實現
理論上可生成N級
支援返回上級
原創作品，轉載註明出處。
#!-*- coding:utf-8 -*-
__author__ = 'David'
class CreateMenu(object):
def __init__(self):
self.temp_menu = {} # 臨時選單，儲存每級選單的字典。
def run(self, menu_dict, initial=0):
'''
:param menu_dict: 原始選單字典，格式：{1:{2:{3:{4:{5:'value'}}}}},可以返回5的值
:param initial: 選單等級控制引數，預設0級，最上層選單
:return: 返回值是字典中最小引數值，字串格式。
'''
self.temp_menu[initial] = menu_dict
if type(menu_dict) is dict:
temp = {}
i = 1 # 選單標號數字
# 生成選擇選單
for k, v in menu_dict.items():
t = str(i)
temp[t] = k
print('{}: {}'.format(i, k))
i += 1
print('0', 'back')
choose = input('-->:').strip()
if choose.isdigit(): # 判斷如果輸入的是數字，繼續判斷選擇的是哪一項
if choose == '0' and initial == 0: # 等於0，並且和初始選單一樣，不做任何操作
return self.run(menu_dict, initial)
elif choose == '0': # 等於0，不和初始選單一樣，則返回
del self.temp_menu[initial] # 從臨時層級列表裡刪除本級
initial -= 1 # 修改完上一級引數
return self.run(self.temp_menu[initial], initial)
elif int(choose) > i or int(choose) < 1: # 如果選擇的數字超出選項值，報錯，重新運行當前選單
print('Input Error!')
return self.run(menu_dict, initial)
else: # 正常進入下級選單
print(temp[choose])
initial += 1 # 選單層級加1
menu_dict = menu_dict[temp[choose]] # 獲取選擇的選項值
if type(menu_dict) is dict: # 如果選擇的值是字典，則判斷為有子選單
return self.run(menu_dict, initial) # 顯示子選單選項
else: # 如果不是字典，則返回選項對應的值
return menu_dict
else: # 如果不是數字，再執行一遍函式，引數都是和之前一樣的
print('Input Error!')
return self.run(menu_dict, initial)
else: # 如果輸入的引數不是字典則報錯。
print('Data type is Error!')
if __name__ == '__main__':
teacher_menu = {'管理班級': {'建立班級': 'create_grade', '新增學員': 'add_stu'},
'上課': 'create_record',
'管理成績': 'mgr_grade',
'退出': 'exit'}
menu = CreateMenu()
value = menu.run(teacher_menu)
print('<value:{}>'.format(value))