view 1-Ruby/acts_as_csv_module.rb @ 26:d50ff917c163

Change function calls so methods get patched in anyway
author IBBoard <dev@ibboard.co.uk>
date Fri, 20 Jan 2017 21:03:13 +0000
parents cd874e58dbc5
children 2ea3b5e9b41b
line wrap: on
line source

#---
# Excerpted from "Seven Languages in Seven Weeks",
# published by The Pragmatic Bookshelf.
# Copyrights apply to this code. It may not be used to create training material, 
# courses, books, articles, and the like. Contact us if you are in doubt.
# We make no guarantees that this code is fit for any purpose. 
# Visit http://www.pragmaticprogrammer.com/titles/btlang for more book information.
#---
module ActsAsCsv
  def self.included(base)
    base.extend ClassMethods
    # The book doesn't do this, but it is possible to add the methods
    # without having the inheriting class call a second function to do it
    base.acts_as_csv
  end
  
  module ClassMethods
    def acts_as_csv
      include InstanceMethods
    end
  end
  
  module InstanceMethods   
    def read
      @csv_contents = []
      filename = self.class.to_s.downcase + '.txt'
      file = File.new(filename)
      @headers = file.gets.chomp.split(', ')

      file.each do |row|
        @csv_contents << row.chomp.split(', ')
      end
    end
    
    attr_accessor :headers, :csv_contents
    def initialize
      read 
    end
  end
end

class RubyCsv  # no inheritance! You can mix it in
  include ActsAsCsv
  # Also, nothing to do beyond the "include" - it automatically defines the methods!
end

m = RubyCsv.new
puts m.headers.inspect
puts m.csv_contents.inspect