CMD版HR人力资源管理理系统

print("欢迎来到HR人力资源管理理系统")
menu = ("1:查看员工信息", "2:添加员工信息", "3:修改员工信息", "4:删除员工信息", "Q:退出")
def menu_op():                                      #定义菜单函数,选择操作
    print(menu)
    msg = input("请做出你的选择或退出:")
    if msg=="1":
        look()
    elif msg=="2":
        add()
    elif msg=="3":
        amend()
    elif msg=="4":
        re()
    elif msg.upper()=="Q":
        exit()
    else:
        print("请重新输入!!")
        menu_op()
def look():
    with open("emp.db", "r+", encoding="utf-8") as f:
        people = f.readlines()                              #每行以列表形式,读取文件所有内容
        #print(people)
        msg = input("请输入你要查看的员工ID(Q进入菜单):")
        if msg.isdigit():
            for i in people:                                #
                if i[0] == msg:                             #根据ID选择列表内员工
                    print(i)
            look()
        elif msg.upper()=="Q":
            menu_op()
        else:
            print("请正确输入!!")
def add():
    while True:
        with open("emp.db","r+",encoding="utf-8") as f:
            people = f.readlines()
            f.seek(0)
            print(people)
            msg = input("请输入您的ID,姓名,薪水(用逗号分隔):")
            #print(msg.split(",")[0])
            if msg.upper()=="Q":
                menu_op()
            elif len(msg.split(",")) !=3:
                print("你的输入有误,请重新输入!!")
            else:
                for i in people:
                    if i.strip()==msg:
                        print("你输入的员工已存在!!")
                        add()
                if len(people) < 1 or i.strip()!=msg:
                    f.seek(0,2)
                    f.write(msg + "
")
                    f.flush()
                    f.close()
                    add()
def amend():
    with open("emp.db", "r+", encoding="utf-8") as f:
        people = f.readlines()
        print(people)
        msg = input("请输入你要修改的员工ID(Q退出):")
        if msg.isdigit():
            for i in people:
                if i.split(",")[0] == msg:
                    print(i)
            msg1 = input("请重新输入您的ID,姓名,薪水(用逗号分隔,Q退出):")
            if msg1.upper() == "Q":
                amend()
            elif len(msg1.split(",")) != 3:
                print("你的输入有误,请重新输入!!")
            else:
                for i in people:
                    if msg==i[0]:
                        people.remove(i)
                        people.append(msg1+"
")
                        f.seek(0)
                        f.write("".join(people))
                        f.flush()
                        f.close()
                        amend()
        elif msg.upper() == "Q":
            menu_op()
def re():
    with open("emp.db", "r+", encoding="utf-8") as f:
        people = f.readlines()
        print(people)
        msg = input("请输入你要删除的员工ID(Q退出):")
        if msg.isdigit():
            for i in people:
                if msg == i[0]:
                    people.remove(i)
            #people[int(msg)-1] = msg1
                    f.seek(0)
                    f.truncate()
                    f.write("".join(people))
                    f.flush()
                    f.close()
                    re()
        elif msg.upper()=="Q":
            menu_op()
        else:
            print("请重新输入!!")
menu_op()