seed_dump 0.2.4 → 0.3.0

data/CHANGELOG.rdoc

@@ -1,5 +1,9 @@
 == Seed Dump
 
+== 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)

data/VERSION

@@ -1 +1 @@
-0.2.4
+0.3.0

data/lib/seed_dump.rb

@@ -1,4 +1,5 @@
 # SeedDump
 module SeedDump 
   require 'seed_dump/railtie' if defined?(Rails)
+  require 'seed_dump/perform' if defined?(Rails)
 end

data/lib/seed_dump/perform.rb

@@ -0,0 +1,84 @@
+module SeedDump
+  class Perform
+
+    @opts = {}
+    @ar_options = {} 
+    @indent = ""
+    @models = []
+    @seed_rb = "" 
+    @id_set_string = ""
+    @verbose = true
+
+    def self.setup(env)
+      # config
+      @opts['with_id'] = !env["WITH_ID"].nil?
+      @opts['no-data'] = !env['NO_DATA'].nil?
+      @opts['models']  = env['MODELS'] || (env['MODEL'] ? env['MODEL'] : "")
+      @opts['file']    = env['FILE'] || "#{Rails.root}/db/seeds.rb"
+      @opts['append']  = (!env['APPEND'].nil? && File.exists?(opts['file']) )
+      @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.constantize }
+    end
+
+    def self.loadModels
+      Dir['app/models/*.rb'].sort.each do |f|
+        model = File.basename(f, '.*').camelize.constantize
+        @models.push model if @opts['models'].include?(model) || @opts['models'].empty? 
+      end
+    end
+
+    def self.dumpAttribute(a_s,r,k,v)
+      v = r.attribute_for_inspect(k)
+      if k == 'id' && @opts['with_id']
+        @id_set_string = "{ |c| c.#{k} = #{v} }.save"
+      else
+        a_s.push("#{k.to_sym.inspect} => #{v}") unless k == 'id' && !@opts['with_id']
+      end 
+    end
+
+    def self.dumpModel(model)
+      create_hash = ""
+      arr = []
+      arr = model.find(:all, @ar_options) unless @opts['no-data']
+      arr = arr.empty? ? [model.new] : arr
+      arr.each_with_index { |r,i| 
+        attr_s = [];
+        r.attributes.each { |k,v| dumpAttribute(attr_s,r,k,v) }
+        create_hash << "\n#{model}.create" << '( ' << attr_s.join(', ') << ' )' << @id_set_string
+      } 
+      create_hash
+    end
+
+    def self.dumpModels
+      @seed_rb = ""
+      @models.sort.each do |model|
+          @id_set_string = ''
+          puts "Adding #{model} seeds." if @verbose
+          @seed_rb << dumpModel(model) << "\n\n"
+      end
+    end
+
+    def self.writeFile
+      File.open(@opts['file'], (@opts['append'] ? "a" : "w")) { |f|
+        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
+
+    def self.run(env)
+
+      setup env
+
+      loadModels
+
+      puts "Appending seeds to #{@opts['file']}." if @opts['append']
+      dumpModels
+
+      puts "Writing #{@opts['file']}."
+      writeFile
+
+      puts "Done."
+    end
+  end
+end

data/lib/tasks/seed_dump.rake

@@ -3,73 +3,8 @@ namespace :db do
     desc "Dump records from the database into db/seeds.rb"
     task :dump => :environment do
 
-      # config
-      opts = {}
-      opts['with_id'] = !ENV["WITH_ID"].nil?
-      opts['no-data'] = !ENV['NO_DATA'].nil?
-      opts['models']  = ENV['MODELS'] || (ENV['MODEL'] ? ENV['MODEL'] : "")
-      opts['file']    = ENV['FILE'] || "#{Rails.root}/db/seeds.rb"
-      opts['append']  = (!ENV['APPEND'].nil? && File.exists?(opts['file']) )
-      ar_options      = ENV['LIMIT'].to_i > 0 ? { :limit => ENV['LIMIT'].to_i } : {}
-      indent          = " " * (ENV['INDENT'].nil? ? 2 : ENV['INDENT'].to_i)
+      SeedDump::Perform::run(ENV)
 
-      models = opts['models'].split(',').collect {|x| x.underscore.singularize }
-
-      new_line = "\n"
-      puts "Appending seeds to #{opts['file']}." if opts['append']
-
-      seed_rb = ""
-      Dir['app/models/*.rb'].sort.each do |f|
-	      model_name  = File.basename(f, '.*')
-        if models.include?(model_name) || models.empty? 
-
-          puts "Adding #{model_name.camelize} seeds."
-
-      	  create_hash = ""
-
-      	  model = model_name.camelize.constantize
-      	  arr = []
-      	  arr = model.find(:all, ar_options) unless opts['no-data']
-      	  arr = arr.empty? ? [model.new] : arr
-      	  arr.each_with_index { |r,i| 
-
-            attr_s = [];
-
-            id_set_string = ''
-            r.attributes.each { |k,v|
-	      v = r.attribute_for_inspect(k)
-              if k == 'id' && opts['with_id']
-                id_set_string = "{ |c| c.#{k} = #{v} }.save"
-              else
-                attr_s.push("#{k.to_sym.inspect} => #{v}") unless k == 'id' && !opts['with_id']
-              end 
-            }
-            create_hash << (i > 0 ? "#{new_line}" : new_line) << "#{model_name.camelize}.create" << '( ' << attr_s.join(', ') << ' )' << id_set_string
-          } 
-          seed_rb << "#{create_hash}#{new_line}#{new_line}"
-        end
-      end
-
-      File.open(opts['file'], (opts['append'] ? "a" : "w")) { |f|
-
-      puts "Writing #{opts['file']}."
-
-      unless opts['append']
-          cont =<<HERE
-# Autogenerated by the db:seed:dump task
-# Do not hesitate to tweak this to your needs
-HERE
-          f << cont
-        end
-
-	cont =<<HERE
-#{seed_rb}
-HERE
-      f << cont
-
-      puts "Done."
-
-}
     end
   end
 end

data/seed_dump.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{seed_dump}
-  s.version = "0.2.4"
+  s.version = "0.3.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff"]
@@ -22,6 +22,7 @@ Gem::Specification.new do |s|
     "Rakefile",
     "VERSION",
     "lib/seed_dump.rb",
+    "lib/seed_dump/perform.rb",
     "lib/seed_dump/railtie.rb",
     "lib/tasks/seed_dump.rake",
     "seed_dump.gemspec",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.3.0
   prerelease: 
 platform: ruby
 authors:
@@ -26,6 +26,7 @@ files:
 - Rakefile
 - VERSION
 - lib/seed_dump.rb
+- lib/seed_dump/perform.rb
 - lib/seed_dump/railtie.rb
 - lib/tasks/seed_dump.rake
 - seed_dump.gemspec