WordPress - Menú de compras

class Articulo:

    def __init__(self, nombre, inventario):

        self.nombre = nombre

        self.inventario = inventario

    def restar_inventario(self, cantidad):

        if cantidad <= self.inventario:

            self.inventario -= cantidad

            print(f"Se han restado {cantidad} unidades de {self.nombre} del inventario.")

        else:

            print(f"No hay suficiente inventario de {self.nombre}.")

# Crear objetos de artículos

articulo1 = Articulo("Artículo 1", 10)

articulo2 = Articulo("Artículo 2", 5)

articulo3 = Articulo("Artículo 3", 8)

articulo4 = Articulo("Artículo 4", 3)

articulo5 = Articulo("Artículo 5", 12)

# Menú de compras

menu = {

    1: articulo1,

    2: articulo2,

    3: articulo3,

    4: articulo4,

    5: articulo5

}

# Función para mostrar el menú y realizar la compra

def realizar_compra():

    print("Menú de compras:")

    for key, value in menu.items():

        print(f"{key}. {value.nombre} - Inventario: {value.inventario}")

    opcion = int(input("Seleccione el número del artículo que desea comprar: "))

    cantidad = int(input("Ingrese la cantidad que desea comprar: "))

    if opcion in menu:

        articulo = menu[opcion]

        articulo.restar_inventario(cantidad)

    else:

        print("Opción inválida.")

# Ejecutar el programa

realizar_compra()