python学习day07---三级目录优化

"""
需求
三级目录
①展示出省级目录
②用户选择省份
    1.跳出对应的地级市
    2.可以返回上一级菜单
    3.可以选择地级市
    4.可以退出程序
③选择地级市
    1.跳出对应的县
    2.可以返回上一级菜单
    3.可以退出

"""
chinese_city = {
    '江苏省': {
        '苏州市': ['姑苏区', '高新区', '工业园区'],
        '镇江市': ['润州区', '京口区', '丹阳市'],
        '南京市': ['鼓楼区', '雨花台区', '建邺区']
    },
    '浙江省': {
        '苏市': ['姑区', '高区', '工区'],
        '镇市': ['润区', '京区', '丹市'],
        '南市': ['鼓区', '雨区', '建区']
    },
    '广东省': {
        '州市': ['苏区', '新区', '园区'],
        '江市': ['州区', '口区', '阳市'],
        '京市': ['楼区', '台区', '邺区']
    }

}

current_level = chinese_city
parents_level = []

while True:
    for key in current_level:
        print(key)
    choice = input("请输入")
    if choice in current_level:
        parents_level.append(current_level)
        current_level = current_level[choice]
    elif choice == 'quit':
        break
    elif choice == 'return':
        if parents_level:
            current_level = parents_level.pop()
    else:
        print("输入错误")