Crear un menú en python (no GUI)

En esta ocasión deseo compartir una forma (muy personal) de realizar menús en Python. Tal y como se puede ver en el título, no me refiero a un menú gráfico sino a uno muy tradicional, de línea de comandos. Debo aclarar que el menú que verán a continuación puede ser implementado sin la utilización de una clase, sin embargo, este enfoque brinda la ventaja de contar con código bien organizado, fácil de leer, depurar y modificar.

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

# menu.py

#----------------------------------------------------------------------- Módulos
import os
import sys

#-------------------------------------------------------------------- Constantes
LIMPIAR = "clear" if sys.platform.startswith("linux") else "cls"

#------------------------------------------------------------------------ Clases
class Menu(object) :
    """Lista de funciones opcionales, las cuales le permitirán al usuario
    realizar operaciones matemáticas básicas.

    Argumentos
        prompt -- Es el prompt del menú (default '>')
    """

    def __init__(self, prompt=">") :
        self.prompt = prompt + " "
        self.opciones = [
            self.opcion_sumar,
            self.opcion_restar,
            self.opcion_multiplicar,
            self.opcion_dividir,
            self.opcion_salir,
        ]
        self.presentacion = "------------------ Bienvenido ------------------\n"

        for numero, opcion in enumerate(self.opciones, 1) :
            self.presentacion += "{0}. {1}\n".format(numero, opcion.__name__[7:])

    def loop(self) :
        while True:
            print self.presentacion

            try:
                seleccion = int(raw_input(self.prompt))

                if seleccion >= 1 and seleccion < len(self.opciones) :
                    op1, op2 = self.pedir_operandos()
                    # Los indices van desde 0 hasta len(self.opciones)-1
                    resultado = self.opciones[seleccion - 1](op1, op2)
                    raw_input("\nEl resultado es {0}".format(resultado))

                # Caso especial para la opción "salir"
                elif seleccion == len(self.opciones) :
                    self.opciones[seleccion - 1]()

                else:
                    raw_input("Error: Opción invalida")

            except ValueError:
                raw_input("Error: Debes introducir un número")

            except ZeroDivisionError:
                raw_input("Acaso quieres destruir el universo!?")

            except KeyboardInterrupt:
                break

            os.system(LIMPIAR)

    def pedir_operandos(self) :
        while True:
            print "Por favor, ingrese el primer operando"

            try:
                op1 = int(raw_input(self.prompt))
            except ValueError:
                print "Error: Debes introducir un número\n"
            else:
                break

        while True:
            print "Por favor, ingrese el segundo operando"

            try:
                op2 = int(raw_input(self.prompt))
            except ValueError:
                print "Error: Debes introducir un número\n"
            else:
                break

        return op1, op2

    def opcion_sumar(self, op1, op2) :
        return op1 + op2

    def opcion_restar(self, op1, op2) :
        return op1 - op2

    def opcion_multiplicar(self, op1, op2) :
        return op1 * op2

    def opcion_dividir(self, op1, op2) :
        return op1 / op2

    def opcion_salir(self) :
        # Aquí podría ir un mensaje de despedida
        raise KeyboardInterrupt()

Para poder ver nuestro menú andando solo necesitamos importarlo en el script (archivo python o incluso el interprete interactivo) en el cual será utilizado

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

# main.py

import menu

mi_menu = menu.Menu()
mi_menu.loop()

Y listo. Espero que les haya sido de ayuda 😉

Anuncios