练习5：学员管理系统

# 需求：进入系统显示系统功能界面，功能如下：
# 1. 添加学员(把学员信息存到字典中，再把字典放到列表中。[{}])
# 2. 删除学员
# 3. 修改学员信息
# 4. 查询学员信息
# 5. 显示所有学员信息
# 6. 退出系统
# 系统共6个功能，用户根据⾃⼰需求选取。
# 根据⽤用户输入的功能序号，执行不同的功能(函数)
#     1. 定义函数
#     2 .调⽤用函数

# 步骤分析：
# 1. 显示功能界⾯面
def student_info():
    print('显示系统功能-------------------------------')
    print('1. 添加学员信息')
    print('2. 删除学员信息')
    print('3. 修改学员信息')
    print('4. 查询学员信息')
    print('5. 显示所有学员信息')
    print('6. 退出系统')
    print('-' * 50)

info = []

# 添加学员
def add_info():
    '''添加学员函数'''
    new_name = input('请输入姓名：')
    new_id = input('请输入学号：')
    new_tel = input('请输入手机号：')
    global info

    for i in info:
        if new_name == i['name']:
            print('该用户已存在！！')
            return  # 退出当前函数，后面添加信息的代码不执行

    info_dict = {}
    info_dict['id'] = new_id
    info_dict['name'] = new_name
    info_dict['tel'] = new_tel
    info.append(info_dict)
    print(info)

# 删除学员
def del_info():
    """删除学员"""
    del_name = input('要删除的学员姓名：')
    global info
    for i in info:
        if i['name'] == del_name:
            info.remove(i)
            break
    else:
        print('该学员不存在！！！')
    print(info)

# 修改学员信息
def modify_info():
    """修改学员函数"""
    modify_name = input('请输入要修改的姓名：')
    global info
    for i in info:
        if i['name'] == modify_name:
            i['tel'] = input('请输入新的手机号：')
            break
    else:
        print('该学员不存在！！！')
    print(info)

# 查询学员
def search_info():
    """查询学员函数"""
    search_name = input('请输入要查询的学员姓名：')
    for i in info:
        if search_name == i['name']:
            print('学员信息---------------------')
            print(f"姓名是{i['name']}, 学号是{i['id']}, 手机号是{i['tel']}")
            break
    else:
        print('该学员不存在！！！')
    print(info)

#显示所有学员信息
def all_info():
    """显示所有学员函数"""
    print('学号\t姓名\t手机号')
    for i in info:
        print(f"{i['id']}\t{i['name']}\t{i['tel']}")



# 2. 用户输入功能序号, 系统功能需要循环使用，直到6才退出
while True:
    student_info()
    student_num = int(input('请输入功能序号：'))

    if student_num ==  1:
        # print('添加学员信息')
        add_info()
    elif student_num == 2:
        # print('删除学员信息')
        del_info()
    elif student_num == 3:
        # print('修改学员信息')
        modify_info()
    elif student_num == 4:
        # print('查询学员信息')
        search_info()
    elif student_num == 5:
        # print('显示所有学员信息')
        all_info()
    elif student_num == 6:
        # print('退出系统')
        exit_flag = input('确定要退出吗？yes  or no ')
        if exit_flag == 'yes':
            break
    else:
        print('输入错误，请重新输入！！！')