# _*_coding:utf-8_*_
# 作者：王佃元
# 日期：2019/12/15
'''
三级菜单新增功能需求
1.需将地区内容保存至文件中
2.用户可对地区内容进行增删改
'''
menu = {
'上海':{
'闵行区':{
'老闵行':{
'紫竹科技园':{},
'上海交大':{},
},
'新闵行':{
'浦江镇':{},
'漕河泾':{},
},
},
'黄浦区':{
'外滩':{
'东方明珠':{},
'南京步行街':{},
'豫园':{},
},
'建筑':{
'人民广场':{},
'大剧院':{},
},
},
},
'深圳':{
'福田区':{
'华强北':{
'华强电子城':{},
'赛格电子城':{},
},
'步行街':{
'东门步行街':{},
'荔枝公园':{},
},
},
'南山区':{
'科技园':{
'南山科技园':{},
'西丽科技园':{},
},
'手机园':{
'酷派科技园':{},
'oppo科技园':{},
},
},
},
}
with open('three_level_menu', 'r', encoding='utf8') as menu_read:
zone = eval(menu_read.read().strip())
msg = '''*******请输入地区*******
输入b返回上一层
输入q退出
***********************'''
current_layer = zone
parent_layers = []
user_enter_quit = False
while not user_enter_quit:
print(msg)
for key in current_layer:
print(key)
user_enter = input("please input :")
if user_enter in current_layer:
parent_layers.append(current_layer) #每一次的当前层对应字典追加至列表中
current_layer = current_layer[user_enter] #将下一层字典赋给当前层
elif user_enter.lower() == 'b' and parent_layers != []:
current_layer = parent_layers.pop() #默认删除列表中最后一个元素，并返回该元素值。使用pop不会增加列表长度
elif user_enter.lower() == 'q':
current_layer = parent_layers.pop(0)
user_enter_quit = True
elif user_enter not in current_layer: #如果用户输入内容不在字典中，则将用户输入作为key添加至一个新字典
if user_enter == 'mod':
original_zone = input('请输入修改的地区：')
mod_zone = input('请输入要修改内容：')
current_layer[mod_zone] = current_layer[original_zone]
current_layer.pop(original_zone)
elif user_enter == 'rem':
rem_enter = input('请输入要删除的地区：')
current_layer.pop(rem_enter)
elif len(parent_layers) < 3:
add_enter = input('对不起，无此地区，请重新输入添加地区：')
current_layer[add_enter] = {} #用户输入内容写入字典后，则进入下一层，提示用户继续输入下一个地点
# print(current_layer[user_enter])
else:
print('最后一层')
else:
print('please enter again!')
with open('three_level_menu', 'w', encoding='utf8') as menu_write:
menu_write.write(str(current_layer).strip())
