seed_dump 1.0.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6af3d1433e6c77d4aa40fa1ea392cc9ac21ee728
-  data.tar.gz: eb9a5434bc51a69c1f24828c91be082679bf6668
+  metadata.gz: 1a7a3bc7780fa30de0f2ffccb171249879058fa5
+  data.tar.gz: dad1c603abb084874344d01263e6dae6d88ea03d
 SHA512:
-  metadata.gz: a995c0615b35cc3a7332a8d2eef59dea404ece69fa9b8ec10ab46ff3c0338fed07de039c14673ec582fa4cb86877ad90c5f6f81c1ac5a76cb30e60e282546595
-  data.tar.gz: 59aeb2c7666dfba8471406b8428598025f2e3a522f15fdd6419a06179c4671bfce6ca89eb14ee121a12522b0e1923200f9ba728737136903d077b234ce4c0921
+  metadata.gz: 9e1f0bfa9e904d6b0b6756b45caa39afee9576f5661c155485d23f430ff87509f488f6b95e247e4ede7f3b41171229b9e49e989677acb3b658be5345e343f4d5
+  data.tar.gz: 74606c3f956c8b5118e83edea597f152e8e02b7913ee1af6ffcba3f9c63f0700d0f34042bfe50f2e536b72b29bfcb1a27d4c58682d9f16926b3699ff7afe1655

data/Gemfile

@@ -3,6 +3,10 @@ source 'https://rubygems.org'
 gem 'activesupport'
 gem 'activerecord'
 
+group :development, :test do
+  gem 'byebug'
+end
+
 group :development do
   gem 'jeweler', :git => 'git://github.com/technicalpickles/jeweler.git'
 end
@@ -10,4 +14,5 @@ end
 group :test do
   gem 'rspec'
   gem 'sqlite3'
+  gem 'database_cleaner'
 end

data/README.md

@@ -5,7 +5,7 @@ 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.
 
-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) or earlier.
+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
 -------------
@@ -24,12 +24,12 @@ Result:
     # Autogenerated by the db:seed:dump task
     # Do not hesitate to tweak this to your needs
 
-    products = Product.create([
+    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" }
     ])
 
-    users = User.create([
+    users = User.create!([
       { :id => 1, :password => "123456", :username => "test_1" },
       { :id => 2, :password => "234567", :username => "tes2" }
     ])
@@ -42,9 +42,9 @@ 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`:
+If you want the dump to use `create` rather than `create!`:
 
-    rake db:seed:dump CREATE_METHOD='create!'
+    rake db:seed:dump CREATE_METHOD='create'
 
 There are more environment variables that can be set— see below for all of them.
 
@@ -64,9 +64,9 @@ Or install it by hand:
 All environment variables
 -------------------------
 
-`APPEND`: Append the data to the file instead of overwriting it.
+`APPEND`: If set to `true`, append the data to the file instead of overwriting it.  Default: `false`.
 
-`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!'`).
+`CREATE_METHOD`: Use the specified create method rather than `create!` (the default).
 
 `FILE`: Use a different output file.  Default: `db/seeds.rb`.
 
@@ -76,14 +76,6 @@ All environment variables
 
 `MODEL[S]`: A model name or a comma-separated list of models.  Default: all models.
 
-`NO_DATA`: Don't dump any data from the db, instead generate empty `create` options. 
+`WITH_ID`: If set to `true`, include the `:id` in the `create` options.  Default: `false`.
 
-`WITH_ID`: Include the `:id` in the `create` options.
-
-`TIMESTAMPS`: If true, include the `:created_by` and `:updated_by` timestamps.  If false, don't include them.  Default: true.
-
-`SKIP_CALLBACKS`: Deactivate callbacks while importing.
-
-`PG_SCHEMA`: Postgres schema support.
-
-`MODEL_DIR`: Specify an alternate model dir.
+`TIMESTAMPS`: If set to `true`, include the `:created_by` and `:updated_by` timestamps.  Default: `true`.

data/VERSION

@@ -1 +1 @@
-1.0.0
+2.0.0

data/lib/seed_dump.rb

@@ -1,5 +1,7 @@
-# SeedDump
-module SeedDump 
+require 'seed_dump/dump_methods'
+
+class SeedDump
+  include SeedDump::DumpMethods
+
   require 'seed_dump/railtie' if defined?(Rails)
-  require 'seed_dump/perform' if defined?(Rails)
 end

data/lib/seed_dump/dump_methods.rb

@@ -0,0 +1,151 @@
+require "true"
+require "clip"
+
+class SeedDump
+  module DumpMethods
+
+    def initialize
+      @opts = {}
+      @ar_options = {}
+      @indent = ""
+      @models = []
+      @seed_rb = ""
+      @id_set_string = ""
+    end
+
+    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
+
+    def log(msg)
+      puts msg if @opts['debug']
+    end
+
+    def models
+      @models
+    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
+
+      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_model(model)
+      @id_set_string = ''
+      create_hash = ""
+      rows = []
+
+
+      model.find_each(batch_size: (@limit || 1000)) do |record|
+        attr_s = [];
+
+        record.attributes.select {|x| x.is_a?(String) }.each do |k,v|
+          dump_attribute(attr_s, record, k, v)
+        end
+
+        rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
+
+        break if rows.length == @limit
+      end
+
+      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('')
+      else
+        "\n#{model}.#{@opts['create_method']}([\n" << rows.join(",\n") << "\n])\n"
+      end
+
+    end
+
+    def dump_models
+      if @models.empty?
+        Rails.application.eager_load!
+
+        @seed_rb = ""
+
+        @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 }
+
+      @models.each do |model|
+        puts "Adding #{model} seeds." if @opts['verbose']
+
+        @seed_rb << dump_model(model) << "\n\n"
+      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]
+
+      if value.is_a?(String) && value.length > 50
+        "#{value}".inspect
+      elsif value.is_a?(Date) || value.is_a?(Time)
+        %("#{value.to_s(:db)}")
+      else
+        value.inspect
+      end
+    end
+
+    def output
+      @seed_rb
+    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/railtie.rb

@@ -1,9 +1,9 @@
-module SeedDump 
+class SeedDump
   class Railtie < Rails::Railtie
 
     rake_tasks do
       load "tasks/seed_dump.rake"
-    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::Perform.new.run(ENV)
+      SeedDump.new.run(ENV)
 
     end
   end

data/seed_dump.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "seed_dump"
-  s.version = "1.0.0"
+  s.version = "2.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-09-18"
+  s.date = "2013-10-16"
   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 = [
@@ -25,16 +25,13 @@ Gem::Specification.new do |s|
     "VERSION",
     "lib/clip.rb",
     "lib/seed_dump.rb",
-    "lib/seed_dump/perform.rb",
+    "lib/seed_dump/dump_methods.rb",
     "lib/seed_dump/railtie.rb",
     "lib/tasks/seed_dump.rake",
     "lib/true.rb",
     "seed_dump.gemspec",
-    "spec/models/abstract_sample.rb",
-    "spec/models/child_sample.rb",
-    "spec/models/nested/sample.rb",
-    "spec/models/sample.rb",
-    "spec/seed_dump_perform_spec.rb",
+    "spec/helpers.rb",
+    "spec/seed_dump_spec.rb",
     "spec/spec_helper.rb"
   ]
   s.homepage = "https://github.com/rroblak/seed_dump"
@@ -48,15 +45,18 @@ Gem::Specification.new do |s|
     if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
       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<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<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<jeweler>, [">= 0"])
   end
 end

data/spec/helpers.rb

@@ -0,0 +1,69 @@
+# Mock Rails.application.eager_load! and define some
+# Rails models for use in specs.
+class Rails
+  def self.application
+    self
+  end
+
+  def self.eager_load!
+    @already_called ||= false
+
+    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('ChildSample', Class.new(AbstractSample))
+
+      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
+    end
+  end
+end
+
+module Helpers
+  def create_db
+    ActiveRecord::Migration.verbose = false
+
+    ActiveRecord::Schema.define(:version => 1) do
+      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'
+        t.float    'float'
+        t.decimal  'decimal'
+        t.datetime 'datetime'
+        t.datetime 'timestamp'
+        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 'empty_models', force: true
+    end
+  end
+
+  def load_sample_data
+    Rails.application.eager_load!
+
+    Sample.create!
+
+    ChildSample.create!
+  end
+end

data/spec/seed_dump_spec.rb

@@ -0,0 +1,157 @@
+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

data/spec/spec_helper.rb

@@ -1,15 +1,30 @@
-require 'seed_dump/perform'
+require 'seed_dump'
 require 'active_support/core_ext/string'
+require 'active_support/descendants_tracker'
 require 'active_record'
+require 'byebug'
+require 'database_cleaner'
+require './spec/helpers'
+
+ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
 
 RSpec.configure do |config|
   config.treat_symbols_as_metadata_keys_with_true_values = true
-  config.run_all_when_everything_filtered = true
-  config.filter_run :focus
 
-  # Run specs in random order to surface order dependencies. If you find an
-  # order dependency and want to debug it, you can fix the order by providing
-  # the seed, which is printed after each run.
-  #     --seed 1234
   config.order = 'random'
+
+  config.include Helpers
+
+  config.before(:suite) do
+    DatabaseCleaner.strategy = :transaction
+    DatabaseCleaner.clean_with(:truncation)
+  end
+
+  config.before(:each) do
+    DatabaseCleaner.start
+  end
+
+  config.after(:each) do
+    DatabaseCleaner.clean
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Rob Halff
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-09-18 00:00:00.000000000 Z
+date: 2013-10-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -39,6 +39,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  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
@@ -70,16 +84,13 @@ files:
 - VERSION
 - lib/clip.rb
 - lib/seed_dump.rb
-- lib/seed_dump/perform.rb
+- lib/seed_dump/dump_methods.rb
 - lib/seed_dump/railtie.rb
 - lib/tasks/seed_dump.rake
 - lib/true.rb
 - seed_dump.gemspec
-- spec/models/abstract_sample.rb
-- spec/models/child_sample.rb
-- spec/models/nested/sample.rb
-- spec/models/sample.rb
-- spec/seed_dump_perform_spec.rb
+- spec/helpers.rb
+- spec/seed_dump_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/rroblak/seed_dump
 licenses: []

data/lib/seed_dump/perform.rb

@@ -1,201 +0,0 @@
-require "true"
-require "clip"
-
-module SeedDump
-  class Perform
-
-    def initialize
-      @opts = {}
-      @ar_options = {}
-      @indent = ""
-      @models = []
-      @last_record = []
-      @seed_rb = ""
-      @id_set_string = ""
-      @model_dir = 'app/models/**/*.rb'
-    end
-
-    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['no-data'] = env['NO_DATA'].true?
-      @opts['skip_callbacks'] = env['SKIP_CALLBACKS'].true?
-      @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 }
-      @opts['schema']  = env['PG_SCHEMA']
-      @opts['model_dir']  = env['MODEL_DIR'] || @model_dir
-      @opts['create_method']  = env['CREATE_METHOD'] || 'create'
-    end
-
-    def log(msg)
-      puts msg if @opts['debug']
-    end
-
-    def load_models
-      log("Searching in #{@opts['model_dir']} for models")
-
-      Dir[File.join(Dir.pwd, @opts['model_dir'])].sort.each do |f|
-        log("Processing file #{f}")
-
-        dirname, basename = File.split(f)
-
-        dir_array = dirname.split(File::SEPARATOR)
-
-        # Find index of last occurence of 'models' in path
-        models_index = nil
-        dir_array.each_with_index {|x, i| models_index = i if x == 'models'}
-
-        model_dir_array = dir_array[models_index + 1..-1]
-
-        # Initialize nested model namespaces
-        model_dir_array.inject(Object) do |parent, child|
-          child = child.camelize
-
-          if parent.const_defined?(child)
-            parent.const_get(child)
-          else
-            parent.const_set(child, Module.new)
-          end
-        end
-
-        require f
-
-        model = File.join(model_dir_array + [File.basename(basename, '.rb')]).camelize
-
-        log("Detected model #{model}")
-
-        @models << model if @opts['models'].include?(model) || @opts['models'].empty?
-      end
-    end
-
-    def models
-      @models
-    end
-
-    def last_record
-      @last_record
-    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
-
-      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_model(model)
-      @id_set_string = ''
-      @last_record = []
-      create_hash = ""
-      options = ''
-      rows = []
-      arr = nil
-      unless @opts['no-data']
-        arr = model.all
-        arr.limit(@ar_options[:limit]) if @ar_options[:limit]
-      end
-      arr = arr.empty? ? [model.new] : arr
-
-      arr.each_with_index { |r,i|
-        attr_s = [];
-        r.attributes.each do |k,v|
-          pushed_key = dump_attribute(attr_s,r,k,v)
-          @last_record.push k if pushed_key
-        end
-        rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
-      }
-
-      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]#{options})\n"
-        end
-        maxsarr.join('')
-      else
-        "\n#{model}.#{@opts['create_method']}([\n" << rows.join(",\n") << "\n]#{options})\n"
-      end
-
-    end
-
-    def dump_models
-      @seed_rb = ""
-      @models.sort.each do |model|
-          m = model.constantize
-          if m.ancestors.include?(ActiveRecord::Base) && !m.abstract_class
-            puts "Adding #{model} seeds." if @opts['verbose']
-
-            if @opts['skip_callbacks']
-              @seed_rb << "#{model}.reset_callbacks :save\n"
-              @seed_rb << "#{model}.reset_callbacks :create\n"
-              puts "Callbacks are disabled." if @opts['verbose']
-            end
-
-            @seed_rb << dump_model(m) << "\n\n"
-          else
-            puts "Skipping non-ActiveRecord model #{model}..." if @opts['verbose']
-          end
-      end
-    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]
-
-      if value.is_a?(String) && value.length > 50
-        "#{value}".inspect
-      elsif value.is_a?(Date) || value.is_a?(Time)
-        %("#{value.to_s(:db)}")
-      else
-        value.inspect
-      end
-    end
-
-    def set_search_path(path, append_public=true)
-        path_parts = [path.to_s, ('public' if append_public)].compact
-        ActiveRecord::Base.connection.schema_search_path = path_parts.join(',')
-    end
-
-    def run(env)
-      setup env
-
-      set_search_path @opts['schema'] if @opts['schema']
-
-      load_models
-
-      puts "Appending seeds to #{@opts['file']}." if @opts['append']
-      dump_models
-
-      puts "Writing #{@opts['file']}."
-      write_file
-
-      puts "Done."
-    end
-  end
-end

data/spec/models/abstract_sample.rb

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

data/spec/models/child_sample.rb

@@ -1,3 +0,0 @@
-class ChildSample < AbstractSample
-end
-

data/spec/models/nested/sample.rb

@@ -1,2 +0,0 @@
-class Nested::Sample < ActiveRecord::Base
-end

data/spec/models/sample.rb

@@ -1,2 +0,0 @@
-class Sample < ActiveRecord::Base
-end

data/spec/seed_dump_perform_spec.rb

@@ -1,118 +0,0 @@
-require 'spec_helper'
-
-describe SeedDump::Perform do
-  before(:all) do
-    ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
-
-    ActiveRecord::Schema.define(:version => 1) do
-      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'
-        t.float    'float'
-        t.decimal  'decimal'
-        t.datetime 'datetime'
-        t.datetime 'timestamp'
-        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
-    end
-  end
-
-  before do
-    @sd = SeedDump::Perform.new
-
-    @env = {'MODEL_DIR' => 'spec/models/*.rb',
-            'FILE' => Dir.pwd + '/spec/db/seeds.rb',
-            'VERBOSE' => false,
-            'DEBUG' => false}
-  end
-
-  it 'should load models from the specified directory' do
-    @sd.setup(@env)
-
-    @sd.load_models
-
-    @sd.models.should eq(["AbstractSample", "ChildSample", "Sample"])
-  end
-
-  it 'should support nested models' do
-    @env['MODEL_DIR'] = 'spec/models/**/*.rb'
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.models.should eq(['AbstractSample', 'ChildSample', 'Nested::Sample', 'Sample'])
-  end
-
-  it 'should not include timestamps if the TIMESTAMPS parameter is false' do
-    @env['TIMESTAMPS'] = false
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.dump_models
-
-    @sd.last_record.should_not include('created_at')
-  end
-
-  it 'should include timestamps if the TIMESTAMPS parameter is true' do
-    @env['TIMESTAMPS'] = true
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.dump_models
-
-    @sd.last_record.should include('created_at')
-  end
-
-  it 'should include ids if the WITH_ID parameter is true' do
-    @env['WITH_ID'] = true
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.dump_models
-
-    @sd.last_record.should include('id')
-  end
-
-  it 'should skip abstract models' do
-    @env['MODELS'] = 'AbstractSample'
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.dump_models
-
-    @sd.last_record.should eq([])
-  end
-
-  it 'should use the create method specified in the CREATE_METHOD parameter' do
-    @env['CREATE_METHOD'] = 'create!'
-
-    @sd.setup @env
-
-    @sd.load_models
-
-    @sd.dump_models
-
-    @sd.instance_variable_get(:@seed_rb).should include('create!')
-  end
-end