Python王者荣耀系统

'''
王者荣耀管理系统：
角色：姓名，性别，职业

添加角色
删除角色
修改角色
查询角色
显示所有角色
退出系统
'''
import time

all_role = []  # 存放所有角色的容器
print('~~~~~~~~~~~~~~~~~~~~欢迎来到王者荣耀角色管理系统~~~~~~~~~~~~~~~~~~~~~~~~~')
while True:
    select = input('请选择功能:\n 1.添加角色 \n 2.删除角色 \n 3.修改角色'
                   '\n 4.查询角色 \n 5.显示所有角色 \n 6.退出系统\n')

    if select == '1':
        print('添加角色：')
        name = input('\t请输入角色姓名：')
        gender = input('\t请输入角色性别：')
        job = input('\t请输入角色职业：')
        role = [name, gender, job]
        all_role.append(role)
        print('成功添加了{}到王者荣耀系统\n'.format(name))
    elif select == '2':
        print('删除角色')
        role_name = input('请输入需要删除的角色姓名：')
        for role in all_role:
            if role_name in role:
                ask = input('是否确定删除（y/n）：')
                if ask.upper() == 'Y':
                    all_role.remove(role)
                    print('成功删除{}'.format(role_name))
                    break
                else:
                    break
        else:
            print('本系统不存在角色{}'.format(role_name))
            break
    elif select == '3':
        print('修改角色')
        role_name = input('请输入需要修改的角色的姓名：')
        for role in all_role:
            if role_name in role:
                name = input('请输入修改的姓名：')
                gender = input('请输入修改的性别')
                job = input('请输入修改的职业：')
                role[0] = name
                role[1] = gender
                role[2] = job
                print('修改成功！')
                break
        else:
            print('没有您需要修改的信息')
            break
    elif select == '4':
        print('查询')
        role_name = input('请输入需要查询的角色姓名：')
        for role in all_role:
            if role_name == role[0]:
                print('{}{}{}'.format(role[0], role[1], role[2]))
                break
        else:
            print('您输入的角色不存在！')
            break
    elif select == '5':
        print('显示所有角色信息：')
        print('\t 姓名 \t 性别 \t职业')
        for role in all_role:
            print(role[0].center(11), end='')
            print(role[1].center(9), end='')
            print(role[2].center(8))
    elif select == '6':
        print('正在退出本系统')
        time.sleep(3)
        print('退出成功！！！感谢您的使用')
        break
    else:
        print('输入错误，请重新输入\n')