#!/usr/bin/env python#-*- coding:utf-8 -*-#Author:Huanglinsheng

importos#查询方式一：select * from data_staff.txt where age >= 22#查询方式二：select * from data_staff.txt where dept = "IT"#查询方式三：select * from data_staff.txt where enroll_date like "2013"

defSearch():

data=input("Input your select info:")

data= data.split(' ')

con= data[7]

asp= data[5]

count=0

with open('data_staff.txt','r',encoding='utf-8') as f:if asp == 'age':for line inf:if int(line.split(',')[2]) >=int(con):print(line)

count+=1

elif asp == 'dept':for line inf:if line.split(',')[4] incon:print(line)

count+=1

else:for line inf:if line.split(',')[5].split('-')[0] incon:print(line)

count+=1

print('查询结束，共查到符合条件的信息 %d 条' %count)#添加语法： name,age,phone,dept,enroll-date

defAdd():

data= input("请输入要添加员工的信息：")

list_data= data.strip().split(',')

list_all=[]

f= open('data_staff.txt','r+')for line inf:

list_all.append(line.strip().split(',')[3])if list_data[2] inlist_all:print("该用户已经存在！！")else:for line inf:

f.write(line)

staff_id= str(len(list_all) + 1)

list_data.insert(0,str(staff_id))

f.write('\n')

f.write(','.join(list_all))

f.close()print("添加成功")#删除语法：delete from staff_table where staff_id = 12

defDelete():

staff_id= input("输入您要删除员工的Staff_id:")

staff_id= staff_id.strip().split(' ')[6]

f= open('data_staff.txt','r')

f1= open('new_data_staff.txt','w')for line inf:

in_list= line.split(',')if in_list[0]

f1.write(line)elif in_list[0] >staff_id:

in_list[0]= str(int(in_list[0]) - 1)

f1.write(','.join(in_list))else:continuef.close()

f1.close()

os.remove("data_staff.txt")

os.rename('new_data_staff.txt','data_staff')print("删除成功！！")#修改请输入(注意空格和没有引号)：UPDATE staff_table SET dept = IT where dept = 运维

defChange():

data= input("请输入你要修改的信息：")

old= data.split(' ')[5]

new= data.split(' ')[9]

f= open('data_staff.txt','r',encoding='utf-8')

f1= open('new_data_staff','w',encoding='utf-8')for line inf:if old inline:

line=line.replace(old,new)

f1.write(line)

f.close()

f1.close()

os.remove('data_staff')

os.rename('new_data_staff', 'data_staff')print('修改成功')

msg_dict={'1':Search,'2':Add,'3':Delete,'4':Change,'5':'退出'}whileTrue:print("""1:查询

2:添加

3:删除

4:修改

5:退出""")

choice= input("Input your choice:")if choice not inmsg_dict:print("Input error!Pls Input one more time.")continue

if int(choice) == 5:

exit()else:

msg_dict[choice]()