Contents

require 'tap/tasks/load'
require 'csv'

module Tap
  module Tasks
    class Load
      
      # :startdoc::task reads csv data
      #
      # Load CSV data as an array of arrays, selecting the specified rows and
      # columns.
      # 
      #   % tap load/csv 'a,b,c.d,e,f' --row-sep '.' -: inspect
      #   [["a", "b", "c"], ["d", "e", "f"]]
      # 
      # Note this task is quite inefficient in that it will load all data
      # before making a selection; large files or edge selections may benefit
      # from an alternate task.
      #
      class Csv < Load
        
        config :columns, nil, :short => :C, &c.range_or_nil   # Specify a range of columns
        config :rows, nil, :short => :R, &c.range_or_nil      # Specify a range of rows
        
        config :col_sep, ',', :short => :c, &c.string_or_nil  # The column separator (",")
        config :row_sep, $/, :short => :r, &c.string_or_nil   # The row separator ("\r\n" or "\n")
        
        # Loads the io data as CSV, into an array of arrays.
        def load(io)
          data = parse(io.read)
          
          if rows
            data = data[rows]
          end
          
          if columns
            data.collect! do |cols|
              cols[columns]
            end
          end
          
          data
        end
        
        private
        
        if RUBY_VERSION >= '1.9'
          def parse(str)
            CSV.parse(str, :col_sep => col_sep, :row_sep => row_sep)
          end
        else
          def parse(str)
            CSV.parse(str, col_sep, row_sep)
          end
        end
      end 
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
tap-tasks-0.8.0	lib/tap/tasks/load/csv.rb
tap-tasks-0.7.0	lib/tap/tasks/load/csv.rb