seed_dump 2.0.0 → 3.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1a7a3bc7780fa30de0f2ffccb171249879058fa5
-  data.tar.gz: dad1c603abb084874344d01263e6dae6d88ea03d
+  metadata.gz: 616a5857e516b0f3dbeacfa6ccc958481edd2af3
+  data.tar.gz: d92c064c2970dc8ea0743a57db2749721a9a501e
 SHA512:
-  metadata.gz: 9e1f0bfa9e904d6b0b6756b45caa39afee9576f5661c155485d23f430ff87509f488f6b95e247e4ede7f3b41171229b9e49e989677acb3b658be5345e343f4d5
-  data.tar.gz: 74606c3f956c8b5118e83edea597f152e8e02b7913ee1af6ffcba3f9c63f0700d0f34042bfe50f2e536b72b29bfcb1a27d4c58682d9f16926b3699ff7afe1655
+  metadata.gz: 46f9fcdcb191fef7bfd6c880662bc3a4ac2b84ae29d82d8d2d5dc24095117d074c7d2d27c50670530ce929fd1b21d5a71297b0b3cc173eb3aa7745126991c093
+  data.tar.gz: d141c0da2700d4cf927f10ec7e46b64e2a53e5df7a85addc00ac4ead7fbebbc9feebe761d2d8460c3254278bcc893126a9e84e66be4aa98c70ca5ea818fe4aee

data/Gemfile

@@ -5,6 +5,7 @@ gem 'activerecord'
 
 group :development, :test do
   gem 'byebug'
+  gem 'factory_girl'
 end
 
 group :development do

data/MIT-LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2010 Rob Halff
+Copyright (c) 2013 Ryan Oblak
 
 Permission is hereby granted, free of charge, to any person obtaining
 a copy of this software and associated documentation files (the

data/README.md

@@ -3,35 +3,51 @@ Seed Dump
 
 Seed Dump is a Rails 4 plugin that adds a rake task named `db:seed:dump`.
 
-It allows you to create a `db/seeds.rb` from the existing data in your database.
+It allows you to create seed data files from the existing data in your database.
+
+You can also use Seed Dump from the Rails console.  See below for usage examples.
 
 Note: if you want to use Seed Dump with Rails 3 or earlier, use [version 0.5.3](http://rubygems.org/gems/seed_dump/versions/0.5.3).
 
-Example Usage
--------------
+Installation
+------------
+
+Add it to your Gemfile with:
+
+    gem 'seed_dump'
+
+Or install it by hand:
+
+    $ gem install seed_dump
+
+Examples
+--------
+
+### Rake task
 
 Dump all data directly to `db/seeds.rb`:
 
     $ rake db:seed:dump
 
-Dump only data from the users and products table and dump a maximum amount of 2 records:
+Result:
 
-    $ rake db:seed:dump MODELS=User,Product LIMIT=2
+    Product.create!([
+      {category_id: 1, description: "Long Sleeve Shirt", name: "Long Sleeve Shirt"},
+      {category_id: 3, description: "Plain White Tee Shirt", name: "Plain T-Shirt"}
+    ])
+    User.create!([
+      {id: 1, password: "123456", username: "test_1"},
+      {id: 2, password: "234567", username: "test_2"}
+    ])
 
-Result:
+Dump only data from the users table and dump a maximum amount of 1 record:
 
-    $ cat db/seeds.rb
-    # Autogenerated by the db:seed:dump task
-    # Do not hesitate to tweak this to your needs
+    $ rake db:seed:dump MODELS=User,Product LIMIT=1
 
-    products = Product.create!([
-      { :category_id => 1, :description => "Long Sleeve Shirt", :name => "Long Sleeve Shirt" },
-      { :category_id => 3, :description => "Plain White Tee Shirt", :name => "Plain T-Shirt" }
-    ])
+Result:
 
-    users = User.create!([
-      { :id => 1, :password => "123456", :username => "test_1" },
-      { :id => 2, :password => "234567", :username => "tes2" }
+    User.create!([
+      {id: 1, password: "123456", username: "test_1"}
     ])
 
 Append to `db/seeds.rb` instead of overwriting it:
@@ -40,42 +56,42 @@ Append to `db/seeds.rb` instead of overwriting it:
 
 Use another output file instead of `db/seeds.rb`:
 
-    rake db:seed:dump FILE=db/categories.rb
-
-If you want the dump to use `create` rather than `create!`:
-
-    rake db:seed:dump CREATE_METHOD='create'
+    rake db:seed:dump FILE=db/seeds/users.rb
 
-There are more environment variables that can be set— see below for all of them.
+There are more options that can be set— see below for all of them.
 
+### Console
 
-Installation
-------------
+Output a dump of all User records:
 
-Add it to your Gemfile with:
+    irb(main):001:0> puts SeedDump.dump(User)
+    User.create!([
+      {id: 1, password: "123456", username: "test_1"},
+      {id: 2, password: "234567", username: "test_2"}
+    ])
 
-    gem 'seed_dump'
+Write the dump to a file:
 
-Or install it by hand:
+    irb(main):002:0> SeedDump.dump(User, file: 'db/seeds.rb')
 
-    $ gem install seed_dump
+Append the dump to a file:
 
+    irb(main):003:0> puts SeedDump.dump(User, file: 'db/seeds.rb', append: true)
 
-All environment variables
--------------------------
+Options are specified as a Hash to the second argument.
 
-`APPEND`: If set to `true`, append the data to the file instead of overwriting it.  Default: `false`.
+Options
+-------
 
-`CREATE_METHOD`: Use the specified create method rather than `create!` (the default).
+Options are common to both the Rake task and the console, except where noted.
 
-`FILE`: Use a different output file.  Default: `db/seeds.rb`.
+`append`: If set to `true`, append the data to the file instead of overwriting it.  Default: `false`.
 
-`LIMIT`: Dump no more then this amount of data.  Default: no limit.
+`exclude`: Attributes to be excluded from the dump.  Default: `id, created_at, updated_at`.
 
-`MAX`: Split one create action per model into several create actions with MAX elements in each.  Default: no limit.  Useful for large data dumping to reduce memory usage.
+`file`: Write to the specified output file.  Default in Rake task is `db/seeds.rb`.  Console returns the dump as a string by default.
 
-`MODEL[S]`: A model name or a comma-separated list of models.  Default: all models.
+`limit`: Dump no more then this amount of data.  Default: no limit.  Rake task only.  In the console just pass in an ActiveRecord::Relation with the appropriate limit (e.g. `SeedDump.dump(User.limit(5))`).
 
-`WITH_ID`: If set to `true`, include the `:id` in the `create` options.  Default: `false`.
+`model[s]`: Restrict the dump to the specified comma-separated list of models.  Default: all models.  Rake task only.
 
-`TIMESTAMPS`: If set to `true`, include the `:created_by` and `:updated_by` timestamps.  Default: `true`.

data/Rakefile

@@ -10,6 +10,7 @@ begin
     gem.email = 'rroblak@gmail.com'
     gem.homepage = 'https://github.com/rroblak/seed_dump'
     gem.authors = ['Rob Halff', 'Ryan Oblak']
+    gem.license = 'MIT'
   end
   Jeweler::GemcutterTasks.new
 rescue LoadError

data/VERSION

@@ -1 +1 @@
-2.0.0
+3.0.0

data/lib/seed_dump.rb

@@ -1,7 +1,10 @@
+require 'seed_dump/dump_methods/enumeration'
 require 'seed_dump/dump_methods'
+require 'seed_dump/environment'
 
 class SeedDump
-  include SeedDump::DumpMethods
+  extend Environment
+  extend DumpMethods
 
   require 'seed_dump/railtie' if defined?(Rails)
 end

data/lib/seed_dump/dump_methods.rb

@@ -1,151 +1,96 @@
-require "true"
-require "clip"
-
 class SeedDump
   module DumpMethods
+    include Enumeration
 
-    def initialize
-      @opts = {}
-      @ar_options = {}
-      @indent = ""
-      @models = []
-      @seed_rb = ""
-      @id_set_string = ""
-    end
+    def dump(records, options = {})
+      return nil if records.count == 0
 
-    def setup(env)
-      # config
-      @opts['verbose'] = env["VERBOSE"].true? || env['VERBOSE'].nil?
-      @opts['debug'] = env["DEBUG"].true?
-      @opts['with_id'] = env["WITH_ID"].true?
-      @opts['timestamps'] = env["TIMESTAMPS"].true? || env["TIMESTAMPS"].nil?
-      @opts['models']  = env['MODELS'] || 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
-      @indent          = " " * (env['INDENT'].nil? ? 2 : env['INDENT'].to_i)
-      @opts['create_method']  = env['CREATE_METHOD'] || 'create!'
-
-      @limit = (env['LIMIT'].to_i > 0) ? env['LIMIT'].to_i : nil
-
-      @models = @opts['models'].split(',').collect {|x| x.strip.underscore.singularize.camelize.constantize }
-    end
+      io = open_io(options)
 
-    def log(msg)
-      puts msg if @opts['debug']
-    end
+      write_records_to_io(records, io, options)
 
-    def models
-      @models
+      ensure
+        io.close if io.present?
     end
 
-    def dump_attribute(a_s, r, k, v)
-      pushed = false
-      if v.is_a?(BigDecimal)
-        v = v.to_s
-      else
-        v = attribute_for_inspect(r,k)
-      end
+    private
 
-      unless k == 'id' && !@opts['with_id']
-        if (!(k == 'created_at' || k == 'updated_at') || @opts['timestamps'])
-          a_s.push("#{k.to_sym.inspect} => #{v}")
-          pushed = true
-        end
-      end
-      pushed
-    end
+    def dump_record(record, options)
+      attribute_strings = []
 
-    def dump_model(model)
-      @id_set_string = ''
-      create_hash = ""
-      rows = []
+      options[:exclude] ||= [:id, :created_at, :updated_at]
 
+      # We select only string attribute names to avoid conflict
+      # with the composite_primary_keys gem (it returns composite
+      # primary key attribute names as hashes).
+      record.attributes.select {|key| key.is_a?(String) }.each do |attribute, value|
+        attribute_strings << dump_attribute_new(attribute, value) unless options[:exclude].include?(attribute.to_sym)
+      end
 
-      model.find_each(batch_size: (@limit || 1000)) do |record|
-        attr_s = [];
+      "{#{attribute_strings.join(", ")}}"
+    end
 
-        record.attributes.select {|x| x.is_a?(String) }.each do |k,v|
-          dump_attribute(attr_s, record, k, v)
-        end
+    def dump_attribute_new(attribute, value)
+      "#{attribute}: #{value_to_s(value)}"
+    end
 
-        rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
+    def value_to_s(value)
+      value = case value
+              when BigDecimal
+                value.to_s
+              when Date, Time, DateTime
+                value.to_s(:db)
+              else
+                value
+              end
+
+      value.inspect
+    end
 
-        break if rows.length == @limit
-      end
+    def open_io(options)
+      if options[:file].present?
+        mode = options[:append] ? 'a+' : 'w+'
 
-      if @opts['max']
-        splited_rows = rows.each_slice(@opts['max']).to_a
-        maxsarr = []
-        splited_rows.each do |sr|
-          maxsarr << "\n#{model}.#{@opts['create_method']}([\n" << sr.join(",\n") << "\n])\n"
-        end
-        maxsarr.join('')
+        File.open(options[:file], mode)
       else
-        "\n#{model}.#{@opts['create_method']}([\n" << rows.join(",\n") << "\n])\n"
+        StringIO.new('', 'w+')
       end
-
     end
 
-    def dump_models
-      if @models.empty?
-        Rails.application.eager_load!
+    def write_records_to_io(records, io, options)
+      io.write("#{model_for(records)}.create!([\n  ")
 
-        @seed_rb = ""
+      enumeration_method = if records.is_a?(ActiveRecord::Relation) || records.is_a?(Class)
+                             :active_record_enumeration
+                           else
+                             :enumerable_enumeration
+                           end
 
-        @models = ActiveRecord::Base.descendants.select do |model|
-                    (model.to_s != 'ActiveRecord::SchemaMigration') && \
-                     model.table_exists? && \
-                     model.exists?
-        end
-      end
-
-      @models.sort! { |a, b| a.to_s <=> b.to_s }
+      send(enumeration_method, records, io, options) do |record_strings, last_batch|
+        io.write(record_strings.join(",\n  "))
 
-      @models.each do |model|
-        puts "Adding #{model} seeds." if @opts['verbose']
-
-        @seed_rb << dump_model(model) << "\n\n"
+        io.write(",\n  ") unless last_batch
       end
 
-      @seed_rb
-    end
-
-    def write_file
-      File.open(@opts['file'], (@opts['append'] ? "a" : "w")) { |f|
-        f << "# encoding: utf-8\n"
-        f << "# Autogenerated by the db:seed:dump task\n# Do not hesitate to tweak this to your needs\n" unless @opts['append']
-        f << "#{@seed_rb}"
-      }
-    end
-
-    #override the rails version of this function to NOT truncate strings
-    def attribute_for_inspect(r,k)
-      value = r.attributes[k]
+      io.write("\n])\n")
 
-      if value.is_a?(String) && value.length > 50
-        "#{value}".inspect
-      elsif value.is_a?(Date) || value.is_a?(Time)
-        %("#{value.to_s(:db)}")
+      if options[:file].present?
+        nil
       else
-        value.inspect
+        io.rewind
+        io.read
       end
     end
 
-    def output
-      @seed_rb
+    def model_for(records)
+      if records.is_a?(Class)
+        records
+      elsif records.respond_to?(:model)
+        records.model
+      else
+        records[0].class
+      end
     end
 
-    def run(env)
-      setup env
-
-      puts "Appending seeds to #{@opts['file']}." if @opts['append']
-      dump_models
-
-      puts "Writing #{@opts['file']}."
-      write_file
-
-      puts "Done."
-    end
   end
 end

data/lib/seed_dump/dump_methods/enumeration.rb

@@ -0,0 +1,73 @@
+class SeedDump
+  module DumpMethods
+    module Enumeration
+      def active_record_enumeration(records, io, options)
+        # If the records don't already have an order,
+        # order them by primary key ascending.
+        if !records.respond_to?(:arel) || records.arel.orders.blank?
+          records.order("#{records.quoted_table_name}.#{records.quoted_primary_key} ASC")
+        end
+
+        num_of_batches, batch_size, last_batch_size = batch_params_from(records, options)
+
+        # Loop through each batch
+        (1..num_of_batches).each do |batch_number|
+
+          record_strings = []
+
+          last_batch = (batch_number == num_of_batches)
+
+          cur_batch_size = if last_batch
+                             last_batch_size
+                           else
+                             batch_size
+                           end
+
+          # Loop through the records of the current batch
+          records.offset((num_of_batches - 1) * batch_size).limit(cur_batch_size).each do |record|
+            record_strings << dump_record(record, options)
+          end
+
+          yield record_strings, last_batch
+        end
+      end
+
+      def enumerable_enumeration(records, io, options)
+        num_of_batches, batch_size = batch_params_from(records, options)
+
+        record_strings = []
+
+        batch_number = 1
+
+        records.each_with_index do |record, i|
+          record_strings << dump_record(record, options)
+
+          last_batch = (i == records.length - 1)
+
+          if (record_strings.length == batch_size) || last_batch
+            yield record_strings, last_batch
+
+            record_strings = []
+            batch_number += 1
+          end
+        end
+      end
+
+      def batch_params_from(records, options)
+        batch_size = batch_size_from(records, options)
+
+        count = records.count
+
+        [((count / batch_size) + 1), batch_size, (count % batch_size)]
+      end
+
+      def batch_size_from(records, options)
+        if options[:batch_size].present?
+          options[:batch_size].to_i
+        else
+          1000
+        end
+      end
+    end
+  end
+end

data/lib/seed_dump/environment.rb

@@ -0,0 +1,33 @@
+class SeedDump
+  module Environment
+
+    def dump_using_environment(env = {})
+      Rails.application.eager_load!
+
+      models = if env['MODEL'] || env['MODELS']
+                 (env['MODEL'] || env['MODELS']).split(',').collect {|x| x.strip.underscore.singularize.camelize.constantize }
+               else
+                 ActiveRecord::Base.descendants.select do |model|
+                   (model.to_s != 'ActiveRecord::SchemaMigration') && \
+                    model.table_exists? && \
+                    model.exists?
+                 end
+               end
+
+      append = (env['APPEND'] == 'true')
+
+      models.each do |model|
+        model = model.limit(env['LIMIT'].to_i) if env['LIMIT']
+
+        SeedDump.dump(model,
+                      append: append,
+                      batch_size: (env['BATCH_SIZE'] ? env['BATCH_SIZE'].to_i : nil),
+                      exclude: (env['EXCLUDE'] ? env['EXCLUDE'].split(',').map {|e| e.strip.to_sym} : nil),
+                      file: (env['FILE'] || 'db/seeds.rb'))
+
+        append = true
+      end
+    end
+  end
+end
+

data/lib/tasks/seed_dump.rake

@@ -3,7 +3,7 @@ namespace :db do
     desc "Dump records from the database into db/seeds.rb"
     task :dump => :environment do
 
-      SeedDump.new.run(ENV)
+      SeedDump.dump_using_environment(ENV)
 
     end
   end

data/seed_dump.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "seed_dump"
-  s.version = "2.0.0"
+  s.version = "3.0.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff", "Ryan Oblak"]
-  s.date = "2013-10-16"
+  s.date = "2013-12-07"
   s.description = "Dump (parts) of your database to db/seeds.rb to get a headstart creating a meaningful seeds.rb file"
   s.email = "rroblak@gmail.com"
   s.extra_rdoc_files = [
@@ -17,24 +17,28 @@ Gem::Specification.new do |s|
   ]
   s.files = [
     ".rspec",
-    "CHANGELOG.rdoc",
     "Gemfile",
     "MIT-LICENSE",
     "README.md",
     "Rakefile",
     "VERSION",
-    "lib/clip.rb",
     "lib/seed_dump.rb",
     "lib/seed_dump/dump_methods.rb",
+    "lib/seed_dump/dump_methods/enumeration.rb",
+    "lib/seed_dump/environment.rb",
     "lib/seed_dump/railtie.rb",
     "lib/tasks/seed_dump.rake",
-    "lib/true.rb",
     "seed_dump.gemspec",
+    "spec/dump_methods_spec.rb",
+    "spec/environment_spec.rb",
+    "spec/factories/another_samples.rb",
+    "spec/factories/samples.rb",
+    "spec/factories/yet_another_samples.rb",
     "spec/helpers.rb",
-    "spec/seed_dump_spec.rb",
     "spec/spec_helper.rb"
   ]
   s.homepage = "https://github.com/rroblak/seed_dump"
+  s.licenses = ["MIT"]
   s.require_paths = ["lib"]
   s.rubygems_version = "2.0.3"
   s.summary = "{Seed Dumper for Rails}"
@@ -46,17 +50,20 @@ Gem::Specification.new do |s|
       s.add_runtime_dependency(%q<activesupport>, [">= 0"])
       s.add_runtime_dependency(%q<activerecord>, [">= 0"])
       s.add_development_dependency(%q<byebug>, [">= 0"])
+      s.add_development_dependency(%q<factory_girl>, [">= 0"])
       s.add_development_dependency(%q<jeweler>, [">= 0"])
     else
       s.add_dependency(%q<activesupport>, [">= 0"])
       s.add_dependency(%q<activerecord>, [">= 0"])
       s.add_dependency(%q<byebug>, [">= 0"])
+      s.add_dependency(%q<factory_girl>, [">= 0"])
       s.add_dependency(%q<jeweler>, [">= 0"])
     end
   else
     s.add_dependency(%q<activesupport>, [">= 0"])
     s.add_dependency(%q<activerecord>, [">= 0"])
     s.add_dependency(%q<byebug>, [">= 0"])
+    s.add_dependency(%q<factory_girl>, [">= 0"])
     s.add_dependency(%q<jeweler>, [">= 0"])
   end
 end

data/spec/dump_methods_spec.rb

@@ -0,0 +1,111 @@
+require 'spec_helper'
+
+describe SeedDump do
+
+  describe '.dump' do
+    before do
+      Rails.application.eager_load!
+
+      create_db
+
+      3.times { FactoryGirl.create(:sample) }
+
+      @expected_output = "Sample.create!([\n  {string: \"string\", text: \"text\", integer: 42, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 42, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 42, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false}\n])\n"
+    end
+
+    context 'without file option' do
+      it 'should return the dump of the models passed in' do
+        SeedDump.dump(Sample).should eq(@expected_output)
+      end
+    end
+
+    context 'with file option' do
+      before do
+        @filename = Dir::Tmpname.make_tmpname(File.join(Dir.tmpdir, 'foo'), nil)
+      end
+
+      after do
+        File.unlink(@filename)
+      end
+
+      it 'should dump the models to the specified file' do
+        SeedDump.dump(Sample, file: @filename)
+
+        File.open(@filename) { |file| file.read.should eq(@expected_output) }
+      end
+
+      context 'with append option' do
+        it 'should append to the file rather than overwriting it' do
+          SeedDump.dump(Sample, file: @filename)
+          SeedDump.dump(Sample, file: @filename, append: true)
+
+          File.open(@filename) { |file| file.read.should eq(@expected_output + @expected_output) }
+        end
+      end
+    end
+
+    context 'ActiveRecord relation' do
+      it 'should return nil if the count is 0' do
+        SeedDump.dump(EmptyModel).should be(nil)
+      end
+
+      context 'with an order parameter' do
+        it 'should dump the models in the specified order' do
+          Sample.delete_all
+          samples = 3.times {|i| FactoryGirl.create(:sample, integer: i) }
+
+          SeedDump.dump(Sample.order('integer DESC')).should eq("Sample.create!([\n  {string: \"string\", text: \"text\", integer: 2, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 1, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 0, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false}\n])\n")
+        end
+      end
+
+      context 'without an order parameter' do
+        it 'should dump the models sorted by primary key ascending' do
+          Sample.delete_all
+          samples = 3.times {|i| FactoryGirl.create(:sample, integer: i) }
+
+          SeedDump.dump(Sample).should eq("Sample.create!([\n  {string: \"string\", text: \"text\", integer: 0, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 1, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {string: \"string\", text: \"text\", integer: 2, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false}\n])\n")
+        end
+      end
+
+      context 'with a limit parameter' do
+        it 'should dump the number of models specified by the limit when the limit is smaller than the batch size' do
+          expected_output = "Sample.create!([\n  {string: \"string\", text: \"text\", integer: 42, float: 3.14, decimal: \"2.72\", datetime: \"1776-07-04 19:14:00\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false}\n])\n"
+
+          SeedDump.dump(Sample.limit(1)).should eq(expected_output)
+        end
+
+        it 'should dump the number of models specified by the limit when the limit is larger than the batch size but not a multiple of the batch size' do
+          Sample.delete_all
+          4.times { FactoryGirl.create(:sample) }
+
+          SeedDump.dump(Sample.limit(3), batch_size: 2).should eq(@expected_output)
+        end
+      end
+    end
+
+    context 'with a batch_size parameter' do
+      it 'should not raise an exception' do
+
+        SeedDump.dump(Sample, batch_size: 100)
+      end
+    end
+
+    context 'Array' do
+      it 'should return the dump of the models passed in' do
+        SeedDump.dump(Sample.all.to_a, batch_size: 2).should eq(@expected_output)
+      end
+
+      it 'should return nil if the array is empty' do
+        SeedDump.dump([]).should be(nil)
+      end
+    end
+
+    context 'with an exclude parameter' do
+      it 'should exclude the specified attributes from the dump' do
+        expected_output = "Sample.create!([\n  {text: \"text\", integer: 42, decimal: \"2.72\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {text: \"text\", integer: 42, decimal: \"2.72\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false},\n  {text: \"text\", integer: 42, decimal: \"2.72\", time: \"2000-01-01 03:15:00\", date: \"1863-11-19\", binary: \"binary\", boolean: false}\n])\n"
+
+        SeedDump.dump(Sample, exclude: [:id, :created_at, :updated_at, :string, :float, :datetime]).should eq(expected_output)
+      end
+    end
+  end
+end

data/spec/environment_spec.rb

@@ -0,0 +1,136 @@
+require 'spec_helper'
+
+describe SeedDump do
+  describe '.dump_using_environment' do
+    before(:all) do
+      create_db
+    end
+
+    before(:each) do
+      Rails.application.eager_load!
+
+      FactoryGirl.create(:sample)
+    end
+
+    describe 'APPEND' do
+      it "should specify append as true if the APPEND env var is 'true'" do
+        SeedDump.should_receive(:dump).with(anything, include(append: true))
+
+        SeedDump.dump_using_environment('APPEND' => 'true')
+      end
+
+      it "should specify append as false the first time if the APPEND env var is not 'true' (and true after that)" do
+        FactoryGirl.create(:another_sample)
+
+        SeedDump.should_receive(:dump).with(anything, include(append: false)).ordered
+        SeedDump.should_receive(:dump).with(anything, include(append: true)).ordered
+
+        SeedDump.dump_using_environment('APPEND' => 'false')
+      end
+    end
+
+    describe 'BATCH_SIZE' do
+      it 'should pass along the specified batch size' do
+        SeedDump.should_receive(:dump).with(anything, include(batch_size: 17))
+
+        SeedDump.dump_using_environment('BATCH_SIZE' => '17')
+      end
+
+      it 'should pass along a nil batch size if BATCH_SIZE is not specified' do
+        SeedDump.should_receive(:dump).with(anything, include(batch_size: nil))
+
+        SeedDump.dump_using_environment
+      end
+    end
+
+    describe 'EXCLUDE' do
+      it 'should pass along any attributes to be excluded' do
+        SeedDump.should_receive(:dump).with(anything, include(exclude: [:baggins, :saggins]))
+
+        SeedDump.dump_using_environment('EXCLUDE' => 'baggins,saggins')
+      end
+    end
+
+    describe 'FILE' do
+      it 'should pass the FILE parameter to the dump method correctly' do
+        SeedDump.should_receive(:dump).with(anything, include(file: 'blargle'))
+
+        SeedDump.dump_using_environment('FILE' => 'blargle')
+      end
+
+      it 'should pass db/seeds.rb as the file parameter if no FILE is specified' do
+        SeedDump.should_receive(:dump).with(anything, include(file: 'db/seeds.rb'))
+
+        SeedDump.dump_using_environment
+      end
+    end
+
+    describe 'LIMIT' do
+      it 'should apply the specified limit to the records' do
+        relation_double = double('ActiveRecord relation double')
+        Sample.should_receive(:limit).with(5).and_return(relation_double)
+
+        SeedDump.should_receive(:dump).with(relation_double, anything)
+        SeedDump.stub(:dump)
+
+        SeedDump.dump_using_environment('LIMIT' => '5')
+      end
+    end
+
+    describe 'MODEL' do
+      it 'if MODEL is not specified it should dump all non-empty models' do
+        FactoryGirl.create(:another_sample)
+
+        [Sample, AnotherSample].each do |model|
+          SeedDump.should_receive(:dump).with(model, anything)
+        end
+
+        SeedDump.dump_using_environment
+      end
+
+      it 'if MODEL is specified it should only dump the specified model' do
+        FactoryGirl.create(:another_sample)
+
+        SeedDump.should_receive(:dump).with(Sample, anything)
+
+        SeedDump.dump_using_environment('MODEL' => 'Sample')
+      end
+    end
+
+    describe 'MODELS' do
+      it 'if MODELS is not specified it should dump all non-empty models' do
+        FactoryGirl.create(:another_sample)
+
+        [Sample, AnotherSample].each do |model|
+          SeedDump.should_receive(:dump).with(model, anything)
+        end
+
+        SeedDump.dump_using_environment
+      end
+
+      it 'if MODELS is specified it should only dump those models' do
+        FactoryGirl.create(:another_sample)
+        FactoryGirl.create(:yet_another_sample)
+
+        SeedDump.should_receive(:dump).with(Sample, anything)
+        SeedDump.should_receive(:dump).with(AnotherSample, anything)
+
+        SeedDump.dump_using_environment('MODELS' => 'Sample, AnotherSample')
+      end
+    end
+
+    it 'should run ok without ActiveRecord::SchemaMigration being set (needed for Rails Engines)' do
+      schema_migration = ActiveRecord::SchemaMigration
+
+      ActiveRecord.send(:remove_const, :SchemaMigration)
+
+      SeedDump.stub(:dump)
+
+      begin
+        SeedDump.dump_using_environment
+      ensure
+        ActiveRecord.const_set(:SchemaMigration, schema_migration)
+      end
+    end
+   end
+ end

data/spec/factories/another_samples.rb

@@ -0,0 +1,14 @@
+FactoryGirl.define do
+  factory :another_sample do
+    string   'string'
+    text     'text'
+    integer  42
+    float    3.14
+    decimal  2.72
+    datetime DateTime.parse('July 4, 1776 7:14pm UTC')
+    time     Time.parse('3:15am UTC')
+    date     Date.parse('November 19, 1863')
+    binary   'binary'
+    boolean  false
+  end
+end

data/spec/factories/samples.rb

@@ -0,0 +1,14 @@
+FactoryGirl.define do
+  factory :sample do
+    string   'string'
+    text     'text'
+    integer  42
+    float    3.14
+    decimal  2.72
+    datetime DateTime.parse('July 4, 1776 7:14pm UTC')
+    time     Time.parse('3:15am UTC')
+    date     Date.parse('November 19, 1863')
+    binary   'binary'
+    boolean  false
+  end
+end

data/spec/factories/yet_another_samples.rb

@@ -0,0 +1,14 @@
+FactoryGirl.define do
+  factory :yet_another_sample do
+    string   'string'
+    text     'text'
+    integer  42
+    float    3.14
+    decimal  2.72
+    datetime DateTime.parse('July 4, 1776 7:14pm UTC')
+    time     Time.parse('3:15am UTC')
+    date     Date.parse('November 19, 1863')
+    binary   'binary'
+    boolean  false
+  end
+end

data/spec/helpers.rb

@@ -11,16 +11,12 @@ class Rails
     if !@already_called
       Object.const_set('Sample', Class.new(ActiveRecord::Base))
 
-      Object.const_set('AbstractSample', Class.new(ActiveRecord::Base))
-      AbstractSample.abstract_class = true
+      Object.const_set('AnotherSample', Class.new(ActiveRecord::Base))
 
-      Object.const_set('ChildSample', Class.new(AbstractSample))
+      Object.const_set('YetAnotherSample', Class.new(ActiveRecord::Base))
 
       Object.const_set('NoTableModel', Class.new(ActiveRecord::Base))
 
-      Object.const_set('Nested', Module.new)
-      Nested.const_set('Sample', Class.new(ActiveRecord::Base))
-
       Object.const_set('EmptyModel', Class.new(ActiveRecord::Base))
 
       @already_called = true
@@ -33,20 +29,43 @@ module Helpers
     ActiveRecord::Migration.verbose = false
 
     ActiveRecord::Schema.define(:version => 1) do
-      create_table 'child_samples', :force => true do |t|
-        t.string   'name'
+      create_table 'samples', :force => true do |t|
+        t.string   'string'
+        t.text     'text'
+        t.integer  'integer'
+        t.float    'float'
+        t.decimal  'decimal'
+        t.datetime 'datetime'
+        t.time     'time'
+        t.date     'date'
+        t.binary   'binary'
+        t.boolean  'boolean'
         t.datetime 'created_at', :null => false
         t.datetime 'updated_at', :null => false
       end
 
-      create_table 'samples', :force => true do |t|
+      create_table 'another_samples', :force => true do |t|
+        t.string   'string'
+        t.text     'text'
+        t.integer  'integer'
+        t.float    'float'
+        t.decimal  'decimal'
+        t.datetime 'datetime'
+        t.time     'time'
+        t.date     'date'
+        t.binary   'binary'
+        t.boolean  'boolean'
+        t.datetime 'created_at', :null => false
+        t.datetime 'updated_at', :null => false
+      end
+
+      create_table 'yet_another_samples', :force => true do |t|
         t.string   'string'
         t.text     'text'
         t.integer  'integer'
         t.float    'float'
         t.decimal  'decimal'
         t.datetime 'datetime'
-        t.datetime 'timestamp'
         t.time     'time'
         t.date     'date'
         t.binary   'binary'

data/spec/spec_helper.rb

@@ -1,13 +1,19 @@
 require 'seed_dump'
-require 'active_support/core_ext/string'
-require 'active_support/descendants_tracker'
+
+require 'active_support'
 require 'active_record'
+
 require 'byebug'
+
 require 'database_cleaner'
+require 'factory_girl'
+
 require './spec/helpers'
 
 ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
 
+FactoryGirl.find_definitions
+
 RSpec.configure do |config|
   config.treat_symbols_as_metadata_keys_with_true_values = true
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 3.0.0
 platform: ruby
 authors:
 - Rob Halff
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-16 00:00:00.000000000 Z
+date: 2013-12-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -53,6 +53,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: factory_girl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: jeweler
   requirement: !ruby/object:Gem::Requirement
@@ -76,24 +90,28 @@ extra_rdoc_files:
 - README.md
 files:
 - .rspec
-- CHANGELOG.rdoc
 - Gemfile
 - MIT-LICENSE
 - README.md
 - Rakefile
 - VERSION
-- lib/clip.rb
 - lib/seed_dump.rb
 - lib/seed_dump/dump_methods.rb
+- lib/seed_dump/dump_methods/enumeration.rb
+- lib/seed_dump/environment.rb
 - lib/seed_dump/railtie.rb
 - lib/tasks/seed_dump.rake
-- lib/true.rb
 - seed_dump.gemspec
+- spec/dump_methods_spec.rb
+- spec/environment_spec.rb
+- spec/factories/another_samples.rb
+- spec/factories/samples.rb
+- spec/factories/yet_another_samples.rb
 - spec/helpers.rb
-- spec/seed_dump_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/rroblak/seed_dump
-licenses: []
+licenses:
+- MIT
 metadata: {}
 post_install_message: 
 rdoc_options: []

data/CHANGELOG.rdoc

@@ -1,56 +0,0 @@
-== 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
- * add TIMESTAMP flag and checks against attr_accessible [limratana].
- * modifications for better testing
- * make MODEL_DIR actually work
-
-== 0.4.0 / 2012-07-28
-
- * fix rdoc issue.
- * automatically use :without_protection if created_at, updated_at are present.
- * Properly dump decimal columns
- * support nested models [h6y3]
- * add WITHOUT_PROTECTION flag [Alexey Yurchenko]
- * add utf-8 encoding comment [Alexey Yurchenko, Alexey Poimtsev, n0b0dy]
- * add SKIP_CALLBACKS flag [Alexey Yurchenko]
- * fixed typo in README [Andrey Ognevsky]
- * add PG_SCHEMA flag [Luka Novsak]
-
-== 0.3.4 / 2011-07-14
-
- * Skip models that aren't Derived from ActiveRecord [iconoclast]
- * override the AR version of attribute_for_inspect to NOT truncate strings [sc0ttman]
-
-== 0.3.3 / 2011-07-14
-
- * solve sorting issue
-
-== 0.3.2 / 2011-04-22
-
- * fix broken APPEND option
-
-== 0.3.1 / 2011-04-22
-
- * use different syntax for multiple rows
-
-== 0.3.0 / 2011-04-22
-
- * refactor to make testing possible
-
-== 0.2.4 / 2011-04-22
-
- * properly quote any value (also fixes Time value generation)
- * solve issue when specifying WITH_ID [David Guthu]
-

data/lib/clip.rb

@@ -1,6 +0,0 @@
-# by digitalross: http://stackoverflow.com/questions/1604305/all-but-last-element-of-ruby-array
-class Array
-  def clip n=1
-    take size - n
-  end
-end

data/lib/true.rb

@@ -1,9 +0,0 @@
-# http://veerasundaravel.wordpress.com/2010/10/26/string-to-boolean-conversion-in-ruby/
-class Object 
-  def true? # to boolean
-    return false if self.nil?
-    return true if self == true || self =~ (/(true|t|yes|y|1)$/i)
-    return false if self == false || self.nil? || self =~ (/(false|f|no|n|0)$/i)
-    raise ArgumentError.new('invalid value for Boolean: "#{self}"')
-  end
-end

data/spec/seed_dump_spec.rb

@@ -1,157 +0,0 @@
-require 'spec_helper'
-
-describe SeedDump do
-  describe '#dump_models' do
-    before(:all) do
-      create_db
-    end
-
-    before(:each) do
-      @sd = SeedDump.new
-
-      @env = {'FILE' => Dir.pwd + '/spec/db/seeds.rb',
-              'VERBOSE' => false,
-              'DEBUG' => false}
-
-      ActiveSupport::DescendantsTracker.clear
-    end
-
-    it 'should not include timestamps if the TIMESTAMPS parameter is false' do
-      Rails.application.eager_load!
-
-      @env['MODELS'] = 'Sample'
-      @env['TIMESTAMPS'] = false
-
-      @sd.setup @env
-
-      load_sample_data
-
-      @sd.dump_models.should match(/^\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil }\n\]\)\n\n\n$/)
-    end
-
-    it 'should include timestamps if the TIMESTAMPS parameter is true' do
-      Rails.application.eager_load!
-
-      @env['MODELS'] = 'Sample'
-      @env['TIMESTAMPS'] = true
-
-      load_sample_data
-
-      @sd.setup @env
-
-      @sd.dump_models.should match(/^\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n$/)
-    end
-
-    it 'should include ids if the WITH_ID parameter is true' do
-      Rails.application.eager_load!
-
-      @env['MODELS'] = 'Sample'
-      @env['WITH_ID'] = true
-
-      @sd.setup @env
-
-      load_sample_data
-
-      @sd.dump_models.should match(/^\nSample\.create!\(\[\n  { :id => \d+, :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n$/)
-    end
-
-    it 'should respect the MODELS parameter' do
-      Rails.application.eager_load!
-
-      @env['MODELS'] = 'Sample'
-
-      @sd.setup @env
-
-      load_sample_data
-
-      @sd.dump_models.should match(/\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it 'should use the create method specified in the CREATE_METHOD parameter' do
-      load_sample_data
-
-      @env['MODELS'] = 'Sample'
-      @env['CREATE_METHOD'] = 'create'
-
-      @sd.setup @env
-
-      @sd.dump_models.should match(/\nSample\.create\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it "should use 'create!' as the default create method" do
-      load_sample_data
-
-      @env['MODELS'] = 'Sample'
-
-      @sd.setup @env
-
-      @sd.dump_models.should match(/\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it "should return the contents of the dump" do
-      load_sample_data
-
-      @env['MODELS'] = 'Sample'
-
-      @sd.setup @env
-
-      @sd.dump_models.should match(/\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it 'should run ok without ActiveRecord::SchemaMigration being set (needed for Rails Engines)' do
-      schema_migration = ActiveRecord::SchemaMigration
-
-      ActiveRecord.send(:remove_const, :SchemaMigration)
-
-      begin
-        @sd.setup @env
-
-        @sd.dump_models
-      ensure
-        ActiveRecord.const_set(:SchemaMigration, schema_migration)
-      end
-    end
-
-    it "should skip any models whose tables don't exist" do
-      @sd.setup @env
-
-      load_sample_data
-
-      @sd.dump_models.should match(/\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it "should skip any models that don't have have any rows" do
-      @sd.setup @env
-
-      @sd.dump_models.should_not include('EmptyModel')
-    end
-
-    it 'should respect the LIMIT parameter' do
-      load_sample_data
-      load_sample_data
-
-      @env['MODELS'] = 'Sample'
-      @env['LIMIT'] = '1'
-
-      @sd.setup @env
-
-      @sd.dump_models.should match(/\nSample\.create!\(\[\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}", :updated_at => "\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}" }\n\]\)\n\n\n/)
-    end
-
-    it 'should only pull attributes that are returned as strings' do
-      load_sample_data
-
-      @env['MODELS'] = 'Sample'
-      @env['LIMIT'] = '1'
-
-      @sd.setup @env
-
-      original_attributes = Sample.new.attributes
-      attributes = original_attributes.merge(['col1', 'col2', 'col3'] => 'ABC')
-
-      Sample.any_instance.stub(:attributes).and_return(attributes)
-
-      @sd.dump_models.should eq("\nSample.create!([\n  { :string => nil, :text => nil, :integer => nil, :float => nil, :decimal => nil, :datetime => nil, :timestamp => nil, :time => nil, :date => nil, :binary => nil, :boolean => nil, :created_at => nil, :updated_at => nil }\n])\n\n\n")
-    end
-  end
-end