python实现简易工资管理系统（Salary Manage）源码

一、需求：

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()