Python工资管理系统

实验名称：工资管理系统
实验要求：
Alex 100000
Rain 80000
Egon 50000
Yuan 30000
-----以上是info.txt文件-----
实现效果：
从info.txt文件中读取员工及其工资信息，最后将修改或增加的员工工资信息也写入原info.txt文件。
效果演示：

1. 查询员工工资

2. 修改员工工资

3. 增加新员工记录

4. 删除员工信息

5. 退出

   >:1
   请输入要查询的员工姓名（例如：Alex）：Alex
   Alex的工资是：100000。

6. 查询员工工资

7. 修改员工工资

8. 增加新员工记录

9. 删除员工信息

10. 退出

    >:2
    请输入要修改的员工姓名和工资，用空格分隔（例如：Alex 10）：Alex 10
    修改成功！

11. 查询员工工资

12. 修改员工工资

13. 增加新员工记录

14. 删除员工信息

15. 退出

    >:3
    请输入要增加的员工姓名和工资，共空格分割（例如：Eric 100000）：Eric 100000
    增加成功！

16. 查询员工工资

17. 修改员工工资

18. 增加新员工记录

19. 删除员工信息

20. 退出

    >:4
    请输入要增加的员工姓名和工资，共空格分割（例如：Eric 100000）：Eric 100000
    增加成功！

21. 查询员工工资

22. 修改员工工资

23. 增加新员工记录

24. 删除员工信息

25. 退出

    >:5
    再见！

代码实现：

#!/usr/bin/env python

#-*- coding:utf-8 -*-

import os

######查询操作######

def select():

    name=input("请输入要查询的员工姓名（例如：Alex）:").strip() #空格分段

    flag=False #判断条件

    with open("info.txt","r")as reads:

        for line in reads: #遍历

            if line.split()[0]==name:  ###[alex 1000],[Yuan 9999][0]==name

                flag=True

                print("员工信息",name,line.split()[1])  #[alex 1000][1]==1000

            else:  ###跳过，知道遍历结束为查询到，执行if not flag的判断

                continue

    if not flag:

        print("\033[31;1m未找到%s员工信息\033[0m"%name)

#####修改操作######

def revise():

    reff=input("请输入要修改的员工姓名和工资，用空格分隔（例如：Alex 10）：").strip()  ##reff: "Alex 10000"

    reff_list=reff.split() #reff_list : ['Alex','1000']  class list

    flag=False

    with open("info.txt","r")as reads,open("info",'w')as writes:

        for line in reads: #遍历  line "alex 10000'

            if line.split()[0] == reff_list[0]:   ##line "alex" == reff_list ["alex",""]

                writes.write(reff+"\n")   #修改成功，True

                flag=True

            else:###

                writes.write(line)

    os.remove("info_bak.txt")

    os.rename("info.txt","info_bak.txt")

    os.rename("info","info.txt")

    if flag:

        print("修改成功")

    else:

        print("未找到需要修改的\033[31;1m%s\033[0m员工信息" % reff)

def app():

    rtff=input("请输入要增加的员工姓名和工资，共空格分割（例如：Eric 100000）：").strip() ##以空格作为分隔符

    rtff_list=rtff.split()

    flag=False

    with open("info.txt","r")as reads:

        for line in reads:

            if line.split()[0] == rtff_list[0]: ##当相同时，print 以重复。

                flag = True

            else:

                continue

    if flag:

        print("\033[31;1m该%s用户已存在\033[0m"%rtff)

    else:

        with open("info.txt", "a") as writes:

            writes.write("\n"+rtff)

        print("增加成功")

######删除操作#######

def delete():

    name=input("请输入要删除的员工姓名（例如：Alex）:").strip()

    flag=False

    with open("info.txt","r") as reads,open("info","w")as writes:

        for line in reads:

            if line.split()[0]==name:

                flag=True

                continue

            else:

                writes.write(line)

    os.remove("info_bak.txt")

    os.rename("info.txt", "info_bak.txt")

    os.rename("info", "info.txt")

    if flag:

        print("删除成功")

    else:

        print("未找到需要修改的\033[31;1m%s\033[0m员工信息" % name)

#####菜单#####

def main():

    menu={

        "1":select,

        "2":revise,

        "3":app,

        "4":delete,

        "5":exit,

    }

    while True:

        print("""

        请选择：

            1,查询员工信息

            2,修改员工工资

            3,增加员工信息

            4,删除员工信息

            5,退出工资系统

        """)

        choice=input(">>>").strip()  ##choice :"1,2,3,4"

        if choice in menu:  ###这里是menu字典的key值等于choice

            menu[choice]() ###进入对应的value

if __name__ == "__main__":

    main()