Python进阶篇：文件系统的操作

通过一个例子来熟悉文件的基本操作：创建文件，读取文件，修改文件，删除文件，重命名文件，判断文件是否存在

 '''
 编写可供查询的员工信息表--学号 姓名  年龄 班级
    1. 提供格式化查询接口
    2. 允许用户添加或者删除员工信息
 '''
 import os
 employee_dir = 'employee_list'
 index=employee_dir+"/index"

 if(not os.path.exists(employee_dir)):
     os.mkdir(employee_dir,777)

 def add_student(id,name,age,_class):
     local_file = employee_dir+"/"+str(id)
     if(os.path.isfile(local_file)):
         print("id: %s 已经存在，请重新添加" %(id))
     else:
         f=open(local_file,"w")
         f.write(name+"\t"+str(age)+"\t"+_class)
         f.close()
         f=open(index,"a")
         f.write(str(id)+"\t"+name+"\n")
         f.close()
         print("id=%s添加成功" %id)

 def select_student(**kwargs):
     for key in kwargs.keys():
         if(key=='id'):
             local_file = employee_dir+"/"+str(kwargs[key])
             if(not os.path.isfile(local_file)):
                 print("不存在id:",kwargs[key])
             else:
                 f=open(local_file,"r")
                 print(kwargs[key],f.read())
         elif(key=='name'):
             if(not os.path.isfile(index)):
                 print("不存在name:",kwargs[key])
             else:
                 i=0
                 f=open(index,"r")
                 for line in f.readlines():
                     id,name = line.strip().split("\t")
                     if(name == kwargs[key]):
                         i=i+1
                         select_student(id=id)
                 if(i==0):
                     print("不存在name:",kwargs[key])
         else:
             print("无效的获取方式，只能通过id和name来获取!")

 def modify_student(id,name,age,_class):
     local_file = employee_dir+"/"+str(id)
     if(os.path.isfile(local_file)):
         os.remove(local_file)
         rm_line(index,id)
         add_student(id,name,age,_class)
         print("id=%s修改成功" %id)
     else:
         print("id: %s 不存在，请先添加" %(id))

 def delete_student(**kwargs):
     for key in kwargs.keys():
         if(key=='id'):
             local_file = employee_dir+"/"+str(kwargs[key])
             if(not os.path.isfile(local_file)):
                 print("不存在id:",kwargs[key],",无法删除")
             else:
                 os.remove(local_file)
                 rm_line(index,kwargs[key])
                 print("删除id=%s成功" %kwargs[key])
         elif(key=='name'):
             if(not os.path.isfile(index)):
                 print("不存在name:",kwargs[key],",无法删除")
             else:
                 i=0
                 f=open(index,"r")
                 for line in f.readlines():
                     id,name = line.strip().split("\t")
                     if(name == kwargs[key]):
                         i=i+1
                         delete_student(id=id)
                 if(i==0):
                     print("不存在name:",kwargs[key],",无法删除")
         else:
             print("无效的删除方式，只能通过id和name来删除!")

 def rm_line(file,id):
     if(not os.path.isfile(file)):
         print("不存在文件:",file)
     else:
         f= open(file,"r")
         of = open(file+".tmp","w")
         for line in f.readlines():
             localid,localname = line.strip().split("\t")
             if(str(id)==localid):
                 pass
             else:
                 of.write(line,)
         f.close()
         of.close()
         if(os.path.isfile(file+".tmpt")): os.remove(file+".tmpt")
         os.rename(file,file+".tmpt")
         os.rename(file+".tmp",file)