python 函数练习

coding:utf-8

当前系统日期:2022/8/5

函数练习

学员管理系统

学员信息：学号、名字、手机号

增删改查学员、退出系统

打印要操作的信息

输入对应的数字表示使用的功能

1：增加 2：删除 3：修改 4：查看 5：退出

user_info = {}

def print_info() -> str:
    """
    系统功能也没
    :return:功能选择 str
    """
    print('-'*30)
    print('1：增加 2：删除 3：修改 4：查询 5：退出')
    print('-' * 30)
    return input('请输入数字，选择对应的功能:')

def add_info():
    """
    增加学员信息
    :return: bool
    """
    user_id = input('请输入学员的学号：')
    if verify_info(user_id):
        print(f'学号为：{user_id}的学员已存在，请重新选择对应的功能')
    else:
        user_name = input('请输入学员的姓名：')
        user_phone = input('请输入学员的手机号：')
        user_info[user_id] = {'user_name': user_name, 'user_phone': user_phone}
        print(f'学员增加成功，学员的学号为：{user_id} -> {user_info[user_id]}')

def verify_info(user_id):
    """
    确认学号是否存在
    :param user_id:学号
    :return: bool
    """
    if user_id in user_info.keys():
        return True
    else:
        print(f'学号为：{user_id}的学员未录入')
        return False

def search_info():
    """
    查询学员信息
    :return: bool
    """
    print('-'*30)
    print('1：查询单个学员的信息 2：查询全部学员的信息')
    print('-' * 30)
    data = input('请输入数字，选择对应的查询功能:')
    if data == '1':
        user_id = input('请输入学员的学号：')
        if verify_info(user_id):
            print(f'学员的学号为：{user_id}，学员的信息为{user_info[user_id]}')
    elif data == '2':
        print(f'所有学员的信息为：{user_info.items()}')
    else:
        print('功能选择错误，请重新选择')

def del_info():
    """
    删除学员信息
    :return: bool
    """
    user_id = input('请输入学员的学号：')
    if verify_info(user_id):
        del user_info[user_id]
        print(f'学员的学号为：{user_id}的学员信息已删除')
        verify_info(user_id)  # 再次确认删除成功

def update_info():
    """
    修改学员信息
    :return: bool
    """
    user_id = input('请输入学员的学号：')
    if verify_info(user_id):
        print('-' * 30)
        print('1：修改姓名 2：修改手机号')
        print('-' * 30)
        _ = input('请输入数字，选择对应的功能:')
        if _ == '1':
            user_name = input('请输入学员的姓名：')
            user_info[user_id]['user_name'] = user_name
            print(f'学员的学号为：{user_id}的学员信息已修改为->{user_info[user_id]}')
        elif _ == '2':
            user_phone = input('请输入学员的手机号：')
            user_info[user_id]['user_phone'] = user_phone
            print(f'学员的学号为：{user_id}的学员信息已修改为->{user_info[user_id]}')
        else:
            print('功能选择错误，请重新选择')

if __name__ == '__main__':  # 程序的入口

    while True:
        value = print_info()
        if value == '1':
            add_info()  # 调用增加信息函数
        elif value == '2':
            del_info()  # 调用删除信息函数
        elif value == '3':
            update_info()  # 调用修改信息函数
        elif value == '4':
            search_info()  # 调用查询信息函数
        elif value == '5':
            break  # 结束程序运行
        else:
            print('功能选择错误，请重新选择')