Python3练习：对员工信息文件，实现增删改查操作

1.练习要求：

2.数据文件（data_staff.txt)

1,Alex Li,22,13651054684,运维,2013-02-04
2,Jack Wang,20,13312331232,HR,2014-06-03
3,Mike Cao,20,15504231232,Sales,2013-05-06
4,Jack Chen,34,12404231232,HR,2011-02-01
5,Lu Haojie,21,15204231232,运维,2013-08-12

3.实现代码

A.主程序部分：采用字典类型来实现函数的调用（key对应的value是函数名，通过Name()语法可以调用函数执行）,利用os模块的remove()和rename()实现文件的修改后的旧文件删除和新文件的重命名。

 import os
 msg_dict={
     '':Search,
     '':Add,
     '':Delete,
     '':Change,
     '':'退出'
 }
 
 while True:
     print( """
    1:查询
    2:添加
    3:删除
    4:修改
    5:退出
    """)
     choice=input('输入序号：')
     if choice not in msg_dict:
         print("输入错误，请重新输入")
         continue
     if int(choice)==5:
         exit()
     else:
         msg_dict[choice]()

B.查询函数：主要通过字符串常用操作的split()实现从字符串到列表的转换，join()实现列表到字符串的转换

 def Search():
      # 查询方式一：select * from staff_table where age >= 22
      # 查询方式二：select * from staff_table where dept = "IT"
      # 查询方式三：select * from staff_table where enroll_date like "2013"
 
      data=input('输入您的查询信息：')
      data=data.split(' ')
      con=data[7]
      asp=data[5]
      count=0
      with open('data_staff','r',encoding='utf-8') as f:
         if asp=='age':
             for line in f:
               if int(line.split(',')[2])>=int(con):
                  print(line)
                  count+=1
         elif asp=='dept':
             for line in f:
                if line.split(',')[4] in con:
                    print(line)
                    count += 1
         else:
             for line in  f:
                if line.split(',')[5].split('-')[0] in con:
                    print(line)
                    count += 1
         print('查询结束，共查到符合条件的信息 %d 条 ' % count)

C.添加函数：利用‘r+’的可读写模式来实现读写（注意这种方式下的write()是追加到文件末尾的）

 def Add():
     # 添加语法： name,age,phone,dept,enroll-date
     data=input('输入您要添加的员工信息：')
     list_data=data.strip().split(',')
     list_all=[]
     f=open('data_staff','r+')
     for line in f:
         list_all.append(line.strip().split(',')[3])
     if list_data[2] in list_all:
         print("该用户已经存在")
         f.close()
     else:
         for line in f:
             f.write(line)
         staff_id=str(len(list_all)+1)
         list_data.insert(0,str(staff_id))
         f.write('\n')
         f.write(','.join(list_data))
         f.close()
         print('添加成功')

D.删除函数：

 def Remove():
     # 删除语法：delete from staff_table where staff_id = 12
     staff_id=input("输入您要删除员工的Staff_id:")
     staff_id=staff_id.strip().split(' ')[6]
     f=open('data_staff','r')
     f1=open('new_data_staff','w')
     for line in f:
         in_list=line.split(',')
         if in_list[0]<staff_id:
             f1.write(line)
         elif in_list[0]>staff_id:
             in_list[0]=str(int(in_list[0])-1)
             f1.write(','.join(in_list))
         else:
             continue
     f.close()
     f1.close()
     os.remove('data_staff')
     os.rename('new_data_staff','data_staff')
     print('删除成功')

E.修改函数:

 def Change():
     # 修改请输入（注意空格和没有引号）：UPDATE staff_table SET dept = IT where dept = 运维
     data=input("请输入您要修改的信息：")
     old=data.split(' ')[5]
     new=data.split(' ')[9]
     f=open('data_staff','r',encoding='utf-8')
     f1=open('new_data_staff','w',encoding="utf-8")
     for line in f:
         if old in line:
              line=line.replace(old,new)
         f1.write(line)
     f.close()
     f1.close()
     os.remove('data_staff')
     os.rename('new_data_staff','data_staff')
     print('修改成功')