martes, 6 de diciembre de 2011

Manejo de ficheros en Python

Un ejemplo sencillo sobre ficheros en Python


import os
productos = []

class producto:
    """ Clase para cada producto """
    ID = 0
    nombre = ""
    cantidad = 0
    def __init__(self, ID, nombre, cantidad ):
        self.ID = ID
        self.nombre = nombre
        self.cantidad = cantidad

def registrarProducto():
    os.system('clear')
    print "="*80,"\t\tREGISTRAR PRODUCTO\n","="*80
    p = producto(long(raw_input("ID: ")),raw_input("Nombre:"),
                  int(raw_input("Cantidad: ")))
    productos.append(p)
    try:
        fichero = open("archivo.txt", "a") #abre un fichero existente
    except:
        fichero = open("archivo.txt","w") #se crea el fichero si no existe   
    fichero.write(str(p.ID)+","+p.nombre+","+ str(p.cantidad)+"\n")   
    fichero.close()
   
def verProductos():
    os.system('clear')
    print "="*80,"\t\tLISTA DE PRODUCTOS\n","="*80
    print "ID/Nombre/cantidad\n"
    try:
        fichero = open("archivo.txt","r")
        for linea in fichero:
            print linea.strip()    
        fichero.close()
    except:
        print "El fichero no existe, porfavor creelo"  

def printMenu():   
    print "="*80,"\t\tMENU\n","="*80   
    print "Escoja una opcion: "   
    print "\t1.Registrar un producto"
    print "\t2.Ver lista de productos"
    print "\t3.salir"   
    return raw_input("OPCION: ") 
        
def menu():
    opcion = 0
    while opcion != '3' :       
        opcion = printMenu()               
        if opcion == '1':
            registrarProducto()
        elif opcion == '2':
            verProductos()
menu()
exit()

No hay comentarios:

Publicar un comentario