smilodon.rb in smilodon-0.2.3

- old
+ new

@@ -1,7 +1,8 @@
 require 'smilodon/errors'
 require 'smilodon/logger'
+require 'csv'
 
 # Smilodon includes helper methods to ease parsing data files.
 # Assigning a header and iterating over rows is handled by the
 # module via a simple configuration.
 # 
@@ -28,49 +29,52 @@
     DIRECTORY = 'files'
 
     # Default data file type is CSV.
     TYPE = 'csv'
 
-    # Attribute accessors for the directory, file name, header, count, rows and before hook.
-    attr_accessor :logger, :directory, :file, :type, :header, :count, :rows, :before
+    # Attribute accessors for the directory, file name, header, rows and before hook.
+    attr_accessor :logger, :directory, :files, :type, :header, :rows, :before
 
     # Configuration helper.
     #
-    # @param [String] file The data file name.
+    # @param [Array] args an array of strings with an optional options hash
     # @option options [String] :directory ('db/populate/data_files') The location of the data file.
     # @option options [String] :type ('csv') The data file type.
     # @option options [Boolean] :header Set true if the file has a header.
     # @option options [String] :before The method to call before the run.
-    def populates(file, options = {})
+    def populates(*args)
       # Setup the logger to log populator warnings and messages.
       self.logger = PopulateLogger.setup
 
+      options = args.last.is_a?(Hash) ? args.pop : {}
+
+      self.files = args
       self.directory = options[:directory] || DIRECTORY
-      self.file = file
       self.type = options[:type] || TYPE
       self.header = options[:header]
       self.before = options[:before]
     end
 
     # Parses the data file content and processes each row.
     #
     # @return [Boolean] Returns true if all rows are processed successfully.
     def run
 
-      # Call the before hook if defined.
-      #
-      # @usage
-      #   populates 'TestFile', :before => :inactivate
-      send(before) if before
+      files.each do |f|
+        # Call the before hook if defined.
+        #
+        # @usage
+        #   populates 'TestFile', :before => :inactivate
+        send(before) if before
 
-      rows = parser.parse(read)
-      self.count = header ? rows.count - 1 : rows.count
-      self.header = rows.shift if header
-      count.times { process(rows.shift) }
+        rows = parser.parse(read(f))
+        rows.shift if header
+        rows.count.times { process(rows.shift) }
+      end
 
       # Return true when all rows are processed.
-      return true
+      true
     end
 
     # Stub method to be defined in the extended module.
     # 
     # @raise [Exception] Raises an exception if the extended module does not override the method.
@@ -80,10 +84,11 @@
 
     private
     
     # The parser to use based on the type of data file.
     #
+    # @param [String] Name of file to be read
     # @return [Parser, #parse] Returns the parser class to use.
     def parser
       if type == 'csv'
         require 'csv'
         return CSV::Reader
@@ -91,21 +96,26 @@
     end
 
     # Absolute path for the data file.
     # 
     # @return [String] The absolute path.
-    def path
-      "#{Rails.root}/#{directory}/#{file}.#{type}"
+    def path(file)
+      if defined?(Rails) 
+        "#{Rails.root}/#{directory}/#{file}.#{type}"
+      else
+        "#{directory}/#{file}.#{type}"
+      end
     end
 
     # Reads the data file.
     #
+    # @param [String] Name of file to be read
     # @return [File] The data file contents.
     # @raise [DataFileNotConfigured] Raises an exception when the file is not configured.
     # @raise [MissingDataFile] Raises an exception when the configured file is missing.
-    def read
+    def read(file)
       raise DataFileNotConfigured unless file
-      raise MissingDataFile unless File.exist?(path)
-      File.read path
+      raise MissingDataFile unless File.exist?(path(file))
+      File.read path(file)
     end
   end
 end