员工信息管理系统函数封装版

def add():
    print('开始添加用户')
    info_dic = {}
    user_id = input('请输入用户编号>>>:').strip()
    # 判断用户编号是否存在
    for i1 in info_lis:
        if user_id in i1['user_id']:
            print('用户编号已存在，请重新输入')
            return 1
    else:
        name = input('请输入名字>>>:').strip()
        age = input('请输入年龄>>>:').strip()
        salary = input('请输入薪资>>>:').strip()
        info_dic['user_id'] = user_id
        info_dic['name'] = name
        info_dic['age'] = age
        info_dic['salary'] = salary
        info_lis.append(info_dic)
        print('用户：{} 添加成功'.format(name))
        return 1


def ls():
    print('开始查看用户')
    # 若用户列表为空，提示用户
    if len(info_lis) == 0:
        print('暂无用户可查看。。。')
        return 1
    # 用户列表不为空执行查询功能
    else:
        while True:
            ls1 = input('选择：1.查看全部用户/2.查看指定用户>>>:').strip()
            if ls1 == '1':
                for index in range(0, len(info_lis)):
                    print(info_lis[index])
                break
            elif ls1 == '2':
                choice_id = input('请输入想要查看的用户id>>>:').strip()
                for user in info_lis:
                    if user['user_id'] == choice_id:
                        print(user)
                        break

                else:
                    print('输入用户不存在')
                    continue
                return 1
            else:
                print('输入有误，请重新输入')
                continue
        return 1


def del_info():
    print('开始删除用户')
    # 若用户列表为空，提示用户
    if len(info_lis) == 0:
        print('暂无用户可删除。。。')
        return 1
    # 用户列表不为空执行删除功能
    else:
        del_id = input('请输入想要删除的用户id>>>:').strip()
        for i2 in info_lis:
            if i2['user_id'] == del_id:
                info_lis.remove(i2)
                print('已删除')
                return 1
        print('用户不存在')
        return 1


def salary_adjust():
    # 若用户列表为空，提示用户
    if len(info_lis) == 0:
        print('暂无用户可修改。。。')
        return 1
    # 用户列表不为空执行修改功能
    else:
        change_id = input('请输入想要修改的用户id>>>:').strip()
        change_salary = input('薪资修改为>>>:').strip()
        for i3 in info_lis:
            if i3['user_id'] == change_id:
                i3['salary'] = change_salary
                print('薪资已修改')
                return 1
        else:
            print('更改失败，用户不存在')
        return 1


def quit_sys():
    print('已退出系统！！！')
    return


# 构造功能字典
func_dic = {'1': ['添加用户', add], '2': ['查看用户', ls], '3': ['删除用户', del_info], '4': ['修改薪资', salary_adjust],
            '5': ['退出系统', quit_sys]}

# 构造用户信息列表
info_lis = []

flag = True
while flag:
    # 打印初始界面
    for i in range(1, len(func_dic) + 1):
        print(i, func_dic.get(str(i))[0])

    choice = input('请输入想要的功能编号>>>:').strip()
    while choice.isdigit() and choice in func_dic:
        res = func_dic.get(choice)[1]()
        if res:
            break
        else:
            flag = False
            break
    else:
        print('输入有误，请重新输入')
        continue