Class: Roo::Base

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/roo/base.rb

Overview

Base class for all other types of spreadsheets

Direct Known Subclasses

CSV, Excel, Excel2003XML, Excelx, Google, OpenOffice

Constant Summary collapse

TEMP_PREFIX =
"oo_"

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(filename, options = {}, file_warning = :error, tmpdir = nil) ⇒ Base

Returns a new instance of Base.



48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# File 'lib/roo/base.rb', line 48

def initialize(filename, options={}, file_warning=:error, tmpdir=nil)
  @filename = filename
  @options = options

  @cell = {}
  @cell_type = {}
  @cells_read = {}

  @first_row = {}
  @last_row = {}
  @first_column = {}
  @last_column = {}

  @header_line = 1
  @default_sheet = self.sheets.first
  @header_line = 1
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(m, *args) ⇒ Object

when a method like spreadsheet.a42 is called convert it to a call of spreadsheet.cell(‘a’,42)



372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
# File 'lib/roo/base.rb', line 372

def method_missing(m, *args)
  # #aa42 => #cell('aa',42)
  # #aa42('Sheet1')  => #cell('aa',42,'Sheet1')
  if m =~ /^([a-z]+)(\d)$/
    col = Roo::Base.letter_to_number($1)
    row = $2.to_i
    if args.empty?
      cell(row,col)
    else
      cell(row,col,args.first)
    end
  else
    super
  end
end

Instance Attribute Details

#default_sheetObject

Returns the value of attribute default_sheet.



21
22
23
# File 'lib/roo/base.rb', line 21

def default_sheet
  @default_sheet
end

#header_lineObject

sets the line with attribute names (default: 1)



24
25
26
# File 'lib/roo/base.rb', line 24

def header_line
  @header_line
end

#headersObject (readonly)

Returns the value of attribute headers.



21
22
23
# File 'lib/roo/base.rb', line 21

def headers
  @headers
end

Instance Method Details

#column(columnnumber, sheet = nil) ⇒ Object

returns all values in this column as an array column numbers are 1,2,3,… like in the spreadsheet



273
274
275
276
277
278
279
280
281
282
# File 'lib/roo/base.rb', line 273

def column(columnnumber,sheet=nil)
  if columnnumber.class == String
    columnnumber = Roo::Excel.letter_to_number(columnnumber)
  end
  sheet ||= @default_sheet
  read_cells(sheet)
  first_row(sheet).upto(last_row(sheet)).map do |row|
    cell(row,columnnumber,sheet)
  end
end

#each(options = {}) ⇒ Object

you can also pass in a :clean => true option to strip the sheet of odd unicode characters and white spaces around columns



425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
# File 'lib/roo/base.rb', line 425

def each(options={})
  if options.empty?
    1.upto(last_row) do |line|
      yield row(line)
    end
  else
    if options[:clean]
      options.delete(:clean)
      @cleaned ||= {}
      @cleaned[@default_sheet] || clean_sheet(@default_sheet)
    end

    if options[:header_search]
      @headers = nil
      @header_line = row_with(options[:header_search])
    elsif [:first_row,true].include?(options[:headers])
      @headers = []
      row(first_row).each_with_index {|x,i| @headers << [x,i + 1]}
    else
      set_headers(options)
    end

    headers = @headers ||
      Hash[(first_column..last_column).map do |col|
        [cell(@header_line,col), col]
      end]

    @header_line.upto(last_row) do |line|
      yield(Hash[headers.map {|k,v| [k,cell(line,v)]}])
    end
  end
end

#each_with_pagenameObject

iterate through all worksheets of a document



396
397
398
399
400
# File 'lib/roo/base.rb', line 396

def each_with_pagename
  self.sheets.each do |s|
    yield sheet(s,true)
  end
end

#empty?(row, col, sheet = nil) ⇒ Boolean

true if cell is empty

Returns:

  • (Boolean)


309
310
311
312
313
314
315
316
# File 'lib/roo/base.rb', line 309

def empty?(row, col, sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  row,col = normalize(row,col)
  contents = cell(row, col, sheet)
  !contents || (celltype(row, col, sheet) == :string && contents.empty?) \
    || (row < first_row(sheet) || row > last_row(sheet) || col < first_column(sheet) || col > last_column(sheet))
end

#find(*args) ⇒ Object

find a row either by row number or a condition Caution: this works only within the default sheet -> set default_sheet before you call this method (experimental. see examples in the test_roo.rb file)



219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
# File 'lib/roo/base.rb', line 219

def find(*args) # :nodoc
  options = (args.last.is_a?(Hash) ? args.pop : {})
  result_array = options[:array]
  header_for = Hash[1.upto(last_column).map do |col|
    [col, cell(@header_line,col)]
  end]
  #-- id
  if args[0].class == Fixnum
    rownum = args[0]
    if @header_line
      [Hash[1.upto(self.row().size).map {|j|
        [header_for.fetch(j), cell(rownum,j)]
      }]]
    else
      self.row(rownum).size.times.map {|j|
        cell(rownum,j + 1)
      }
    end
  #-- :all
  elsif args[0] == :all
    rows = first_row.upto(last_row)

    # are all conditions met?
    if (conditions = options[:conditions]) && !conditions.empty?
      column_with = header_for.invert
      rows = rows.select do |i|
        conditions.all? { |key,val| cell(i,column_with[key]) == val }
      end
    end

    rows.map do |i|
      if result_array
        self.row(i)
      else
        Hash[1.upto(self.row(i).size).map do |j|
          [header_for.fetch(j), cell(i,j)]
        end]
      end
    end
  end
end

#first_column(sheet = nil) ⇒ Object

returns the number of the first non-empty column



122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
# File 'lib/roo/base.rb', line 122

def first_column(sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  if @first_column[sheet]
    return @first_column[sheet]
  end
  impossible_value = 999_999 # more than a spreadsheet can hold
  result = impossible_value
  @cell[sheet].each_pair {|key,value|
    x = key.last.to_i # _to_string(key).split(',')
    result = [result, x].min if value
  } if @cell[sheet]
  result = nil if result == impossible_value
  @first_column[sheet] = result
  result
end

#first_column_as_letter(sheet = nil) ⇒ Object

first non-empty column as a letter



76
77
78
# File 'lib/roo/base.rb', line 76

def first_column_as_letter(sheet=nil)
  Roo::Base.number_to_letter(first_column(sheet))
end

#first_row(sheet = nil) ⇒ Object

returns the number of the first non-empty row



86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# File 'lib/roo/base.rb', line 86

def first_row(sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  if @first_row[sheet]
    return @first_row[sheet]
  end
  impossible_value = 999_999 # more than a spreadsheet can hold
  result = impossible_value
  @cell[sheet].each_pair {|key,value|
    y = key.first.to_i # _to_string(key).split(',')
    result = [result, y].min if value
  } if @cell[sheet]
  result = nil if result == impossible_value
  @first_row[sheet] = result
  result
end

#infoObject

returns information of the spreadsheet document and all sheets within this document.



320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
# File 'lib/roo/base.rb', line 320

def info
  without_changing_default_sheet do
    result = "File: #{File.basename(@filename)}\n"+
      "Number of sheets: #{sheets.size}\n"+
      "Sheets: #{sheets.join(', ')}\n"
    n = 1
    sheets.each {|sheet|
      self.default_sheet = sheet
      result << "Sheet " + n.to_s + ":\n"
      unless first_row
        result << "  - empty -"
      else
        result << "  First row: #{first_row}\n"
        result << "  Last row: #{last_row}\n"
        result << "  First column: #{Roo::Base.number_to_letter(first_column)}\n"
        result << "  Last column: #{Roo::Base.number_to_letter(last_column)}"
      end
      result << "\n" if sheet != sheets.last
      n += 1
    }
    result
  end
end

#last_column(sheet = nil) ⇒ Object

returns the number of the last non-empty column



140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
# File 'lib/roo/base.rb', line 140

def last_column(sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  if @last_column[sheet]
    return @last_column[sheet]
  end
  impossible_value = 0
  result = impossible_value
  @cell[sheet].each_pair {|key,value|
    x = key.last.to_i # _to_string(key).split(',')
    result = [result, x].max if value
  } if @cell[sheet]
  result = nil if result == impossible_value
  @last_column[sheet] = result
  result
end

#last_column_as_letter(sheet = nil) ⇒ Object

last non-empty column as a letter



81
82
83
# File 'lib/roo/base.rb', line 81

def last_column_as_letter(sheet=nil)
  Roo::Base.number_to_letter(last_column(sheet))
end

#last_row(sheet = nil) ⇒ Object

returns the number of the last non-empty row



104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# File 'lib/roo/base.rb', line 104

def last_row(sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  if @last_row[sheet]
    return @last_row[sheet]
  end
  impossible_value = 0
  result = impossible_value
  @cell[sheet].each_pair {|key,value|
    y = key.first.to_i # _to_string(key).split(',')
    result = [result, y].max if value
  } if @cell[sheet]
  result = nil if result == impossible_value
  @last_row[sheet] = result
  result
end

#parse(options = {}) ⇒ Object



458
459
460
461
462
463
464
465
466
# File 'lib/roo/base.rb', line 458

def parse(options={})
  ary = []
  if block_given?
    each(options) {|row| ary << yield(row)}
  else
    each(options) {|row| ary << row}
  end
  ary
end

#reloadObject

reopens and read a spreadsheet document



302
303
304
305
306
# File 'lib/roo/base.rb', line 302

def reload
  ds = @default_sheet
  reinitialize
  self.default_sheet = ds
end

#row(rownumber, sheet = nil) ⇒ Object

returns all values in this row as an array row numbers are 1,2,3,… like in the spreadsheet



263
264
265
266
267
268
269
# File 'lib/roo/base.rb', line 263

def row(rownumber,sheet=nil)
  sheet ||= @default_sheet
  read_cells(sheet)
  first_column(sheet).upto(last_column(sheet)).map do |col|
    cell(rownumber,col,sheet)
  end
end

#row_with(query, return_headers = false) ⇒ Object



468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
# File 'lib/roo/base.rb', line 468

def row_with(query,return_headers=false)
  query.map! {|x| Array(x.split('*'))}
  line_no = 0
  each do |row|
    line_no += 1
    # makes sure headers is the first part of wildcard search for priority
    # ex. if UPC and SKU exist for UPC*SKU search, UPC takes the cake
    headers = query.map do |q|
      q.map {|i| row.grep(/#{i}/i)[0]}.compact[0]
    end.compact

    if headers.length == query.length
      @header_line = line_no
      return return_headers ? headers : line_no
    elsif line_no > 100
      raise "Couldn't find header row."
    end
  end
end

#set(row, col, value, sheet = nil) ⇒ Object

set a cell to a certain value (this will not be saved back to the spreadsheet file!)



286
287
288
289
290
291
292
293
294
295
296
297
298
299
# File 'lib/roo/base.rb', line 286

def set(row,col,value,sheet=nil) #:nodoc:
  sheet ||= @default_sheet
  read_cells(sheet)
  row, col = normalize(row,col)
  cell_type = case value
              when Fixnum then :float
              when String, Float then :string
              else
                raise ArgumentError, "Type for #{value} not set"
              end

  set_value(row,col,value,sheet)
  set_type(row,col,cell_type,sheet)
end

#sheet(index, name = false) ⇒ Object

access different worksheets by calling spreadsheet.sheet(1) or spreadsheet.sheet(‘SHEETNAME’)



390
391
392
393
# File 'lib/roo/base.rb', line 390

def sheet(index,name=false)
  @default_sheet = String === index ? index : self.sheets[index]
  name ? [@default_sheet,self] : self
end

#to_csv(filename = nil, sheet = nil) ⇒ Object

write the current spreadsheet to stdout or into a file



187
188
189
190
191
192
193
194
195
196
197
198
199
200
# File 'lib/roo/base.rb', line 187

def to_csv(filename=nil,sheet=nil)
  sheet ||= @default_sheet
  if filename
    File.open(filename,"w") do |file|
      write_csv_content(file,sheet)
    end
    return true
  else
    sio = StringIO.new
    write_csv_content(sio,sheet)
    sio.rewind
    return sio.read
  end
end

#to_matrix(from_row = nil, from_column = nil, to_row = nil, to_column = nil, sheet = nil) ⇒ Object

returns a matrix object from the whole sheet or a rectangular area of a sheet



203
204
205
206
207
208
209
210
211
212
213
214
# File 'lib/roo/base.rb', line 203

def to_matrix(from_row=nil, from_column=nil, to_row=nil, to_column=nil,sheet=nil)
  require 'matrix'

  sheet ||= @default_sheet
  return Matrix.empty unless first_row

  Matrix.rows((from_row||first_row(sheet)).upto(to_row||last_row(sheet)).map do |row|
    (from_column||first_column(sheet)).upto(to_column||last_column(sheet)).map do |col|
      cell(row,col)
    end
  end)
end

#to_xmlObject

returns an XML representation of all sheets of a spreadsheet file



345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
# File 'lib/roo/base.rb', line 345

def to_xml
  Nokogiri::XML::Builder.new do |xml|
    xml.spreadsheet {
      self.sheets.each do |sheet|
        self.default_sheet = sheet
        xml.sheet(:name => sheet) { |x|
          if first_row and last_row and first_column and last_column
            # sonst gibt es Fehler bei leeren Blaettern
            first_row.upto(last_row) do |row|
              first_column.upto(last_column) do |col|
                unless empty?(row,col)
                  x.cell(cell(row,col),
                    :row =>row,
                    :column => col,
                    :type => celltype(row,col))
                end
              end
            end
          end
        }
      end
    }
  end.to_xml
end

#to_yaml(prefix = {}, from_row = nil, from_column = nil, to_row = nil, to_column = nil, sheet = nil) ⇒ Object

returns a rectangular area (default: all cells) as yaml-output you can add additional attributes with the prefix parameter like: oo.to_yaml(“sheet” => “1”)



160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
# File 'lib/roo/base.rb', line 160

def to_yaml(prefix={}, from_row=nil, from_column=nil, to_row=nil, to_column=nil,sheet=nil)
  sheet ||= @default_sheet
  result = "--- \n"
  return '' unless first_row # empty result if there is no first_row in a sheet

  (from_row||first_row(sheet)).upto(to_row||last_row(sheet)) do |row|
    (from_column||first_column(sheet)).upto(to_column||last_column(sheet)) do |col|
      unless empty?(row,col,sheet)
        result << "cell_#{row}_#{col}: \n"
        prefix.each {|k,v|
          result << "  #{k}: #{v} \n"
        }
        result << "  row: #{row} \n"
        result << "  col: #{col} \n"
        result << "  celltype: #{self.celltype(row,col,sheet)} \n"
        if self.celltype(row,col,sheet) == :time
          result << "  value: #{Roo::Base.integer_to_timestring( self.cell(row,col,sheet))} \n"
        else
          result << "  value: #{self.cell(row,col,sheet)} \n"
        end
      end
    end
  end
  result
end