#!/usr/bin/env python#_*_ coding: utf-8 _*_#author:chenjianwen#email:1071179133@qq.com
importsys,os##定义操作列表
ops = ['查询员工工资','修改员工工资','增加新员工记录','退出']
info={}defget_info():##读取文件得到员工信息
globalinfo
with open('info.txt',mode='r',encoding='utf-8') as f_r:for line inf_r:
line= line.rstrip('\n')ifline:
name= line.split(' ')[0]
slary= line.split(' ')[1]
info[name]={'name':'%s'%name,'slary':'%s'%slary,'other':''}##定义写操作函数
defwrite(name,slary):
with open('info.txt', mode='a+', encoding='utf-8') as f_w:
f_w.write('\n%s %s'%(name,slary))##定义修改员工信息函数
defreplace(old_name,new_name,old_slary,new_slary):
f_r= open('info.txt',mode='r',encoding='utf-8')
f_w= open('info1.txt', mode='w+', encoding='utf-8')for line inf_r:
line=line.strip()if old_name inline:
line=line.replace(old_name,new_name)
line=line.replace(old_slary,new_slary)
f_w.write(line+ '\n')
f_r.close()
f_w.close()whileTrue:
get_info()##打印操作菜单
for key,ops_lists inenumerate(ops):print(key,ops_lists,)
select= input("请选择操作序号:")ifselect.isdigit():pass
else:print("请输入数字")continue
if select.startswith('0'):
input1= input("请输入员工名字:")print("%s的工资是:%s" %(input1,info[input1]['slary']))continue
elif select.startswith('1'):
input1= input("请输入修改员工的名字:")print("%s现在的信息是:名字:%s,工资:%s"%(input1,info[input1]['name'],info[input1]['slary']))
input2= input("名字修改为:")
input3= input("工资修改为:")
replace(input1, input2,info[input1]['slary'], input3)
os.remove('info.txt')
os.rename('info1.txt','info.txt')print("员工信息修改成功")
get_info()continue
elif select.startswith('2'):
input1= input("请输入增加员工的名字:")
input2= input("请输入增加员工的工资:")
write(input1,input2)print("新增员工信息成功")
get_info()continue
elif select.startswith('3'):print("退出成功")
sys.exit(1)