active_seed 1.0.3 → 1.0.4

data/README.rdoc

@@ -6,7 +6,7 @@ Although the inbuilt support for seeding in Rails works well for basic seeding.
 
 Simple add the following to your Gemfile
 
-    gem 'active_seed', :git => "git://github.com/intrica/active_seed.git"
+    gem 'active_seed'
     
 Then run:
 
@@ -42,7 +42,7 @@ The purpose of a set file is to define a "set" of seed files. A set is defined i
 
     rails g active_seed
 
-When seeding, the default set file used will be one named after the current RAILS_ENV. So in devleopment mode it will use the file
+When seeding, the default set file used will be one named after the current RAILS_ENV. So in development mode it will use the file
 
     RAILS_ROOT/db/active_seed/development.yml
 
@@ -120,4 +120,10 @@ Evaluations are handy if you want to interpret data in the column rather than ju
 
 As you can see, wherever the ? appears in the header, it is replaced with the value of the corresponding column.
 
-== Potential Issues
+== Using from within your code
+
+If you would like to seed a set directly from within other code, for example if you wanted to seed the "test" set at the start of your cucumber tests you can add the following to your code:
+
+    require "active_seed"
+
+    ActiveSeed.seed("test")

data/VERSION

@@ -1 +1 @@
-1.0.3
+1.0.4

data/active_seed.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{active_seed}
-  s.version = "1.0.3"
+  s.version = "1.0.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["David Monagle"]
-  s.date = %q{2011-04-12}
+  s.date = %q{2011-05-16}
   s.description = %q{ActiveSeed Gives the ability to seed data by inserting it via activerecord. Also allows you to manage sets of data and seed based on the current environment.}
   s.email = %q{david.monagle@intrica.com.au}
   s.extra_rdoc_files = [
@@ -22,6 +22,7 @@ Gem::Specification.new do |s|
     "VERSION",
     "active_seed.gemspec",
     "lib/active_seed.rb",
+    "lib/active_seed/active_seed.rb",
     "lib/active_seed/engine.rb",
     "lib/active_seed/seed_csv.rb",
     "lib/generators/active_seed/example_html_colors_generator.rb",

data/lib/active_seed.rb

@@ -1 +1,2 @@
 require "active_seed/engine"
+require "active_seed/active_seed"

data/lib/active_seed/active_seed.rb

@@ -0,0 +1,99 @@
+require "csv"
+
+module ActiveSeed
+  def self.seed_csv(modelname, filename)
+    data = CSV.read(filename)
+    if data.size < 2
+      puts "No data found in file " << filename
+      return
+    end
+    header = data[0];
+    processed_header = Array.new
+    evaluations = Array.new
+    statics = Array.new
+    header.each do |h|
+      # Check if the header field has an assignment
+      if h.include? "="
+        c = h.split "="
+        h = c[0] # Replace the header with just the field name
+        c.delete_at(0) # Remove the field name
+        evaluation = c.join("=") # Rejoin the rest of the evaluation (if there was another =)
+        # If there is no question mark in the evaluation string then we set this column to be
+        # static.
+        unless evaluation.include? "?"
+          statics.push(h + "=" + evaluation)
+        else
+          processed_header.push(h)
+          evaluations.push(evaluation)
+        end
+      else
+        evaluations.push(nil)
+        processed_header.push(h)
+      end
+    end
+    header = processed_header
+    data.delete_at(0);
+    puts "Seeding " + data.size.to_s + " record" + (data.size > 1 ? "s" : "")
+    line = 1
+    data.each do |d|
+      line+=1
+      if d.size == header.size
+        code = "model = " + modelname + ".new\n"
+        for count in 0..(header.size - 1) do
+          unless (header[count].strip == "nil")
+            value = d[count]
+            value = "" if value.nil?
+            value = "'" + value.strip.gsub(/'/, "\\\\'") + "'"
+            if evaluations[count].nil?
+              assignment = value
+            else
+              assignment = evaluations[count].split("?")
+              assignment = assignment.join(value)
+            end
+            code += "model." + header[count].strip + "=" + assignment + "\n"
+          end
+        end
+        # Add in the statics
+        statics.each do |s|
+          code += "model." + s + "\n"
+        end
+        code += "unless model.save\n"
+        code += "ActiveSeed::print_errors(model.errors, " + line.to_s + ")"
+        code += "end\n"
+        eval code
+        #puts (line - 1).to_s + "/" + data.size.to_s
+      else
+        puts "Skipping line " + line.to_s + " with mismatch in number of fields (" + d.size.to_s + ")"
+      end
+    end
+    puts " Done"
+  end
+
+  def self.print_errors(errors, line)
+    puts "\nThere were errors on line " + line.to_s
+    errors.each do |e|
+      puts e.to_s
+    end
+  end
+
+  def seed(set)
+    set_file = File.join(::Rails.root.to_s, "db", "active_seed", set + ".yml")
+    if !File.exists?(set_file)
+      puts "Set file doesn't exist: " << set_file
+      return
+    end
+    puts "Seeding from set '" + set + "'"
+    fixture_list = YAML::load_file(set_file)
+    fixture_list.each do |model, sf|
+      seed_file = File.join(::Rails.root.to_s, "db", "active_seed", "data", sf + ".csv")
+      if !File.exists?(seed_file)
+        puts "Seed file doesn't exist: " << seed_file
+      else
+        puts "Seeding '" + seed_file + "'..."
+        ActiveSeed::seed_csv(model, seed_file)
+      end
+    end
+  end
+
+  module_function :seed
+end

data/lib/active_seed/seed_csv.rb

@@ -1,76 +1,2 @@
 require "csv"
 
-def seed_csv(modelname, filename)
-    data = CSV.read(filename)
-    if data.size < 2
-        puts "No data found in file " << filename
-        return
-    end
-    header = data[0];
-    processed_header = Array.new
-    evaluations = Array.new
-    statics = Array.new
-    header.each do |h|        
-    # Check if the header field has an assignment
-        if h.include? "="
-            c = h.split "="
-            h = c[0]        # Replace the header with just the field name
-            c.delete_at(0)  # Remove the field name
-            evaluation = c.join("=") # Rejoin the rest of the evaluation (if there was another =)
-            # If there is no question mark in the evaluation string then we set this column to be
-            # static.
-            unless evaluation.include? "?"
-                statics.push(h + "=" + evaluation)
-            else
-                processed_header.push(h)
-                evaluations.push(evaluation)
-            end
-        else
-            evaluations.push(nil)
-            processed_header.push(h)
-        end
-    end
-    header = processed_header
-    data.delete_at(0);
-    puts "Seeding " + data.size.to_s + " record" + (data.size > 1 ? "s" : "")
-    line = 1
-    data.each do |d|
-        line+=1
-        if d.size == header.size
-            code = "model = " + modelname + ".new\n"
-            for count in 0..(header.size - 1) do
-                unless (header[count].strip == "nil")
-                    value = d[count]
-                    value = "" if value.nil?
-                    value = "'" + value.strip.gsub(/'/, "\\\\'") + "'"
-                    if evaluations[count].nil?
-                        assignment = value
-                    else
-                        assignment = evaluations[count].split("?")
-                        assignment = assignment.join(value)
-                    end
-                    code += "model." + header[count].strip + "=" + assignment + "\n"
-                end
-            end
-            # Add in the statics
-            statics.each do |s|
-                code += "model." + s + "\n"
-            end
-            code += "unless model.save\n"
-                code += "print_errors(model.errors, " + line.to_s + ")"
-            code += "end\n"
-            eval code
-            #puts (line - 1).to_s + "/" + data.size.to_s
-        else
-            puts "Skipping line " + line.to_s + " with mismatch in number of fields (" + d.size.to_s + ")"
-        end
-    end
-    puts " Done"
-end
-
-def print_errors(errors, line)
-    puts "\nThere were errors on line " + line.to_s
-    errors.each do |e|
-        puts e.to_s
-    end
-end

data/lib/tasks/active_seed_tasks.rake

@@ -1,24 +1,9 @@
 namespace :db do
-  desc "Load seed files (from db/seeds) into the current environment's database." 
+  desc "Load seed files (from db/seeds) into the current environment's database."
   task :active_seed => :environment do
-    require "active_seed/seed_csv"
-    set = ENV["set"] 
+    require "active_seed"
+    set = ENV["set"]
     set = ::Rails.env unless !set.nil?
-    set_file = File.join(::Rails.root.to_s, "db", "active_seed", set + ".yml")
-    if !File.exists?(set_file)
-    	puts "Set file doesn't exist: " << set_file
-	    return
-    end
-    puts "Seeding from set '" + set + "'"
-    fixture_list = YAML::load_file(set_file)
-    fixture_list.each do |model, sf| 
-	    seed_file = File.join(::Rails.root.to_s, "db", "active_seed", "data", sf + ".csv")
-    	if !File.exists?(seed_file)
-		    puts "Seed file doesn't exist: " << seed_file
-    	else
-		    puts "Seeding '" + seed_file + "'..." 
-		    seed_csv(model, seed_file) 
-	    end
-    end
+    ActiveSeed.seed(set)
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: active_seed
 version: !ruby/object:Gem::Version 
-  hash: 17
+  hash: 31
   prerelease: 
   segments: 
   - 1
   - 0
-  - 3
-  version: 1.0.3
+  - 4
+  version: 1.0.4
 platform: ruby
 authors: 
 - David Monagle
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-04-12 00:00:00 +10:00
+date: 2011-05-16 00:00:00 +10:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -65,6 +65,7 @@ files:
 - VERSION
 - active_seed.gemspec
 - lib/active_seed.rb
+- lib/active_seed/active_seed.rb
 - lib/active_seed/engine.rb
 - lib/active_seed/seed_csv.rb
 - lib/generators/active_seed/example_html_colors_generator.rb