带返回值的python多级菜单

介绍：

创新互联建站网站建设公司，提供做网站、成都网站建设，网页设计，建网站，PHP网站建设等专业做网站服务;可快速的进行网站开发网页制作和功能扩展；专业做搜索引擎喜爱的网站，是专业的做网站团队，希望更多企业前来合作!

利用函数递归实现

理论上可生成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(''.format(value))