员工管理系统

user_dict = {}


def userid(user_id, remark):
    if user_id in user_dict:
        input_code = input('请输入密码：').strip()
        if input_code == user_dict.get(user_id).get('密码'):
            print('密码正确')
            return user_id
        else:
            print('密码错误')
            # input_id = input('请输入员工编号').strip()
            # return userid(input_id, remark)
    else:
        print('请输入正确编号')
        # input_id = input('请输入员工编号').strip()
        # return userid(input_id, remark)


def check_isdigit(a, remark):
    if a.isdigit():
        return int(a)
    else:
        input.a = input('请输入%s:' % (remark)).strip()
        return check_isdigit(input.a, remark)


def add_user():
    user_id = input('请输入员工编号：')
    if user_id not in user_dict:
        name = input('请输入名字：').strip()
        code = input('请输入密码：').strip()
        age = input('请输入年龄：').strip()
        salary = input('请输入薪水：').strip()
        user_dict1 = {
            '姓名': name,
            '密码': code,
            '年龄': age,
            '薪水': salary,
        }
        user_dict[user_id] = user_dict1
        print(user_dict)
    else:
        print('员工编号已存在')


def updata_user():
    user_id = userid(input('请输入员工编号：').strip(), '员工编号')

    print('''
                    1.修改年龄
                    2.修改薪水
                    ''')
    choice = input('请输入功能编号：').strip()
    if choice == '1':
        new_age = check_isdigit(input('请输入修改年龄：').strip(), '员工年龄')
        user_dict.get(user_id)['年龄'] = new_age
        print(f'编号{user_id}修改为', user_dict[user_id])

    elif choice == '2':
        new_salary = check_isdigit(input('请输入新的薪水：').strip(), '员工薪水')
        user_dict.get(user_id)['薪水'] = new_salary
        print(f'编号{user_id}修改为', user_dict[user_id])
    else:
        print('请输入正确编号')


def del_user():
    user_id = userid(input('请输入员工编号：').strip(), '员工编号')
    res = user_dict.pop(user_id)
    print(f'编号{user_id}员工已删除', res)


def find_user():
    user_id = userid(input('请输入员工编号：').strip(), '员工编号')
    print(f"""
                ------------emp info------------
                        员工编号:{user_id}
                        员工姓名:{user_dict.get(user_id).get('姓名')}
                        员工年龄:{user_dict.get(user_id).get('年龄')}
                        员工薪资:{user_dict.get(user_id).get('薪水')}
                ---------------end--------------
                """)


def find_all_user():
    for k in user_dict:
        print(f"""
            ------------emp info------------
                    员工编号:{k}
                    员工姓名:{user_dict.get(k).get('姓名')}
                    员工年龄:{user_dict.get(k).get('年龄')}
                    员工薪资:{user_dict.get(k).get('薪水')}
            ---------------end--------------
            """)


func_dict = {
    '1': add_user,
    '2': updata_user,
    '3': del_user,
    '4': find_user,
    '5': find_all_user,

}

while True:
    print('''
    1.添加员工信息
    2.修改员工信息
    3.删除员工信息
    4.查看单个员工信息
    5.查看所有员工信息
    6.退出
    ''')
    choice = input('请输入功能指令：').strip()
    if choice in func_dict:
        func_dict.get(choice)()
    else:
        if choice == '6':
            break
        else:
            print('请输入正确功能指令')