实验名称：工资管理系统
实验要求：
Alex 100000
Rain 80000
Egon 50000
Yuan 30000
-----以上是info.txt文件-----
实现效果：
从info.txt文件中读取员工及其工资信息，最后将修改或增加的员工工资信息也写入原info.txt文件。
效果演示：
- 查询员工工资
- 修改员工工资
- 增加新员工记录
- 删除员工信息
- 退出
:1
请输入要查询的员工姓名（例如：Alex）：Alex
Alex的工资是：100000。 - 查询员工工资
- 修改员工工资
- 增加新员工记录
- 删除员工信息
- 退出
:2
请输入要修改的员工姓名和工资，用空格分隔（例如：Alex 10）：Alex 10
修改成功！ - 查询员工工资
- 修改员工工资
- 增加新员工记录
- 删除员工信息
- 退出
:3
请输入要增加的员工姓名和工资，共空格分割（例如：Eric 100000）：Eric 100000
增加成功！ - 查询员工工资
- 修改员工工资
- 增加新员工记录
- 删除员工信息
- 退出
:4
请输入要增加的员工姓名和工资，共空格分割（例如：Eric 100000）：Eric 100000
增加成功！ - 查询员工工资
- 修改员工工资
- 增加新员工记录
- 删除员工信息
- 退出
: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()
