En este post explicaré cómo trabajar con CSV desde el lenguaje de programación Ruby. Vamos a leer un archivo CSV (archivo separado por comas u otro delimitador) y también a escribir contenido a un nuevo archivo.
Nota: la foto del código fue tomada con Polacode y la marca de agua colocada con WaterPy.
Te invito a leer antes cómo escribir y leer archivos en Ruby, para que no te confundas al leer los ejercicios aquí presentados.
Trabajar con archivos CSV en Ruby
Para leer y escribir archivos CSV con Ruby se usa la clase o módulo csv
; se podría hacer a mano pero no hay necesidad de reinventar la rueda.
Aquí veremos cómo hacer ambas operaciones, algunas con clases y otras “simples” pero lo importante es ver cómo se abre el archivo y se le pasan las cosas que se escriben.
Escribir archivo CSV con Ruby
Para escribir a un archivo, una vez abierto, se le pasan los datos en forma de arreglo; así:
archivo << ["Un dato", "Otro dato", "Aquí más datos..."]
En cada llamada se creará una nueva fila o línea.
Eso puede ser llamado en cualquier lugar. En el siguiente ejemplo escribimos los detalles de un producto dentro de un ciclo.
=begin
Escribir contenido CSV a un archivo con Ruby
@author parzibyte
=end
require 'csv' #La clase que nos ayuda
NOMBRE_ARCHIVO = "productos.csv"
# Una simple clase para encapsular todo
class Producto
attr_reader :descripcion, :precio_compra, :precio_venta, :cantidad
def initialize(descripcion, precio_compra, precio_venta, cantidad)
@descripcion = descripcion
@precio_compra = precio_compra
@precio_venta = precio_venta
@cantidad = cantidad
end
end
# Un arreglo de productos que vamos a escribir en el CSV
productos = [
Producto.new("Galletas", 10, 15, 100),
Producto.new("Papas fritas", 15, 17, 10),
Producto.new("Chocolate", 3, 5, 50),
Producto.new("Jugo de manzana", 9, 11, 200),
]
CSV.open(NOMBRE_ARCHIVO, "a") do |archivo| # Abrir archivo en modo [a]ppend
productos.each do |p|
# Le pasamos un arreglo. Cada valor es una columna
archivo << [p.descripcion, p.precio_compra, p.precio_venta, p.cantidad]
end
end
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Por cierto, el archivo de productos.csv es el resultado de escribir a través de Ruby. Como lo dije, lo importante es la línea en donde se escribe directamente; lo demás es el ciclo y la apertura del archivo.
Leer CSV con Ruby
Una vez que ya escribimos, vamos a leer. Con Ruby es muy fácil leer un archivo CSV sin importar el contenido que tenga.
Lo digo de nuevo, en el ejemplo se usa una clase pero lo importante es la parte en donde se obtiene a la variable fila en cada iteración; esa fila es un arreglo indexado por números; es decir, el elemento 0 tiene el primer valor, el elemento 1 el segundo y así sucesivamente.
=begin
Leer contenido CSV de un archivo con Ruby
@author parzibyte
=end
require 'csv' #La clase que nos ayuda
NOMBRE_ARCHIVO = "productos.csv"
# Una simple clase para encapsular todo
class Producto
attr_reader :descripcion, :precio_compra, :precio_venta, :cantidad
def initialize(descripcion, precio_compra, precio_venta, cantidad)
@descripcion = descripcion
@precio_compra = precio_compra
@precio_venta = precio_venta
@cantidad = cantidad
end
# Método to string
def to_s
"Descripción #{descripcion}, P.C. #{precio_compra}, P.V. #{precio_venta}, cantidad #{cantidad}"
end
end
# Un arreglo de productos en donde vamos a poner los datos
productos = []
CSV.foreach(NOMBRE_ARCHIVO) do |fila| # Iterar línea o fila por fila
productos.push Producto.new(fila[0], fila[1], fila[2], fila[3])
end
# Imprimir
puts productos
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Galletas,10,15,100
Papas fritas,15,17,10
Chocolate,3,5,50
Jugo de manzana,9,11,200
Algo que se debe notar es que implementamos el método to_s
de la clase Producto, de este modo al imprimir con puts se pueden imprimir como texto formateado y bonito.
Lo digo de nuevo, no es necesario esto de las clases, ya que obtenemos la fila como arreglo en cada iteración.
Probar y descargar ejemplos
Puedes descargar los ejemplos así como los archivos CSV, instala Ruby y ejecuta rb main.rb para cada caso; así se listará todo en la terminal.
Pingback: Eliminar y agregar elementos a arreglo en Ruby - Parzibyte's blog