integrador con funcion

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

# ejercicio integrador

"""
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.

Para esto se debe modificar el código de las opciones 1 y 2 (agregar un
nuevo alumno y ver la lista de alumnos).

La tercera opción será “Ver la cantidad de cursos de un alumno”. Deberá
solicitar el nombre de un alumno e imprimir en pantalla el número de
cursos que tiene asociados como clave.

La cuarta opción es la de salir, como en la versión anterior.

Usar todo lo aprendido hasta el momento, el programa no debe de frenar
de forma imprevista a causa de un error. Ya que en el material de lectura
se vieron todas las posibles soluciones frente a los problemas que se
puedan presentar.

"""
# Estrategia de resolución:
# 1) construir el template while True
# 2) Diagramar mi estructura de datos


# alumnos = {"Juan":3, "Ana":4, "Victoria":2}
alumnos = {}

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 not alumnos:
        print("No hay alumnos")
    else:
        print("Lista de alumnos:")
        for k,v in alumnos.items():
            print(f"{k} - {v} cursos")




def ingresar(alumnos):
    nombre = input("Nombre: ")
    while True:
        cursos = input("Cursos: ")
        if cursos.isdecimal() and int(cursos) != 0:
            break
        else:
            print("Error: debe ingresar un entero positivo")
    alumnos[nombre] = cursos
    return alumnos


def cantidad_de_cursos(alumnos):
    nombre = input("Nombre: ")
    nombres = list(alumnos.keys())
    if nombre in nombres:
        return f"Cantidad de cursos: {alumnos[nombre]}"
    else:
        return f"No existe el alumno {nombre}"




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

while True:

    print(menu())

    opcion = input("Seleccione una opción: ")

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


    elif opcion == "2":
        alumnos = ingresar(alumnos)
        print("¡El alumno fue añadido a la lista!")


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



    elif opcion == "4":
        print("Gracias por usar este programa ...")
        break

    else:
        print("Opción incorrecta")