实现一个小功能,当输入相应的序号,会执行相关操作,
比如当输入序号1,会执行添加功能
#coding:utf-8
print '''
1 添加数据
2 删除数据
3 修改数据
4 查看数据
5 退出程序
'''
def add():
# 输入姓名,年龄,电话
name=raw_input('name:')
age=raw_input('age:')
phone=raw_input('phone:')
#封装个人信息,放入list里面
stu=[name,age,phone]
#把学生放入数据源data_list
data_list.append(stu)
print data_list
print 'hi,welcome to Beijing!'
def delete():
print '删除数据'
def update():
print 'hi,welcome to Shanghai!'
def query():
print '查看数据'
def exit():
print '退出程序'
#保存数据到txt文档
def save():
f=open('stu1.txt','w')
#遍历数据源,把学生信息写入文档
for stu1 in data_list:
#遍历学生信息
for item in stu1:
f.write(item)
if item==stu1[-1]:
f.write('\n')
print 'save....'
#声明一个list数组
data_list=[]
while 1:
num=input('请选择您的操作:')
if num==1:
add()
# save()
elif num==2:
delete()
elif num==3:
update()
elif num==4:
query()
elif num==5:
exit()
break
else:
print 'error'
save()