python实现三级菜单

# 构建三级菜单的数据类型：

dic = {
    '江西': {
        '南昌': {'南昌大学': {}, '华东交大': {} ,'江西财大': {}},
        '赣州': {'赣南师院': {}, '江西理工大学学院': {}},
        '上饶': {'吃饭大学': {}, '睡觉大学': {}},
    },
    '河南': {
        '郑州': {'郑州大学': {}, '河南工业大学': {}},
        '洛阳': {}
    },
    '河北': {},
    '海南': {}
}

def three_level_menu(dic):
    while 1:
        for k in dic.keys():
            print(k)
        choice = input('输入选择的城市：').strip()
        if choice == 'q'or choice == 'b':
            return choice
        elif choice in dic.keys() and dic[choice]:
            ret = three_level_menu(dic[choice])
            if ret == 'q':
                return ret



three_level_menu(dic)