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

?