员工信息查询程序
1、运行程序
2、选择操作选项:1、查询员工工资 2、修改员工工资 3、增加新员工记录 4、退出系统
流程图
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Mego
def file_test():
with open("yuangong.txt",'r',encoding="utf-8") as f:
file = list(f)
return file
def server():
for i in range(1):
print("********************员工工资系统**********************")
print("\033[94;1m1.查询员工工资\n2.修改员工工资\n3.增加新员工记录\n4.退出\033[0m")
print("*****************************************************")
return i
loginSucces = False
while True:
i = server()
serial_number = input("\033[31;1m请选择你的服务:\033[0m")
if serial_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 in file:
file = i.strip().split()
if name in file:
break
for i in file:
if name in file:
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 in file:
file = ii.strip().split()
if name_old in file:
print(file)
sum_old = file[1]
name_alter,sum_alter = input("\033请输入修改的新员工姓名和工资,用空格分割:").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 in lines:
if name_old in line:
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 in lines:
if name_alter in line:
line = line.replace(sum_old,sum_alter)
f_b.write(line)
f_b.close()
print("\033[31;1m修改成功\033[0m")
break
if loginSucces == True: break
else:
print("\033[31;1m没有此用户!\033[0m\n")
elif serial_number == 3:
try:
name_new, sum_new = input("\033[31;1m请输入要增加的员工姓名和工资,用空格分割:\033[0m").strip().split()
except ValueError:
print("\033[31;1m输入的参数不够!\033[0m\n")
else:
if name_new.isalpha() and sum_new.isdigit():
for ii in file:
file = ii.strip().split()
if name_new not in file:
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")