Module: TSV::XLSX

Defined in:

lib/rbbt/tsv/excel.rb

Class Method Summary

• .read(file, options = {}) ⇒ Object
• .write(tsv, file, options = {}) ⇒ Object

Class Method Details

.read(file, options = {}) ⇒ Object

# File 'lib/rbbt/tsv/excel.rb', line 151

def self.read(file, options = {})
  options = Misc.add_defaults options, :sep2 => /[,|]\s?/
  sheet = Misc.process_options options, :sheet
  header = Misc.process_options options, :header

  header = true unless header == false
  TmpFile.with_file do |filename|
    workbook = RubyXL::Parser.parse file
    sheet    = sheet ? workbook[sheet] : workbook.worksheets.first

    rows = []

    sheet.each do |row|
      next if row.nil?
      rows << row.cells.collect{|c| c.nil? ? nil : c.value}.collect{|c| String === c ? c.gsub("\n", ' ') : c }
    end

    num_values = rows.first.length
    File.open(filename, 'w') do |f|
      if header
        header = rows.shift
        f.puts "#" + header * "\t"
      end

      rows.each do |row| 
        row[num_values-1] ||= nil
        f.puts row * "\t" 
      end
    end

    TSV.open(filename, options)
  end
end

.write(tsv, file, options = {}) ⇒ Object

# File 'lib/rbbt/tsv/excel.rb', line 185

def self.write(tsv, file, options = {})
  sheet = Misc.process_options options, :sheet

  fields, rows = TSV._excel_data(tsv, options)

  book = RubyXL::Workbook.new
  sheet1 = book.worksheets.first
  sheet1.sheet_name = sheet if sheet

  fields.each_with_index do |e,i|
    sheet1.add_cell(0, i, e)
  end

  rows.each_with_index do |cells,i|
    cells.each_with_index do |e,j|
      sheet1.add_cell(i+1, j, e)
    end
  end

  book.write file
end