integrador con dicc

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
La lista de alumnos que creamos en el módulo
anterior ahora debe ser un diccionario, en donde
las claves serán nombres de alumnos y los
valores sus respectivas cantidades de cursos.
"""



#################  funciones  ##############################

def menu():
    return"""
                Menu de opciones
    ---------------------------------------
    1 - Ver la lista de alumnos.
    2 - Añadir un alumno a la lista.
    3 - Ver cantidad de cursos de un alumno
    4 - Salir.
    """


def imprimir(alumnos):
    if alumnos:
        for nombre,cursos in alumnos.items():
            print(f"{nombre} - {cursos} cursos")
    else:
        print("No hay alumnos")


def ingresar_alumno(alumnos):
    nombre = input("Nombre: ")
    cursos = int(input("Cursos: "))
    alumnos[nombre] = cursos
    return alumnos



def ver_cursos(alumnos):
    if not alumnos:
        return f"No hay alumnos"
    nombre = input("Nombre: ")
    for alumno,cursos in alumnos.items():
        if alumno == nombre:
            return f"Cursos: {cursos}"
    return f"No existe el alumno {nombre}"


############################################################

# alumnos = {"Juan":3, "Ana":4, "Luisa":6}
alumnos = {}


while True:
    print(menu())
    opcion = input("Seleccione una opcion: ")

    if opcion == "1":
        imprimir(alumnos)


    elif opcion == "2":
        alumnos = ingresar_alumno(alumnos)
        print("El alumno ha sido añadido exitosamente")


    elif opcion == "3":
        print(ver_cursos(alumnos))



    elif opcion == "4":
        print("¡Gracias por utilizar el programa!")
        break

    else:
        print("Opcion incorrecta")