RegistroInventario.py

# -*- coding: cp1252 -*-
from Producto import Producto

class RegistroInventario:
    def __init__(self):
        self.listaProductos = []
    def nuevoProducto(self):
        if len(self.listaProductos) < 20:
            numero = int(raw_input('Numero: '))
            nombre = raw_input('Nombre: ')
            cantidad = int(raw_input('Cantidad: '))
            precio = float(raw_input('Precio: '))
            self.listaProductos.append(Producto(numero, nombre, cantidad, precio))
        else:
            print 'Lista de productos llena. Elimine uno o mas productos.\n'
    def borrarProducto(self):
        if len(self.listaProductos) != 0:
            num_producto = int(raw_input('Numero del producto?: '))
            indice = self.buscarNumero(num_producto)
            if indice > -1:
                del self.listaProductos[indice]
                print "Producto removido\n"
            else:
                print "El producto no existe.\n"
        else:
            print "Lista de productos vacia...\n"
    def modificarProducto(self):
        if len(self.listaProductos) != 0:
            num_producto = int(raw_input('Numero del producto?: '))
            indice = self.buscarNumero(num_producto)
            if indice > -1:
                self.listaProductos[indice].setNumero(int(raw_input('Numero: ')))
                self.listaProductos[indice].setNombre(raw_input('Nombre: '))
                self.listaProductos[indice].setCantidad(int(raw_input('Cantidad: ')))
                self.listaProductos[indice].setPrecio(float(raw_input('Precio: ')))
                print 'Producto modificado con exito.\n'
            else:
                print 'El producto no existe\n'
        else:
            print 'La lista de productos esta vacia\n'
    def obtenerInfo(self):
        if len(self.listaProductos) != 0:
            num_producto = int(raw_input('Numero del producto?: '))
            indice = self.buscarNumero(num_producto)
            if indice > -1:
                print "Numero\t\tNombre\t\tCantidad\tPrecio"
                print self.listaProductos[indice].getInformacion()
            else:
                print "El numero de producto no se encuentra\n"
        else:
            print 'Lista de producto vacia.\n'
    def buscarNumero(self, numero):
        indice = 0
        aux = 0
        while indice < len(self.listaProductos):
            if self.listaProductos[indice].getNumero() == numero:
                aux = indice
                break
            else:
                aux = -1
            indice = indice + 1
        return aux
    def dineroInvertido(self):
        inversion = 0
        for producto in self.listaProductos:
            aux = producto.getPrecio() * producto.getCantidad()
            inversion = inversion + aux
        print "El dinero invertido en los productos es: %f" % (inversion)
    def listarProductos(self):
        if len(self.listaProductos) != 0:
            etiquetas = "Numero\t\tNombre\t\tCantidad\tPrecio"
            print etiquetas
            for producto in self.listaProductos:
                print producto.getInformacion()
            print ""
        else:
            print 'No hay productos en la lista. Agrege uno o mas desde el menu.\n'
    def main(self):
        opcion = 0
        while(opcion != 7):
            opcion = int(raw_input(
                "1.-Incluir nuevo producto\n\
2.-Borrar producto\n\
3.-Modificar info de producto\n\
4.-Obtener info de producto\n\
5.-Dinero invertido en productos\n\
6.-Listar los productos\n\
7.-Salir\n...:"))
            if opcion == 1:
                self.nuevoProducto()
            elif opcion == 2:
                self.borrarProducto()
            elif opcion == 3:
                self.modificarProducto()
            elif opcion == 4:
                self.obtenerInfo()
            elif opcion == 5:
                self.dineroInvertido()
            elif opcion == 6:
                self.listarProductos()
            elif opcion == 7:
                print 'Saliendo del programa...¡Finalizado!'
            else:
                print "¡Opcion no valida! Intente de nuevo."
obj = RegistroInventario()
obj.main()