garden 0.0.13.pre → 0.0.14.pre

data/garden.gemspec

@@ -14,7 +14,6 @@ Gem::Specification.new do |s|
   s.rubyforge_project = "garden"
 
   s.add_dependency 'spreadsheet'
-  # s.add_dependency 'active_record'
 
   s.files         = `git ls-files`.split("\n")
   s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")

data/lib/garden.rb

@@ -22,12 +22,14 @@ module Garden
   # end
   
   def self.excel file_name_or_path, options=nil
-    filepath = resolve_file_path(file_name_or_path)
+    filepath = File.exists?(file_name_or_path) ? file_name_or_path : File.join(Rails.root, "db/seeds/#{file_name_or_path}.xls")
     Spreadsheets::Excel.new filepath, options
   end
 
-  def self.resolve_file_path name_or_path
-    File.exists?(name_or_path) ? name_or_path : File.join(Rails.root, "db/seeds/#{name_or_path}.xls")
+  def self.csv file_name_or_path, options=nil
+    filepath = File.exists?(file_name_or_path) ? file_name_or_path : File.join(Rails.root, "db/seeds/#{file_name_or_path}.csv")
+    Spreadsheets::CSV.new filepath, options
   end
 
+
 end

data/lib/garden/mediators.rb

@@ -47,7 +47,7 @@ module Garden
         #   puts " ** Could not derive ActiveRecord model from the provided name: #{namish}. Exception: #{e.message}"
         # end
         
-        @instance = @clazz.new
+        # @instance = @clazz.new
         
       end
       
@@ -55,10 +55,10 @@ module Garden
         @clazz != nil
       end
     
-      def parse_headers array
-        # @headers = array.map { |header| header.to_s.parameterize.underscore }
-        @headers = array.map { |header| header.to_s.strip.gsub(/ /, '-').underscore }
-      end
+      # def parse_headers array
+      #   # @headers = array.map { |header| header.to_s.parameterize.underscore }
+      #   @headers = array.map { |header| header.to_s.strip.gsub(/ /, '-').underscore }
+      # end
       
       def reference_by col_name
         @instance_options[:reference] = col_name
@@ -68,11 +68,10 @@ module Garden
       #   @relationships
       # end
       # 
-      def create_instance attributes
+      def row attributes
         Instance.new @clazz, attributes, @instance_options.dup
       end
 
-
       def class_exists?(name)
         begin
           true if Kernel.const_get(name)

data/lib/garden/spreadsheets.rb

@@ -1,58 +1,60 @@
 require 'spreadsheet'
+require 'csv'
 
 module Garden
   module Spreadsheets
-    class Excel
     
+    class AbstractSpreadsheet
       def initialize filepath, options={}
         options ||= {}
-        
         @options = options.reverse_merge! :some_option => "nevermind"
-        
         open filepath
       end
     
+      def open(filepath)
+      end
+      
+      def build_mediator name
+        @mediator = Mediators::Table.new name
+        raise "Invalid mediator for table #{name}" unless @mediator.valid?
+        @mediator.reference_by(@options[:reference]) if @options.has_key?(:reference)
+        @mediator
+      end
+      
+    end
+    
+    class Excel < AbstractSpreadsheet
+    
       def open filepath
-        message = "Planting spreadsheet: #{filepath}"
+        message = "Planting excel spreadsheet: #{filepath}"
         message += " (only these worksheets: #{@options[:only].to_s})" if @options[:only]
         message += " (worksheet #{@options[:worksheet].to_s})" if @options[:worksheet]
         puts message
+        
+        excel_spreadsheet = ::Spreadsheet.open filepath
       
-        @ss = Spreadsheet.open filepath
-        worksheet_names = @options[:only] || @options[:worksheet] || @ss.worksheets.collect { |table| table.name }
+        worksheet_names = @options[:only] || @options[:worksheet] || excel_spreadsheet.worksheets.collect { |table| table.name }
         worksheet_names = [worksheet_names] unless worksheet_names.is_a?(Enumerable)
 
         # Import the worksheets
         worksheet_names.each do |name|
           puts "Parsing table #{name}"
-          parse_table @ss.worksheets.find { |table| table.name == name.to_s }
+          table = excel_spreadsheet.worksheets.find { |table| table.name == name.to_s }
+          parse_worksheet table.name, table.to_a
         end
       
       end
-    
-      def parse_table table
-        # The table object passed in is a Spreadsheet::Worksheet instance.
-
-        table_mediator = Mediators::Table.new table.name
-        if !table_mediator.valid?
-          return
-        end
       
-        # Get the headers. These values will be the attribute names
-        headers = table_mediator.parse_headers table.first.to_a
-        
-        # Set reference column if defined. 
-        if @options.has_key?(:reference)
-          table_mediator.reference_by(@options[:reference])
-        end
-      
-        # Now loop the table rows, inserting records.
-        table.each do |row|
-          next if row.idx == 0
-          table_mediator.create_instance parse_worksheet_row(headers, row)
+      def parse_worksheet name, rows
+        build_mediator name
+        # Expects the first row to be the headers.
+        headers = rows.shift.map { |header| header.to_s.strip.gsub(/ /, '-').underscore }
+        rows.each do |row|
+          attributes = parse_worksheet_row(headers, row)
+          @mediator.row attributes
         end
       end
-    
+      
       def parse_worksheet_row keys, values
         h = {}
         keys.each_index do |index|
@@ -61,6 +63,41 @@ module Garden
         end
         h
       end
+      
     end
+    
+    class CSV < AbstractSpreadsheet
+      def open filepath
+        file = File.open filepath
+        begin
+          rows = ::CSV.parse(file, :headers => true)
+        rescue Exception => e
+          rows = []
+        end
+        
+        # Build the mediator. Assume the table name is the same as the file's name passed in or the :table option
+        table_name = @options.delete(:table) || File.basename(filepath, File.extname(filepath))
+        message = "Planting csv spreadsheet: #{filepath}"
+        message << " (into table #{table_name})"
+        puts message
+        build_mediator table_name
+        
+        rows.each do |row|
+          attributes = row_to_attributes(row)
+          @mediator.row attributes
+        end
+      end
+      
+      def row_to_attributes row
+        h = {}
+        row.each do |ary|
+          key = ary.first.to_sym
+          h[key] = ary.last
+        end
+        h
+      end
+      
+    end
+    
   end
 end

data/lib/garden/version.rb

@@ -1,3 +1,3 @@
 module Garden
-  VERSION = "0.0.13.pre"
+  VERSION = "0.0.14.pre"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: garden
 version: !ruby/object:Gem::Version
-  version: 0.0.13.pre
+  version: 0.0.14.pre
   prerelease: 7
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-10-20 00:00:00.000000000Z
+date: 2011-11-11 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: spreadsheet
-  requirement: &70304106801080 !ruby/object:Gem::Requirement
+  requirement: &70302264679260 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70304106801080
+  version_requirements: *70302264679260
 description: ! 'Allows you to organize your seeds in different formats. Typical seeds.rb,
   yaml fixtures, and a variety of spreadsheet formats. '
 email: