deffile_test():
with open("yuangong.txt",'r',encoding="utf-8") as f:
file=list(f)returnfiledefserver():for i in range(1):print("1.查询员工工资\n2.修改员工工资\n3.增加新员工记录\n4.退出")returni
loginSucces=FalsewhileTrue:
i=server()
serial_number= input("请选择你的服务:")ifserial_number.isdigit():
serial_number=int(serial_number)
file=file_test()if serial_number < 4 and serial_number >=0:if serial_number == 1:
name= input("请输入要查询的员工姓名:").strip()for i infile:
file=i.strip().split()if name infile:break
for i infile:if name infile:
name_sum= file[1]print("\033[32;1m%s的工资是:%s\033[0m\n"%(name,name_sum))break
else:print("\033[31;1m没有此用户!\033[0m\n")elif serial_number == 2:
name_old= input("请输入要修改的员工姓:").strip()for ii infile:
file=ii.strip().split()if name_old infile:print(file)
sum_old= file[1]
name_alter,sum_alter= input("请输入修改的新员工姓名和工资,用空格分割:").strip().split()
with open("yuangong.txt","r",encoding="utf-8") as f:
lines=f.readlines()
with open("yuangong.txt","w",encoding="utf-8") as f_a:for line inlines:if name_old inline:
line=line.replace(name_old,name_alter)
f_a.write(line)
f_a.close()
with open("yuangong.txt", "r", encoding="utf-8") as f:
lines=f.readlines()
with open("yuangong.txt", "w", encoding="utf-8") as f_b:for line inlines:if name_alter inline:
line=line.replace(sum_old,sum_alter)
f_b.write(line)
f_b.close()print("修改成功")break
if loginSucces == True: break
else:print("\033[31;1m没有此用户!\033[0m\n")elif serial_number == 3:try:
name_new, sum_new= input("请输入要增加的员工姓名和工资,用空格分割:").strip().split()exceptValueError:print("\033[31;1m输入的参数不够!\033[0m\n")else:if name_new.isalpha() andsum_new.isdigit():for ii infile:
file=ii.strip().split()if name_new not infile:
with open("yuangong.txt", "a+", encoding="utf-8") as f_c:
f_c.write("\n%s %s"%(name_new,sum_new))print("增加成功")break
if loginSucces == True: break
else:print("\033[31;1m参数类型不对!\033[0m\n")elif serial_number == 4:
exit()else:print("\033[31;1m请输入正确的序号!\033[0m\n")else:print("\033[31;1m输入的不是整数!\033[0m\n")