python 文件操作练习

shilang

# python 3.6 文件打开操作练习　　
#  date：2017-12-26
　　
# _*_ coding:utf-8 _*_
　　

　　
import os,sys
　　

　　
def find():
　　
    find_name = input("请输入要查询的员工姓名：").strip().upper()
　　
    flag=False
　　
    with open("info.txt",'r',encoding='utf-8') as f1:
　　
        for line in f1:
　　
            if line.split()[0]==find_name:
　　
                flag=True
　　
                print("员工姓名:",find_name,"\t工资：",line.split()[1])
　　
            else:
　　
                continue
　　
    if not flag:
　　
        print("未找到员工信息:\033[31;1m{_name}\033[1m".format(_name=find_name))
　　

　　
def modify():
　　
    flag = False
　　
    mod_info1=input("请输入要修改的员工姓名：").strip().upper()
　　
    with open("info.txt", 'r', encoding='utf-8') as f1 , \
　　
        open("info_new", 'w', encoding='utf-8') as f2:
　　
        for line in f1:
　　
            if mod_info1 ==line.split()[0]:
　　

　　
                mod_info2 = input("请输入修改后的工资：").strip()
　　
                if not mod_info2.isdigit():  #判断输入是否为数字
　　
                    print("请输入一个数字！！")
　　
                    sys.exit()  #需等待完善优化，下同
　　
                else:
　　
                    f2.write(mod_info1+'\t'+mod_info2+'\n')
　　
                    flag = True
　　
            else:
　　
                 f2.write(line)
　　
    if os.path.exists('info_bak.txt'):os.remove("info_bak.txt") #如果存在则删除
　　
    os.rename("info.txt","info_bak.txt")
　　
    os.rename("info_new","info.txt")
　　
    if flag:
　　
         print("修改成功")
　　
    else:
　　
         print("未找到员工信息:\033[31;1m{_name}\033[1m".format(_name=mod_info1))
　　

　　
def add():
　　
    flag = False
　　
    add_info1 = input("请输入要增加的员工姓名：").strip().upper()
　　
    with open("info.txt", 'r', encoding='utf-8') as f1 :
　　
        for line in f1:
　　
            if add_info1 == line.split()[0]:
　　
                flag = True
　　
            else:
　　
                continue
　　
    if flag:
　　
        print("员工\033[31;1m{_name}\033[1m信息已存在".format(_name=add_info1))
　　
    else:
　　
        with open("info.txt", 'a', encoding='utf-8') as f2 :
　　
            add_info2 = input("请输入工资：").strip()
　　
            if not add_info2.isdigit():  # 判断输入是否为数字
　　
                print("请输入一个数字！！")
　　
                sys.exit()
　　
            else:
　　
                f2.write(add_info1 + '\t' + add_info2 + '\n')
　　
                print("增加成功！！")
　　

　　
def dele():
　　
    flag = False
　　
    del_info1 = input("请输入要删除的员工姓名：").strip().upper()
　　
    with open("info.txt",'r',encoding='utf-8') as f1 ,\
　　
        open('info_new','w',encoding='utf-8') as f2:
　　
        for line in f1:
　　
            if line.split()[0] == del_info1:
　　
                flag=True
　　
                continue
　　
            else:
　　
                f2.write(line)
　　
    if os.path.exists('info_bak.txt'):os.remove("info_bak.txt") #如果存在则删除
　　
    os.rename("info.txt","info_bak.txt")
　　
    os.rename("info_new","info.txt")
　　
    if flag:
　　
        print("删除成功！！")
　　
    else:
　　
        print("未找到员工信息:\033[31;1m{_name}\033[1m".format(_name=del_info1))
　　

　　
def main():
　　
    menu={
　　
        '1':find,
　　
        '2':modify,
　　
        '3':add,
　　
        '4':dele,
　　
        '5':exit
　　
    }
　　
    while True:
　　
        print("""
　　
        请选择：
　　
        1，查询员工信息
　　
        2，修改员工信息
　　
        3，增加员工信息
　　
        4，删除员工信息
　　
        5，退出系统
　　
        """ )
　　
        choice=input('>>>>').strip()
　　
        if choice in menu:
　　
            menu[choice]()
　　
if __name__=="__main__":
　　
    main()