一、需求：
1.导入文件读取员工的信息和工资信息，最后将增加、删除或修改的员工工资信息写入原来的文件中
2.能够实现员工工资信息的增删改查
3.当增加和修改员工信息时用户用空格分隔员工姓名和薪资
4.实现退出功能
二、思路
1.首先需要将包含有员工薪资信息的文件读出来存储到一个容器里，因为员工姓名和薪资一一对应，因此想到字典的key--->value机制，将员工薪资信息存储到字典中。
2.写一个函数让用户选择该如何操作----->add、delete、modify、search（增删改查）以及退出操作
3.定义四个函数----- >user_add（）、user_delete（）、user_modify（）、user_seach（）
三、实现（源码）
#!/user/bin/env python
#-*-coding:utf-8 -*-
#Author: qinjiaxi
import sys,os,re
operation_list = """
1.增加新员工薪资信息
2.删除已有员工薪资信息
3.修改已有员工薪资信息
4.查询已有员工薪资信息
5.退出操作
"""
user_dict = {}#空字典用来存储员工薪资信息
def user_info():
'''定义员工薪资信息函数'''
with open('info.txt','r') as f:
for i in f:
i = i.strip()#去掉每行字符串前后空格和换行以及缩进
user_dict[i.split()[0]] = i.split()[1]#将员工信息以键值对key(员工姓名)--->value(员工薪资)的方式存储在字典中
print('当前员工的姓名:', i.split()[0]) def user_choice():
'''用户输入选择'''
while True:
print(operation_list)
user_choice = input('请选择操作编号:')
if user_choice.isdigit():
user_choice = int(user_choice)
if user_choice > 5:
print('\033[31;1m请输入正确的操作编号\033[0m')
if user_choice == 1:
user_info_add()
if user_choice == 2:
user_info_del()
if user_choice == 3:
user_info_modify()
if user_choice == 4:
user_info_search()
if user_choice == 5:
sys.exit('退出程序')
else:
print('\033[31;1m请输入正确的操作编号\033[0m') def user_info_add():
'''定义增加新员工薪资信息函数'''
while True:
user_info()
add_user = input('请输入要增加员工的姓名和薪资，并以空格隔开（例如: Qinjiaxi 10000）')
add_user_info = add_user.strip().split()#去掉用户输入前后的空格换行以及缩进,并将姓名和工资以列表形式打印
if len(add_user_info) < 2:
print('\033[31;1m请输入正确的格式\033[0m')
elif not add_user_info[1].isdigit():
print('\033[31;1m请输入正确的数字\033[0m')
else:
name = add_user_info[0].capitalize()#把用户名的第一个字母变成大写
salary = add_user_info[1]
if name in user_dict:#判断姓名是否在字典中
print("\033[31;1m您输入的名字已存在\033[0m")
elif not name.isalpha():#判断姓名是否是全英文
print('\033[31;1m请输入正确的姓名')
else:
with open('info.txt','a') as f:
f.write(name + ' ')
f.write(salary + '\n')
print('已将\033[31;1m%s\033[0m用户信息添加成功' % name)
break def user_info_del():
'''定义删除员工薪资信息函数'''
while True:
user_info()
del_user = input('请输入要删除员工的姓名，例如（Qinjiaxi）')
del_user_info = del_user.capitalize()#把首个字母改成大写，方便用户输入
if del_user_info in user_dict:
confirm = input('请确认是否删除\033[32;1m %s\033[0m用户信息--->按\033[32;1mN\033[0m退出，按任意键确认'% del_user_info)
if confirm.capitalize() != 'N':
list = []#定义一个空列表用于存储不需要删除的员工信息
with open('info.txt', 'r') as f:
lines = f.readlines()
for line in lines:
if not re.search(del_user_info, line):#判断哪些员工不需要删除
list.append(line)
with open('info.txt', 'w') as f:
f.writelines(list)
print('\033[32;1m%s\033[0m已经删除' % del_user_info)
break
else:
print('\033[32;1m%已取消删除\033[0m')
else:
print('\033[32;1m该用户不存在\033[0m') def user_info_modify():
'''定义修改员工薪资信息函数'''
while True:
user_info()
modify_user = input('请输入要修改员工的姓名和薪资，并以空格隔开(例如：Qinjiaxi 10000)')
modify_user_info = modify_user.strip().split()
if len(modify_user_info) < 2:
print('\033[32;1m请输入正确的格式\033[0m')
elif not modify_user_info[1].isdigit():
print('\033[32;1m请输入正确的数字\033[0m')
else:
name = modify_user_info[0].capitalize()#姓名
salary = modify_user_info[1]#薪资
if name in user_dict:
with open('info.txt', 'r') as f:
lines = f.readlines()
with open('info.txt', 'w') as f_w:
for line in lines:
if name in line:
line = line.replace(user_dict[name], salary)#替换薪资
f_w.write(line)
print('已将\033[32;1m%s\033[0m的工资修改为\033[32;1m%s\033[0m元'%(name, salary))
break
else:
print('\033[32m;1m您输入的用户不存在\033[0m') def user_info_search():
'''定义查询员工薪资信息函数'''
while True:
user_info()
search_user = input('请输入要查询的员工姓名(例如：Qinjiaxi)')
search_user_info = search_user.capitalize()
if search_user_info in user_dict:
print('\033[32;1m%s\033[0m工资为\033[32;1m%s\033[0m元' % (search_user_info, user_dict[search_user_info]))
break
else:
print('\033[32;1m您输入的用户不存在\033[0m') user_choice()