介绍:
利用函数递归实现
理论上可生成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))
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。