#!/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()