HR系统

#  ("查看员⼯信息","添加员⼯信息", "修改员⼯信息", "删除员⼯信息", "退出")
import time
import os

def log(content):
    f = open("emp.log", mode="a", encoding="utf-8")
    f.write(content+"\n")
    f.flush()
    f.close()

def chakan(): # 第二个
    print("编号\t\t名字\t\t生日\t\t收入\t\t录入时间")
    f = open("emp.db", mode="r", encoding="utf-8")
    for line in f:
        lst = line.strip().split("_")
        print(lst[0]+"\t"+lst[1]+"\t"+lst[2]+"\t"+lst[3]+"\t"+lst[4])
    print("查看完毕!!!!")

def tianjia(): # 第一个
    flag = True
    while flag:
        id = input("请输入员工的id:")
        # 判定id是否可用
        f = open("emp.db", mode="r", encoding="utf-8")
        for line in f:
            if id == line.strip().split("_")[0]:
                print("对不起. id已经重复了. 不可用. 请重新输入")
                break
        else:
            print("id不重复,可以使用")
            flag = False

    name = input("请输入员工的name:")
    birthday = input("请输入员工的birthday:")
    salary = input("请输入员工的salary:")
    input_time = time.strftime("%Y-%m-%d %H:%M:%S") # 获取系统时间

    f = open("emp.db", mode="a", encoding="utf-8")
    f.write(id+"_"+name+"_"+birthday+"_"+salary+"_"+input_time+"\n")
    f.flush()
    f.close()
    print("添加成功!!!")

    log(f"管理员xxx在{input_time}添加了一个员工, 该员工信息为: {id}_{name}_{birthday}_{salary}_{input_time}")

def xiugai(): # 第四个
    flag = True
    while flag:
        id = input("请输入你要修改的员工的id:")

        f = open("emp.db", mode="r", encoding="utf-8")
        for line in f:
            lst = line.strip().split("_")
            if id == lst[0]:
                print("当前员工的信息是: %s\t%s\t%s\t%s\t%s" % (lst[0], lst[1], lst[2],lst[3],lst[4]) )
                flag = False
                break
        else:
            print("id不存在,请重新输入")
    f.flush()
    f.close()

    salary = input("请输入该员工的最新工资")

    with open("emp.db", mode="r", encoding="utf-8") as f1, \
        open("emp.db_副本", mode="w", encoding="utf-8") as f2:
        for line in f1:
            lst = line.strip().split("_")
            if id == lst[0]:
                line = f"{id}_{lst[1]}_{lst[2]}_{salary}_{time.strftime('%Y-%m-%d %H:%M:%S')}"
            f2.write(line) # 修改

    os.remove("emp.db")
    os.rename("emp.db_副本", "emp.db")

    print("修改员工信息完毕!!!")

def shanchu(): # 第三个
    id = input("请输入你要删除的员工的id")
    with open("emp.db", mode="r", encoding="utf-8") as f1, \
        open("emp.db_副本", mode="w", encoding="utf-8") as f2:
        for line in f1:
            if id == line.strip().split("_")[0]:
                continue
            f2.write(line) # 不要删除的数据

    os.remove("emp.db")
    os.rename("emp.db_副本", "emp.db")
    print("删除完毕!!!!")


menu = ("查看员工信息","添加员工信息", "修改员工信息", "删除员工信息", "退出")

while 1:
    for i in range(len(menu)):
        print(i + 1, menu[i])

    n = input("请输入你要执行的菜单:")
    if n == '1':
        chakan()
    elif n == '2':
        tianjia()
    elif n == "3":
        xiugai()
    elif n == "4":
        shanchu()
    elif n == "5":
        print("程序退出")
        exit()
    else:
        print("对不起. 您输入的菜单不存在. 请重新输入")