jcnetdev-seed-fu 1.0.200807042 → 1.1

data/lib/seed-fu.rb

@@ -3,17 +3,23 @@ module SeedFu
     def self.plant(model_class, *constraints, &block)
       constraints = [:id] if constraints.empty?
       seed = Seeder.new(model_class)
+
+      options = constraints.last if (constraints.last.is_a? Hash)
+      constraints.delete_at(*constraints.length-1) if (constraints.last.is_a? Hash)
+
+      insert_only = constraints.last.is_a? TrueClass
+      constraints.delete_at(*constraints.length-1) if (constraints.last.is_a? TrueClass or constraints.last.is_a? FalseClass)
       seed.set_constraints(*constraints)
       yield seed
-      seed.plant!
+      seed.plant!({:insert_only => insert_only}.merge(options||{}))
     end
-  
+
     def initialize(model_class)
       @model_class = model_class
       @constraints = [:id]
       @data = {}
     end
-  
+
     def set_constraints(*constraints)
       raise "You must set at least one constraint." if constraints.empty?
       @constraints = []
@@ -22,21 +28,24 @@ module SeedFu
         @constraints << constraint.to_sym
       end
     end
-  
+
     def set_attribute(name, value)
       @data[name.to_sym] = value
     end
-  
-    def plant!
+
+    def plant!(options)
+      insert_only = options[:insert_only].nil? ? false : options[:insert_only]
+      enable_validation = options[:validate].nil? ? true : options[:validate]
       record = get
+      return if !record.new_record? and insert_only
       @data.each do |k, v|
         record.send("#{k}=", v)
       end
-      record.save!
-      puts " - #{@model_class} #{condition_hash.inspect}"
+      record.save(enable_validation) || raise(RecordNotSaved)
+      puts " - #{@model_class} #{condition_hash.inspect}"      
       record
     end
-  
+
     def method_missing(method_name, *args) #:nodoc:
       if (match = method_name.to_s.match(/(.*)=$/)) && args.size == 1
         set_attribute(match[1], args.first)
@@ -44,9 +53,9 @@ module SeedFu
         super
       end
     end
-  
+
     protected
-  
+
     def get
       records = @model_class.find(:all, :conditions => condition_hash)
       raise "Given constraints yielded multiple records." unless records.size < 2
@@ -56,7 +65,7 @@ module SeedFu
         return @model_class.new
       end
     end
-  
+
     def condition_hash
       @data.reject{|a,v| !@constraints.include?(a)}
     end
@@ -73,4 +82,15 @@ class ActiveRecord::Base
   def self.seed(*constraints, &block)
     SeedFu::Seeder.plant(self, *constraints, &block)
   end
+
+  def self.seed_many(*constraints)
+    seeds = constraints.pop
+    seeds.each do |seed_data|
+      seed(*constraints) do |s|
+        seed_data.each_pair do |k,v|
+          s.send "#{k}=", v
+        end
+      end
+    end
+  end
 end

data/seed-fu.gemspec

@@ -1,9 +1,9 @@
 Gem::Specification.new do |s|
   s.name = 'seed-fu'
-  s.version = '1.0.200807042'
-  s.date = '2008-07-04'
+  s.version = '1.1'
+  s.date = '2008-09-17'
   
-  s.summary = "Allows easier database seeding of tables"
+  s.summary = "Allows easier database seeding of tables."
   s.description = "Seed Fu is an attempt to once and for all solve the problem of inserting and maintaining seed data in a database. It uses a variety of techniques gathered from various places around the web and combines them to create what is hopefully the most robust seed data system around."
   
   s.authors = ["Michael Bleigh"]

data/spec/seed_fu_spec.rb

@@ -43,6 +43,18 @@ describe SeedFu::Seeder do
     SeededModel.find_by_login("bob").first_name.should == "Steve"
   end
   
+  it "should be able to create models from an array of seed attributes" do
+    SeededModel.seed_many(:title, :login, [
+      {:login => "bob", :title => "Peon", :first_name => "Steve"},
+      {:login => "frank", :title => "Peasant", :first_name => "Francis"},
+      {:login => "harry", :title => "Noble", :first_name => "Harry"}
+    ])
+    
+    SeededModel.find_by_login("bob").first_name.should == "Steve"
+    SeededModel.find_by_login("frank").first_name.should == "Francis"
+    SeededModel.find_by_login("harry").first_name.should == "Harry"
+  end
+  
   #it "should raise an error if constraints are not unique" do
   #  SeededModel.create(:login => "bob", :first_name => "Bob", :title => "Peon")
   #  SeededModel.create(:login => "bob", :first_name => "Robert", :title => "Manager")

data/tasks/seed_fu_tasks.rake

@@ -1,12 +1,13 @@
 namespace :db do
   desc "Loads seed data from db/fixtures for the current environment."
   task :seed => :environment do
-    Dir[File.join(RAILS_ROOT, 'db', 'fixtures', '*.rb')].sort.each { |fixture| 
+    fixture_path = ENV["FIXTURE_PATH"] ? ENV["FIXTURE_PATH"] : "db/fixtures"
+    Dir[File.join(RAILS_ROOT, fixture_path, '*.rb')].sort.each { |fixture| 
       puts "\n== Seeding from #{File.split(fixture).last} " + ("=" * (60 - (17 + File.split(fixture).last.length)))
       load fixture 
       puts "=" * 60 + "\n"
     }
-    Dir[File.join(RAILS_ROOT, 'db', 'fixtures', RAILS_ENV, '*.rb')].sort.each { |fixture| 
+    Dir[File.join(RAILS_ROOT, fixture_path, RAILS_ENV, '*.rb')].sort.each { |fixture| 
       puts "\n== [#{RAILS_ENV}] Seeding from #{File.split(fixture).last} " + ("=" * (60 - (20 + File.split(fixture).last.length + RAILS_ENV.length)))
       load fixture 
       puts "=" * 60 + "\n"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jcnetdev-seed-fu
 version: !ruby/object:Gem::Version 
-  version: 1.0.200807042
+  version: "1.1"
 platform: ruby
 authors: 
 - Michael Bleigh
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2008-07-04 00:00:00 -07:00
+date: 2008-09-17 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -67,6 +67,6 @@ rubyforge_project:
 rubygems_version: 1.2.0
 signing_key: 
 specification_version: 2
-summary: Allows easier database seeding of tables
+summary: Allows easier database seeding of tables.
 test_files: []