seed_dump 0.4.2 → 0.4.3

data/CHANGELOG.rdoc

@@ -1,5 +1,14 @@
 == Seed Dump
 
+== 0.4.3 / 2013-05-11
+
+ * Handle abstract_class setting in ActiveRecord [craigjackson]
+ * added MAX option [OlegPasko]
+
+== 0.4.2 / 2012-11-02
+
+ * Fix for bug resulting from top-level class redefinition [rroblak]
+
 == 0.4.1 / 2012-08-01
 
  * include TIMESTAMPS by default

data/README.rdoc

@@ -68,6 +68,10 @@ FILE:
 
 LIMIT:
  Dump no more then this amount of data, default: no limit
+ 
+MAX:
+ Split one create action per model to several create actions with MAX elements in each, default: no limit
+ It usefull for large data dumping to reduce memory usage
 
 MODEL(S):
  A model name or a comma seperated list of models, default: all models

data/VERSION

@@ -1 +1 @@
-0.4.2
+0.4.3

data/lib/seed_dump/perform.rb

@@ -27,6 +27,7 @@ module SeedDump
       @opts['models']  = env['MODELS'] || (env['MODEL'] ? env['MODEL'] : "")
       @opts['file']    = env['FILE'] || "#{Rails.root}/db/seeds.rb"
       @opts['append']  = (env['APPEND'].true? && File.exists?(@opts['file']) )
+      @opts['max']     = env['MAX'] && env['MAX'].to_i > 0 ? env['MAX'].to_i : nil
       @ar_options      = env['LIMIT'].to_i > 0 ? { :limit => env['LIMIT'].to_i } : {}
       @indent          = " " * (env['INDENT'].nil? ? 2 : env['INDENT'].to_i)
       @opts['models']  = @opts['models'].split(',').collect {|x| x.underscore.singularize.camelize }
@@ -106,15 +107,25 @@ module SeedDump
       if @opts['without_protection']
         options = ', :without_protection => true '
       end
-
-      "\n#{model}.create([\n" << rows.join(",\n") << "\n]#{options})\n"
+      
+      if @opts['max']
+        splited_rows = rows.each_slice(@opts['max']).to_a
+        maxsarr = []
+        splited_rows.each do |sr|
+          maxsarr << "\n#{model}.create([\n" << sr.join(",\n") << "\n]#{options})\n"
+        end 
+        maxsarr.join('')
+      else
+        "\n#{model}.create([\n" << rows.join(",\n") << "\n]#{options})\n"
+      end
+      
     end
 
     def dumpModels
       @seed_rb = ""
       @models.sort.each do |model|
           m = model.constantize
-          if m.ancestors.include?(ActiveRecord::Base)
+          if m.ancestors.include?(ActiveRecord::Base) && !m.abstract_class
             puts "Adding #{model} seeds." if @opts['verbose']
 
             if @opts['skip_callbacks']

data/seed_dump.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "seed_dump"
-  s.version = "0.4.2"
+  s.version = "0.4.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff"]
-  s.date = "2012-11-29"
+  s.date = "2013-05-11"
   s.description = "Dump (parts) of your database to db/seeds.rb to get a headstart creating a meaningful seeds.rb file"
   s.email = "rob.halff@gmail.com"
   s.extra_rdoc_files = [
@@ -29,6 +29,8 @@ Gem::Specification.new do |s|
     "lib/true.rb",
     "seed_dump.gemspec",
     "test/fixtures/samples.yml",
+    "test/models/abstract_sample.rb",
+    "test/models/child_sample.rb",
     "test/models/nested/sample.rb",
     "test/models/sample.rb",
     "test/seed_dump_test.rb",
@@ -36,7 +38,7 @@ Gem::Specification.new do |s|
   ]
   s.homepage = "http://github.com/rhalff/seed_dump"
   s.require_paths = ["lib"]
-  s.rubygems_version = "1.8.24"
+  s.rubygems_version = "1.8.25"
   s.summary = "{Seed Dumper for Rails}"
 
   if s.respond_to? :specification_version then

data/test/models/abstract_sample.rb

@@ -0,0 +1,3 @@
+class AbstractSample < ActiveRecord::Base
+  self.abstract_class = true
+end

data/test/models/child_sample.rb

@@ -0,0 +1,4 @@
+class ChildSample < AbstractSample
+  attr_accessible :name
+end
+

data/test/seed_dump_test.rb

@@ -18,14 +18,14 @@ class SeedDumpTest < ActiveSupport::TestCase
     @env['MODEL_DIR'] = 'test/models/*.rb'
     @sd.setup @env 
     @sd.loadModels
-    assert_equal ["Sample"], @sd.models
+    assert_equal ["AbstractSample", "ChildSample", "Sample"], @sd.models
   end
 
   test "support nested models" do
     @env['MODEL_DIR'] = 'test/models/**/*.rb'
     @sd.setup @env 
     @sd.loadModels
-    assert_equal ["Nested::Sample", "Sample"], @sd.models
+    assert_equal ["AbstractSample", "ChildSample", "Nested::Sample", "Sample"], @sd.models
   end
 
   test "without timestamps" do
@@ -55,5 +55,15 @@ class SeedDumpTest < ActiveSupport::TestCase
     assert @sd.last_record.include?("id"), "WITH_ID must include id"
   end
 
+  test "skip abstract model" do
+    @env['MODELS'] = "AbstractSample"
+    @env['MODEL_DIR'] = 'test/models/*.rb'
+    @env['TIMESTAMPS'] = false
+    @sd.setup @env
+    @sd.loadModels
+    @sd.dumpModels
+    assert_equal [], @sd.last_record
+  end
+
 
 end

data/test/test_helper.rb

@@ -9,7 +9,13 @@ require 'active_record'
 ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
 
 ActiveRecord::Schema.define(:version => 1) do
- create_table "samples", :force => true do |t|
+  create_table "child_samples", :force => true do |t|
+    t.string   "name"
+    t.datetime "created_at", :null => false
+    t.datetime "updated_at", :null => false
+  end
+
+  create_table "samples", :force => true do |t|
     t.string   "string"
     t.text     "text"
     t.integer  "integer"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.4.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-29 00:00:00.000000000 Z
+date: 2013-05-11 00:00:00.000000000 Z
 dependencies: []
 description: Dump (parts) of your database to db/seeds.rb to get a headstart creating
   a meaningful seeds.rb file
@@ -32,6 +32,8 @@ files:
 - lib/true.rb
 - seed_dump.gemspec
 - test/fixtures/samples.yml
+- test/models/abstract_sample.rb
+- test/models/child_sample.rb
 - test/models/nested/sample.rb
 - test/models/sample.rb
 - test/seed_dump_test.rb
@@ -56,7 +58,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.24
+rubygems_version: 1.8.25
 signing_key: 
 specification_version: 3
 summary: ! '{Seed Dumper for Rails}'