首页 > 代码库 > Python实现三层菜单操作
Python实现三层菜单操作
#!/usr/bin/env python3
# encoding:utf-8
# __author__: chenwei
# date: 2017-05-25
# blog: http://www.cnblogs.com/weiandlu/
‘‘‘json数据文件格式如下:
{
‘北京‘:{
‘海淀‘:{
‘五道口‘:{
‘soho‘:{},
‘网易‘:{},
‘google‘:{}
},
‘中关村‘:{
‘爱奇艺‘:{},
‘汽车之家‘:{},
‘youku‘:{},
},
‘上地‘:{
‘百度‘:{},
},
},
‘昌平‘:{
‘沙河‘:{
‘高教园‘:{},
‘北航‘:{},
},
‘天通苑‘:{},
‘回龙观‘:{},
},
‘朝阳‘:{},
‘东城‘:{},
},
‘上海‘:{
‘闵行‘:{
"人民广场":{
‘炸鸡店‘:{}
}
},
‘闸北‘:{
‘火车战‘:{
‘携程‘:{}
}
},
‘浦东‘:{},
},
‘山东‘:{},
}
‘‘‘
import json
def read_data():
with open("administrative_division.json","r",encoding="utf8") as f_read:
data=http://www.mamicode.com/json.load(f_read)
return data
def write_data(data):
with open("administrative_division.json","w",encoding="utf8") as f_write:
json.dump(data,f_write)
data = http://www.mamicode.com/read_data()
current_layer = read_data()
last_layer = [] #在菜单层层深入时记录上一层菜单
layer_count = {‘layer‘:0,‘key‘:[]} #=每进入一层该字典的layer数值增加1,同时将每层选择的城市名append到key的列表中
num_to_menu = {} #将菜单列表打印的序列号与城市名成进行对应
while True:
print("欢迎使用三级菜单系统".center(50,"#"))
for i,v in enumerate(current_layer,1):
print("%s\t%s\t" % (i,v),end=‘‘)
num_to_menu[i] = v
print()
choice=input("请选择对应选项前的字母:\na:添加,d:删除,m:修改,b:返回,q:退出:").strip()
if choice.isdigit() and int(choice) in num_to_menu:
last_layer.append(current_layer)
current_layer = current_layer[num_to_menu[int(choice)]]
layer_count[‘layer‘] += 1
layer_count[‘key‘].append(num_to_menu[int(choice)])
write_data(data)
elif choice == ‘a‘:
city=input("请输入城市名称!:").strip()
if layer_count[‘layer‘] == 1:
if not data[layer_count[‘key‘][0]]:
data[layer_count[‘key‘][0]].update({city:{}})
elif layer_count[‘layer‘] == 2:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][1]].update({city:{}})
elif layer_count[‘layer‘] == 3:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]].update({city:{}})
elif layer_count[‘layer‘] == 4:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]].update({city:{}})
else:
pass
current_layer.update({city:{}})
write_data(data)
elif choice == ‘d‘:
city_num=int(input("请输入删除城市序号!").strip())
layer_count[‘layer‘] += 1
layer_count[‘key‘].append(num_to_menu[city_num])
if layer_count[‘layer‘] == 1:
if not data[layer_count[‘key‘][0]]:
del data[layer_count[‘key‘][0]]
elif layer_count[‘layer‘] == 2:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]:
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]
elif layer_count[‘layer‘] == 3:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]:
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]
elif layer_count[‘layer‘] == 4:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]:
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]
else:
pass
write_data(data)
elif choice == ‘m‘:
city_num=int(input("请输入要修改的城市序号!").strip())
new_name=input("请输入新的城市名称!").strip()
layer_count[‘layer‘] += 1
layer_count[‘key‘].append(num_to_menu[city_num])
if layer_count[‘layer‘] == 1:
if not data[layer_count[‘key‘][0]]:
data[new_name]=data[layer_count[‘key‘][0]]
del data[layer_count[‘key‘][0]]
elif layer_count[‘layer‘] == 2:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][new_name]]=data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]]
elif layer_count[‘layer‘] == 3:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][new_name]]=data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]]
elif layer_count[‘layer‘] == 4:
if not data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]:
data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][new_name]]=data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]
del data[layer_count[‘key‘][0]][layer_count[‘key‘][1]][layer_count[‘key‘][2]][layer_count[‘key‘][3]]
else:
pass
write_data(data)
elif choice == ‘b‘:
if current_layer != data:
current_layer = last_layer[-1]
last_layer.pop()
layer_count[‘layer‘] -= 1
layer_count[‘key‘].pop()
print(layer_count)
continue
else:
print("\033[31;1myou have reach to the top layer!\033[;0m")
elif choice == ‘q‘:
exit(1)
else:
print("\033[1;31;40m请根据提示进行选择!\033[;0m")
Python实现三层菜单操作