Module: DBI::Utils::TableFormatter

Defined in:
lib/dbi/utils.rb

Class Method Summary collapse

Class Method Details

.ascii(header, rows, header_orient = :left, rows_orient = :left, indent = 2, cellspace = 1, pagebreak_after = nil, output = STDOUT) ⇒ Object

TODO: add a nr-column where the number of the column is shown



288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
 
# File 'lib/dbi/utils.rb', line 288

def TableFormatter.ascii(header, rows, 
  header_orient=:left, rows_orient=:left, 
  indent=2, cellspace=1, pagebreak_after=nil,
  output=STDOUT)

  header_orient ||= :left
  rows_orient   ||= :left
  indent        ||= 2
  cellspace     ||= 1

  # pagebreak_after n-rows (without counting header or split-lines)
  # yield block with output as param after each pagebreak (not at the end)

  col_lengths = (0...(header.size)).collect do |colnr|
    [
    (0...rows.size).collect { |rownr|
      value = rows[rownr][colnr]
      (value.nil? ? "NULL" : value).to_s.size
    }.max,
    header[colnr].size
    ].max
  end

  indent = " " * indent

  split_line = indent + "+"
  col_lengths.each {|col| split_line << "-" * (col+cellspace*2) + "+" }

  cellspace = " " * cellspace

  output_row = proc {|row, orient|
    output << indent + "|"
    row.each_with_index {|c,i|
      output << cellspace
      str = (c.nil? ? "NULL" : c).to_s
      output << case orient
      when :left then   str.ljust(col_lengths[i])
      when :right then  str.rjust(col_lengths[i])
      when :center then str.center(col_lengths[i])
      end 
      output << cellspace
      output << "|"
    }
    output << "\n" 
  } 

  rownr = 0
 
  loop do 
    output << split_line + "\n"
    output_row.call(header, header_orient)
    output << split_line + "\n"
    if pagebreak_after.nil?
      rows.each {|ar| output_row.call(ar, rows_orient)}
      output << split_line + "\n"
      break
    end      

    rows[rownr,pagebreak_after].each {|ar| output_row.call(ar, rows_orient)}
    output << split_line + "\n"

    rownr += pagebreak_after

    break if rownr >= rows.size
    
    yield output if block_given?
  end
  
end