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+"	"+str(age)+"	"+_class)
        f.close()
        f=open(index,"a")
        f.write(str(id)+"	"+name+"
")
        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("	")
                    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("	")
                    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("	")
            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)