需求：
1.可进行模糊查询，语法至少支持下面3种:
1.select name,age from staff_table where age > 22
2.select * from staff_table where dept = IT
3.查到的信息，打印后，最后面还要显示查到的条数
2.可创建新员工纪录，以phone做唯一键，staff_id需自增
3.可删除指定员工信息纪录，输入员工id，即可删除
4.可修改员工信息，语法如下:
1.UPDATE staff_table SET dept= 无 where dept = IT
注意：请输入正确的语句进行查询，否则会报错。
#_*_coding:utf-8_*_
import os
def fetch(data):
#查询语法一：select name,age from staff_table where age > 22
#查询语法二：select * from staff_table where dept = IT
data1 = data.split(" ")
directory = ["staff_id", "name", "age", "phone", "dept", "enroll-date"]
if data == ("select name,age from staff_table where age > %s" %(data1[7])):
with open("xinxi", encoding="utf-8") as f:
list = []
list1 = []
list2 = []
for line in f:
i = line.strip().split(",")
w = i[1]
e = i[2]
a = [w, e]
if e > data1[7]:
list2.append(a)
for i in list2:
print(i)
print("查询到 %s 条符合的信息" %len(list2))
else:
with open("xinxi", encoding="utf-8") as f:
list = []
for line in f:
i = line.strip().split(",")
q = i[0]
w = i[1]
e = i[2]
r = i[3]
t = i[4]
y = i[5]
if data == ("select * from staff_table where %s = %s" % (data1[5], i[(directory.index(data1[5]))])):
list.append(i)
else:
continue
for j in list:
print('.'.join(j))
print("查询到 %s 条符合的信息" %len(list))
return 0
def add(data):
#添加语法： name,age,phone,dept,enroll-date (例如：王新凯,22,11111111111,IT,2015-10-31)
data1 = data.split(",")
f= open("xinxi",encoding="utf-8")
all_list = []
list = []
phone_list = []
for line in f:
i = line.strip().split(",")
q = i[3]
phone_list.append(q) if data1[2] in phone_list:
print("手机号已存在")
f.close()
else:
f1 = open("xinxi", "r+", encoding="utf-8")
for line in f1:
lines = line.strip().split(",")
# print(lines)
list.append(lines)
i = line.strip().split(",")
w = str(int(list[-1][0]) + 1)
data1.insert(0, w)
print(data1)
data1 = ','.join(data1)
f1.write("\n")
f1.write(data1)
f1.close()
print("添加成功!!!")
def remove(data):
#删除语法：delete from staff_table where staff_id = 12
data1 = data.split(" ")
if data == ("delete from staff_table where staff_id = %s" %data1[6]):
with open("xinxi", encoding="utf-8") as f:
list = []
for line in f:
i = line.strip().split(",")
i1 = line.splitlines()
q = i[0]
if data1[6] == q:
i2 = ','.join(i1)
print(i2)
list.append(i)
a = i2
f = open("xinxi", encoding="utf-8")
f1 = open("back", "a+", encoding="utf-8")
for i in f:
if a in i:
i = i.replace(a, "").strip()
f1.write(i)
f1.flush()
f.close()
f1.close()
os.remove("xinxi")
os.rename("back","xinxi")
print("删除成功！！！")
return
def change(data):
#修改请输入（注意空格和没有引号）：UPDATE staff_table SET dept = IT where dept = 运维
data1 = data.split(" ")
print(data1)
# a =','.join(data1[3])
# print(a)
directory = ["staff_id", "name", "age", "phone", "dept", "enroll-date"]
var = int(directory.index(data1[3]))
with open("xinxi", encoding="utf-8") as f,\
open("back","w",encoding="utf-8")as f1:
for line in f:
lines = line.strip()
print(lines)
if data1[5] in lines:
lines = lines.replace(data1[5],data1[9])
f1.write(lines)
f1.write("\n")
f1.flush()
os.remove("xinxi")
os.rename("back","xinxi")
print("修改成功!!!") if __name__ == "__main__":
msg = """
1:查询
2:添加
3:删除
4:修改
5:退出
"""
msg_dict = {
"": fetch,
"": add,
"": remove,
"": change,
"": exit,
}
while True:
print(msg)
choice = input("输入序号>>:")
if len(choice) == 0 or choice not in msg_dict: continue
if choice =='':break
data = input("请输入数据>>:").strip()
msg_dict[choice](data)