#-*- Coding:utf-8 -*-
# Author: D.Gray
'''
作业需求
1、从info.txt文件中读取员工及其工资信息，最后将修改或增加的员工工资信息也写入原info.txt文件。
2、能增查改员工工资
3、增、改员工工资用空格分隔
4、实现退出功能
'''
import sys,os,reoperation_lists = '''1.查询工资
2.修改工资
3.增加新员工记录
4.删除员工信息
5.退出'''
user_dict = {}                  #定义一个存储员工姓名及工资的空字典
def user_information():'''定义一个员工信息函数'''with open('info','r') as f:for i in f:i = i.strip()           #剔除字符串中的前后空格和换行user_dict[i.split()[0]] = i.split()[1]  #将员工姓名及工资存储到user_dict字典中，i.split()[0]=员工姓名 作为键#i.split()[1]=员工工资  作为值print('当前员工姓名：',i.split()[0])def user_operations():'''定义一个用户操作的函数'''while True:print(operation_lists)user_operation = input('请选择操作编号>>>:')if user_operation.isdigit():user_operation = int(user_operation)if user_operation > 5:print('\033[31;1m请输入有效操作编号\033[0m')if user_operation == 1:user_enquiries()if user_operation == 2:salary_change()if user_operation == 3:add_users()if user_operation == 4:del_users()if user_operation == 5:sys.exit('程序退出')else:print('\033[31;1m请输入有效操作编号\033[0m')def user_enquiries():'''定义一个用户查询的函数'''while True:user_information()enquirie_name = input('请输入要查询的员工姓名（例如：Alex）：')if enquirie_name.capitalize() in user_dict:     #将输入的员工姓名首字母变大写，方便用户输入print('\033[32;1m%s\033[0m 工资为: \033[32;1m%s\033[0m元'%(enquirie_name.capitalize(),user_dict[enquirie_name.capitalize()]))breakelse:print('\033[31;1m请输入有效用户名\033[0m')def salary_change():'''定义一个修改工资函数'''while True:user_information()change_salary = input('请输入要修改的员工姓名和工资，用空格分隔（例如：Alex 10）：')salary_lists = change_salary.split()   #将用户输入的姓名和工资以列表形式打印if len(salary_lists) < 2:               #判断输入内容格式是否正确 姓名 工资print('\033[31;1m请输入正确格式内容\033[0m')elif not salary_lists[1].isdigit():     #判断输入的工资是否是数字print('\033[31;1m请输入有效工资金额\033[0m')else:_name = salary_lists[0].capitalize()        #定义_name变量存储 用户输入的姓名_salary = salary_lists[1]                   #定义_salary变量存储 用户输入的工资if _name in user_dict:with open('info','r') as f:lines = f.readlines()with open('info','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))breakelse:print('\033[31;1m该用户不存在\033[0m')def add_users():'''定义一个增加员工函数'''while True:user_information()add_user = input('请输入要增加员工姓名和工资，用空格分隔（例如：Eric 100000）：')add_lists = add_user.split()if len(add_lists) < 2:               #判断输入内容格式是否正确 姓名 工资print('\033[31;1m请输入正确格式内容\033[0m')elif not add_lists[1].isdigit():     #判断输入的工资是否是数字print('\033[31;1m请输入有效工资金额\033[0m')else:_name = add_lists[0].capitalize()  # 定义_name变量存储 用户输入的姓名_salary = add_lists[1]  # 定义_salary变量存储 用户输入的工资if _name in user_dict:  #判断输入的姓名是否已存在print('\033[31;1m该用户已存在\033[0m')elif not _name.isalpha():       #判断输入的姓名是否是纯英文print('\033[31;1m请输入正确姓名\033[0m')else:with open('info','a') as f:f.write(_name+' ')f.write(_salary+'\n')print('已将 \033[32;1m%s\033[0m 的信息添加成功' % _name)breakdef del_users():'''定义一个删除员工函数'''while True:user_information()del_user = input('请输入要删除的员工姓名（例如：Alex）：')del_name = del_user.capitalize()if del_name in user_dict:     #将输入的员工姓名首字母变大写，方便用户输入confirm_del = input('是否确认删除\033[32;1m%s\033[0m员工信息>>>任意键确认删除、按N不删除:'%del_user.capitalize())if confirm_del.capitalize() != 'N':         #判断用户选择确认删除list = []                               #定义一个空列表存储不需要删除员工信息with open('info','r') as f:lines = f.readlines()for line in lines:if not re.search(del_name,line):   #判断哪些员工不需要删除   del_name为确认删除的员工
                            list.append(line)with open('info','w') as f:f.writelines(list)                      #将不需要删除员工的信息列表重新写入 info文本中print('\033[32;1m%s\033[0m的员工信息已删除'%del_user.capitalize())breakelse:print('\033[31;1m已取消删除\033[0m')else:print('\033[31;1m该用户不存在\033[0m')user_operations()