seed_scatter 0.0.1 → 0.0.2

data/README.md

@@ -35,10 +35,12 @@ To dump all of your models' information from the database to their respective se
 
     $ rake db:harvest
 
+To re-seed a specific model:
+
+    $ rake db:reseed[MODEL_NAME]
 
 ## Todos
 
-* Re-seed models one at a time
 * Support for namespaced models
 * Support for join tables
 

data/lib/seed_scatter/harvest.rb

@@ -6,21 +6,32 @@ module SeedScatter
 
     def init(model)
       models = if model == :all
-                 Dir['app/models/*.rb'].map {|f| File.basename(f, '.*').camelize.constantize.name }
+                 Dir['app/models/*.rb'].map {|f| File.basename(f, '.*').camelize.constantize }
                else
-                 [model]
+                 [model.camelize.constantize]
                end
       models.each do |m|
-        @model = m.underscore.singularize
+        @model = m
         puts "Harvesting seeds for #{class_name}"
         @harvest_data = []
         dump_model(m)
       end
     end
 
+    def reseed(model)
+      @model = m = model.camelize.constantize
+      truncate(m.table_name)
+      f = File.expand_path(File.join("db","seeds","#{m.table_name}.rb"))
+      puts "Reseeding #{f}"
+      load(f)
+    end
+
+    def truncate(table)
+      ActiveRecord::Base.connection.execute("TRUNCATE #{table}")
+    end
+
     def collect_data(model)
-      klass = model.camelize.constantize
-      klass.all.each do |m|
+      model.all.each do |m|
         @harvest_data << ActiveSupport::JSON.decode(m.attributes.to_json).symbolize_keys
       end
     end
@@ -35,11 +46,11 @@ module SeedScatter
     end
 
     def class_name
-      @model.camelize
+      @model.name
     end
 
     def destination_path(model)
-      "db/seeds/#{model.downcase.pluralize}.rb"
+      "db/seeds/#{model.table_name}.rb"
     end
   end
 

data/lib/seed_scatter/version.rb

@@ -1,3 +1,3 @@
 module SeedScatter
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/tasks/seed_scatter.rake

@@ -7,4 +7,7 @@ namespace :db do
     args.with_defaults(:model => :all)
     SeedScatter::Harvester.new.init args[:model]
   end
+  task :reseed, [:model] => :environment do |t,args|
+    SeedScatter::Harvester.new.reseed args[:model]
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_scatter
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-19 00:00:00.000000000 Z
+date: 2013-07-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails