Python 练习 文件处理

...

#coding = utf - 8
#用户管理模块

__author__='Jerr Q'

file = 'd:\\123.txt'
def add(name,age,phonenum) :
    #添加用户信息
    global file
    with open(file,'a') as f :
        f.write(name + ' '*4 + age + ' '*4 + phonenum)
    print('添加完成')
def revise(style,old,new) :
    #根据选择修改相应的项目
    global file
    with open(file,'r') as f :
        line = f.readlines()
    with open(file,'w') as f1 :
        for i in line :
            if style == 'name' :
            #修改名字
                if old in i :
                    a = len(old)
                    f1.write(new + i[a:])
                else :
                    f1.write(i)
            elif style == 'age' :
            #修改年龄
                if old in i :
                    a = len(old)
                    b = i.index(' ')
                    f1.write(i[0:b+3]+new+i[b+3+a+1:])
                else :
                    f1.write(i)
            elif style == 'phonenum':
            #修改电话
                if old in i :
                    a = len(old)
                    f1.write(i[0:-(a+1)]+new+'\n')
                else :
                    f1.write(i)
        print('修改完成')
def delete(style,target):
    #删除用户信息
    with open(file,'r') as f:
        line = f.readlines()
    with open(file,'w') as f1:
        for i in line :
            #删除单个内容
            # if style == 'name' :
            #     if target in i :
            #         a = len(target)
            #         f1.write(' '*a + i[a:])
            #     else:
            #         f1.write(i)
            # elif style == 'age' :
            #     if target in i :
            #         a = len(target)
            #         b = i.index(' ')
            #         f1.write(i[0:b+3]+' '*a+i[b+3+a+1:])
            #     else:
            #         f1.write(i)
            # elif style == 'phonenum' :
            #     if target in i :
            #         a = len(target)
            #         f1.write(i[0:-(a+1)]+' '*4+'\n')
            #     else:
            #         f1.write(i)
            #删除整行内容
            if target in i :
                continue
            f1.write(i)
        print('删除完毕')
print('欢迎进入用户管理系统')
chosen1 = input('请选择您想进行的操作(add、revise、delete）：')
if chosen1 == 'add' :
    #创建新用户信息到用户信息文件中
    newname = input('请输入您的姓名：')
    newage = input('请输入您的年龄：')
    newphonenum = input('请输入您的电话: ')
    add(newname,newage,newphonenum)
elif chosen1 == 'revise' :
    #修改用户信息
    mod = input('请选择修改类型(name、age、phonenum)：')
    old_content = input('请输入您要修改的原始信息：')
    new_content = input('请输入修改的新信息：')
    revise(mod,old_content,new_content)
elif chosen1 == 'delete' :
    #删除信息
    mod = input('请选择删除类型(name、age、phonenum)：')
    target = input('请输入您要删除的：')
    delete(mod,target)