python返回菜单_带返回值的python多级菜单

介绍:

利用函数递归实现

理论上可生成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) 

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(''.format(value))