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