survey reader for Think Stats

gistfile1.rb

# finally, correctly rewrote the Python script "survey" in Think Stats to Ruby.
# burned the midnight candle to get this right. feels great to succeed

class Table < Hash

  def initialize(name=nil)
    self[:records] = []
    self[:name] = name
  end

  def length
    return self[:records].size
  end

  def add(record)
    self[:records] << record
  end

  def expand(records)
    for record in records
      self.add(record)
    end
  end

  # Here's where the fun begins

  def upload(filename, fields)
    file = File.open(filename, "r")
    file.each_line { |line|
      self.generate(line, fields)
    }
  end

  def generate(line, fields={})
    record = {}
    fields.each { |key, value|
      record[key] = line[value].to_i
    }
    self.add(record)
  end

  def recode
    # not yet implemented
  end

end

# Copyleft 2012, Jared Davis @haiqus