python员工管理系统1.0

def main():
    while True:
        menu()
        choice = input('选择功能').strip()
        if not choice.isdigit():
            print('请输入数字')
            continue
        if choice == '1':
            add()  # 添加信息
        elif choice == '2':
            search()  # 查询特定
        elif choice == '3':
            modify()  # 修改薪资
        elif choice == '4':
            search_all()  # 查询所有
        elif choice == '5':
            delete()  # 删除特定
        elif choice == '0':
            break


import sys


def gohome():
    answer = input('是否回到主菜单？Y/N')
    if answer == 'y' or 'Y':
        return main()
    elif answer == 'n' or 'N':
        sys.exit()


def add():  # 添加信息
    # 编号，姓名，年龄，薪资
    user_id = input('输入编号：').strip()
    with open('user_data_list.txt', 'r', encoding='utf-8') as f1:
        for line in f1:
            real_user_id = line.split('|')
            if real_user_id[0] == user_id:
                print('该编号已存在')
                add()
        name = input('输入姓名：').strip()
        age = input('输入年龄：').strip()
        salary = input('输入薪水：').strip()

    user_data = '%s|%s|%s|%s|\n' % (user_id, name, age, salary)
    with open('user_data_list.txt', 'a', encoding='utf-8') as f:
        f.write(user_data)
    print('%s添加成功' % name)
    gohome()


def search():
    user_id = input('输入编号：').strip()
    with open('user_data_list.txt', 'r', encoding='utf-8') as f1:
        i = 0
        for line in f1:
            real_user_id = line.split('|')
            i += 1
            file = open('user_data_list.txt')
            res = len(file.readlines())

            if real_user_id[0] == user_id:
                print('编号:' + str(real_user_id[0]),
                      '姓名:' + str(real_user_id[1]),
                      '年龄:' + str(real_user_id[2]),
                      '薪资:' + str(real_user_id[3]))
                gohome()
            elif res > i:
                continue
            elif res == i:
                print('查无此人')
        gohome()


def modify():  # 修改薪资
    user_id = input('输入编号：').strip()
    with open('user_data_list.txt', 'r', encoding='utf-8') as f1:
        i = 0
        for line in f1:
            real_user_id = line.split('|')
            i += 1
            file = open('user_data_list.txt')
            res = len(file.readlines())
            res1 = real_user_id[3]

            if real_user_id[0] == user_id:
                new_salary = input('新薪资：').strip() + '\n'
                with open('user_data_list.txt.', 'r', encoding='utf-8') as f:
                    data = f.read()
                with open('user_data_list.txt.', 'w', encoding='utf-8') as f:
                    f.write(data.replace(res1, new_salary))
                gohome()
            elif res > i:
                continue
            elif res == i:
                print('查无此人')
            gohome()


def search_all():  # 查询所有
    with open('user_data_list.txt', 'r', encoding='utf-8') as f1:
        for line in f1:
            real_user_id = line.split('|')
            print('编号:' + str(real_user_id[0]),
                  '姓名:' + str(real_user_id[1]),
                  '年龄:' + str(real_user_id[2]),
                  '薪资:' + str(real_user_id[3]))
            gohome()

def delete():  # 删除特定
    user_id = input('输入编号：').strip()
    with open('user_data_list.txt', 'r', encoding='utf-8') as f1:
        i = 0
        for line in f1:
            real_user_id = line.split('|')
            i += 1
            file = open('user_data_list.txt')
            res = len(file.readlines())

            if real_user_id[0] == user_id:
                with open('a.txt', 'r') as f:
                    lines = f.readlines()
                with open('a.txt', 'w') as f:
                    f.writelines(line for line in lines if user_id not in line)
                gohome()
            elif res > i:
                continue
            elif res == i:
                print('查无此人')
                gohome()

def menu():
    print('员工管理系统')
    print('功能菜单')
    print('1.添加员⼯信息')
    print('2.查询特定员⼯')
    print('3.修改员⼯薪资')
    print('4.查询所有员⼯')
    print('5.删除特定员⼯')
    print('0.退出')


main()