seed_dump 0.4.3 → 0.5.0

data/README.rdoc

@@ -42,7 +42,7 @@ Use another output file instead of db/seeds.rb
 
   rake db:seed:dump FILE=db/categories.rb
 
-By default the :id column will not be added to the generated Create statements
+By default the :id column will not be added to the generated create statements.
 If you do want the :id to be included use WITH_ID:
 
   rake db:seed:dump WITH_ID=1
@@ -54,15 +54,22 @@ It's up to you to edit these and change the values into something meaningful:
 
   rake db:seed:dump MODEL=User NO_DATA=1 APPEND=true 
 
+If you want the dump to use `create!` rather than `create`:
+
+  rake db:seed:dump CREATE_METHOD='create!'
+
 Here is a full example using all of the options above:
 
-  rake db:seed:dump MODELS=Category LIMIT=10 APPEND=true FILE=db/categories.rb WITH_ID=1 NO_DATA=1
+  rake db:seed:dump MODELS=Category LIMIT=10 APPEND=true FILE=db/categories.rb WITH_ID=1 NO_DATA=1 CREATE_METHOD='create!'
 
 == All environment variables
 
 APPEND:
  Append the data to db/seeds.rb instead of overwriting it.
 
+CREATE_METHOD:
+ Use the specified create method rather than 'create' (the default).  Note: if you are using bash and want to use 'create!', be sure to use single quotes on the command line (i.e. CREATE_METHOD='create!').
+
 FILE:
  Use a different output file, default: db/seeds.rb
 

data/VERSION

@@ -1 +1 @@
-0.4.3
+0.5.0

data/lib/seed_dump/perform.rb

@@ -6,11 +6,11 @@ module SeedDump
 
     def initialize
       @opts = {}
-      @ar_options = {} 
+      @ar_options = {}
       @indent = ""
       @models = []
       @last_record = []
-      @seed_rb = "" 
+      @seed_rb = ""
       @id_set_string = ""
       @model_dir = 'app/models/**/*.rb'
     end
@@ -33,6 +33,7 @@ module SeedDump
       @opts['models']  = @opts['models'].split(',').collect {|x| x.underscore.singularize.camelize }
       @opts['schema']  = env['PG_SCHEMA']
       @opts['model_dir']  = env['MODEL_DIR'] || @model_dir
+      @opts['create_method']  = env['CREATE_METHOD'] || 'create'
     end
 
     def loadModels
@@ -57,7 +58,7 @@ module SeedDump
         require f
 
         puts "Detected model #{model}" if @opts['debug']
-        @models.push model if @opts['models'].include?(model) || @opts['models'].empty? 
+        @models.push model if @opts['models'].include?(model) || @opts['models'].empty?
       end
     end
 
@@ -74,13 +75,13 @@ module SeedDump
         v = v.to_s
       else
         v = attribute_for_inspect(r,k)
-      end 
+      end
 
       unless k == 'id' && !@opts['with_id']
         if (!(k == 'created_at' || k == 'updated_at') || @opts['timestamps'])
           a_s.push("#{k.to_sym.inspect} => #{v}")
         end
-      end 
+      end
     end
 
     def dumpModel(model)
@@ -91,34 +92,34 @@ module SeedDump
       rows = []
       arr = []
       arr = model.find(:all, @ar_options) unless @opts['no-data']
-      arr = arr.empty? ? [model.new] : arr 
+      arr = arr.empty? ? [model.new] : arr
 
-      arr.each_with_index { |r,i| 
+      arr.each_with_index { |r,i|
         attr_s = [];
         r.attributes.each do |k,v|
           if ((model.attr_accessible[:default].include? k) || @opts['without_protection'] || @opts['with_id'])
             dumpAttribute(attr_s,r,k,v)
-            @last_record.push k 
+            @last_record.push k
           end
         end
         rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
-      } 
+      }
 
       if @opts['without_protection']
         options = ', :without_protection => true '
       end
-      
+
       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 << "\n#{model}.#{@opts['create_method']}([\n" << sr.join(",\n") << "\n]#{options})\n"
+        end
         maxsarr.join('')
       else
-        "\n#{model}.create([\n" << rows.join(",\n") << "\n]#{options})\n"
+        "\n#{model}.#{@opts['create_method']}([\n" << rows.join(",\n") << "\n]#{options})\n"
       end
-      
+
     end
 
     def dumpModels
@@ -152,7 +153,7 @@ module SeedDump
     #override the rails version of this function to NOT truncate strings
     def attribute_for_inspect(r,k)
       value = r.attributes[k]
-      
+
       if value.is_a?(String) && value.length > 50
         "#{value}".inspect
       elsif value.is_a?(Date) || value.is_a?(Time)

data/seed_dump.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "seed_dump"
-  s.version = "0.4.3"
+  s.version = "0.5.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff"]
-  s.date = "2013-05-11"
+  s.date = "2013-06-12"
   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 = [

data/test/seed_dump_test.rb

@@ -10,20 +10,20 @@ class SeedDumpTest < ActiveSupport::TestCase
      "MODEL_DIR" => 'test/models/**.rb',
      "FILE" => Dir.pwd + '/test/db/seeds.rb',
      "VERBOSE" => false,
-     "DEBUG" => false 
+     "DEBUG" => false
     }
   end
 
   test "load sample model" do
     @env['MODEL_DIR'] = 'test/models/*.rb'
-    @sd.setup @env 
+    @sd.setup @env
     @sd.loadModels
     assert_equal ["AbstractSample", "ChildSample", "Sample"], @sd.models
   end
 
   test "support nested models" do
     @env['MODEL_DIR'] = 'test/models/**/*.rb'
-    @sd.setup @env 
+    @sd.setup @env
     @sd.loadModels
     assert_equal ["AbstractSample", "ChildSample", "Nested::Sample", "Sample"], @sd.models
   end
@@ -65,5 +65,12 @@ class SeedDumpTest < ActiveSupport::TestCase
     assert_equal [], @sd.last_record
   end
 
-
+  test "create method" do
+    @env['MODEL_DIR'] = 'test/models/*.rb'
+    @env['CREATE_METHOD'] = 'create!'
+    @sd.setup @env
+    @sd.loadModels
+    @sd.dumpModels
+    assert @sd.instance_variable_get(:@seed_rb) =~ /create!/, 'CREATE_METHOD must specify the creation method'
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-11 00:00:00.000000000 Z
+date: 2013-06-12 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