integrador 2

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

# Ejercicio integrador

# alumnos = {"Juan":5, "Ana":11, "Josefa":3}  alumnos["Juan"] --> da 5

#################################################
def borrar_pantalla():
    import os
    if os.name == "posix":
        os.system("clear")
    else:
        os.system("cls")


def menu():
    return """
    Ingrese el número de la operación que desea ejecutar:
    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:
        print("Lista de alumnos:")
        for k,v in alumnos.items():
            print(f"{k} - {v} cursos")
    else:
        print("No hay alumnos")


def agregar_alumno():
    nombre = input("Ingrese el nombre del alumno: ")
    while True:
        cursos = input("Ingrese la cantidad de cursos: ")
        if cursos.isdecimal() and int(cursos):
            return [nombre, cursos]
        else:
            print("Error en el ingreso de la cantidad de cursos")


def mostrar_cursos(alumnos):
    nombre = input("Ingrese el nombre del alumno: ")
    if nombre in list(alumnos.keys()):
        print(f"{nombre} - {alumnos[nombre]} cursos")
    else:
        print(f"No existe el alumno {nombre}")



########################################################
alumnos = {}

while True:
    print(menu())
    opcion = input(">>> ")

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

    elif opcion == "2":
        nombre, cursos = agregar_alumno()
        alumnos[nombre] = cursos

    elif opcion == '3':
        mostrar_cursos(alumnos)

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

    else:
        print("La opción ingresada no es correcta, vuelva a intentarlo.")

    input("Presione cualquier tecla para continuar")
    borrar_pantalla()