Python之多级菜单

要求：

1. 三级菜单
2. 可依次选择进入各子菜单

流程图后续再补充

代码如下：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author Jiang

menu={
'陕西':{
    '西安':['大雁塔','兵马俑'],
    '宝鸡':['法门寺','云台山'],
    '汉中':['青木川','武侯祠'],
},
'山西':{
    '太原':['晋祠','纯阳宫'],
    '大同':['恒山','华严寺'],
    '临汾':['壶口瀑布','千佛洞'],
},
'湖南':{
    '长沙':['岳麓山','橘子洲'],
    '株洲':['神农谷','炎帝陵'],
    '衡阳':['衡山','水帘洞'],
},
}

province_list=list(menu.keys())
while True:
    for index,key in enumerate(menu.keys(),1):
        print('省',index,key)
    pro_id=input('请输入省编号，或输入q退出：')
    if pro_id.isdigit():
        pro_id=int(pro_id)
        if pro_id>0 and pro_id<=len(province_list):
            pro_name=province_list[pro_id-1]
            city_list=list(menu[pro_name].keys())
            while True:
                for j in city_list:
                    print('市',city_list.index(j)+1,j)
                city_id=input('请输入市编号，或输入b返回上级菜单，或输入q退出：')
                if city_id.isdigit():
                    city_id=int(city_id)
                    if city_id>0 and city_id<=len(city_list):
                        city_name=city_list[city_id-1]
                        view_list=menu[pro_name][city_name]
                        while True:
                            for i in view_list:
                                print('景点',view_list.index(i)+1,i)
                            view_id=input('请输入景点编号，或输入b返回上级菜单，或输入q退出：')
                            if view_id=='b':
                                break
                            elif view_id=='q':
                                exit()
                            else:
                                print('输入不合格')
                    else:
                        print('您输入的用户%d不存在' %city_id)
                elif city_id=='b':
                    break
                elif city_id=='q':
                    exit()
                else:
                    print('输入不合格')
        else:
            print('您输入的用户%d不存在' %pro_id)
    elif pro_id=='q':
        exit()
    else:
        print('输入不合格')