# -*- coding:utf-8 -*-
def main():
try:
datas = open("data.txt").readlines()
except:
datas = []
choice=1
while choice:
print ("=============================")
print ("1.新增记录")
print ("2.查询和删除数据")
print ("3.保存退出")
print ("=============================")
n=input("请选择:...")
if n == "1":
number=input("请输入学号:")
name = input("请输入姓名:")
age = input("请输入年龄:")
datas.append(number+","+name+","+age+"\n")
elif n == "2":
number = input("请输入要查找的学号:")
#查找数据
for i,data in enumerate(datas):
if data.split(",")[0] == number:
print ("找到一条数据", data)
position=i
break
#确认是否删除数据
if input("是否要删除此条数据?(y/n)")=="y":
del datas[i]
elif n == "3":
open("data.txt", "w").write("".join(datas))
break
main()