python调用函数实现数据的增删改查（2）

1  添加数据

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!'

注意：当添加完数据桌面会新建一个txt文档，添加数据会查看到内容，如下：

2删除数据

def delete():
query()
num=input('请输入要删除的序号：')
while num<0 or num>len(data_list):
num=input('请输入正确的序号：')
del data_list[num-1]
print '删除数据成功！'

3   修改数据，并没有完善好这里

4  查询数据

def query():
count=0
for stu in data_list:
count+=1
print count,'name',stu[0] ,'age:',stu[1],'phone:',stu[2]
print '_'*10

5  退出程序

具体代码如下：

#coding:utf-8
print '''
1 添加数据
2 删除数据
3 修改数据
4 查看数据
5 退出程序
'''
data_list=[]
#定义加载数据函数load_data
def load_data():
f=open('stu.txt','r') #读取数据
res=f.readlines() #读取多行
for stu in res:
print stu
#把学生放入数据源data_list

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():
query()
num=input('请输入要删除的序号：')
while num<0 or num>len(data_list):
num=input('请输入正确的序号：')
del data_list[num-1]
print '删除数据成功！'
def update():
print 'hi,welcome to Shanghai!'
def query():
count=0
for stu in data_list:
count+=1
print count,'name',stu[0] ,'age:',stu[1],'phone:',stu[2]
print '_'*10
def exit():
print '退出程序'
#保存数据到txt文档
def save():
f=open('/Users/mac/Desktop/stu1.txt','w')
#遍历数据源，把学生信息写入文档
for stu1 in data_list:
#遍历学生信息
for item in stu1:
f.write(item)
if item==stu1[-1]:
f.write('\n')
print 'save....'
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()