Sha256: 9cec389aec88b0d815236e73776b78d0ae87dd94731174be8a5461e40e1bc4b4
Contents?: true
Size: 1.16 KB
Versions: 26
Compression:
Stored size: 1.16 KB
Contents
module Turnip
class Table
class WidthMismatch < StandardError
def initialize(expected, actual)
super("Expected the table to be #{expected} columns wide, got #{actual}")
end
end
class ColumnNotExist < StandardError
def initialize(column_name)
super("The column named \"#{column_name}\" does not exist")
end
end
attr_reader :raw
alias_method :to_a, :raw
include Enumerable
def initialize(raw)
@raw = raw
end
def headers
raw.first
end
def rows
raw.drop(1)
end
def hashes
rows.map { |row| Hash[headers.zip(row)] }
end
def rows_hash
raise WidthMismatch.new(2, width) unless width == 2
transpose.hashes.first
end
def transpose
self.class.new(raw.transpose)
end
def each
raw.each { |row| yield(row) }
end
def map_column!(name, strict = true)
index = headers.index(name.to_s)
if index.nil?
raise ColumnNotExist.new(name) if strict
else
rows.each { |row| row[index] = yield(row[index]) }
end
end
private
def width
raw[0].size
end
end
end
Version data entries
26 entries across 26 versions & 1 rubygems