seed_dump 0.4.0 → 0.4.1

data/CHANGELOG.rdoc

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

data/README.rdoc

@@ -78,6 +78,9 @@ NO_DATA:
 WITH_ID:
  Inlcude the +:id+ in the create options
 
+TIMESTAMPS:
+ Include the :created_by and :updated_by timestamps (default)
+
 SKIP_CALLBACKS:
  Deactivate callbacks while importing.
 
@@ -85,7 +88,8 @@ PG_SCHEMA:
  Postgres schema support
 
 WITHOUT_PROTECTION:
- Skip protection for columns that are protected by default. (Automatically activated if such columns are present)
+ Skip protection for columns that are protected by default.
+ Note : WITH_ID and TIMESTAMPS automatically override protection
 
 MODEL_DIR:
  Specify an alternative model dir

data/VERSION

@@ -1 +1 @@
-0.4.0
+0.4.1

data/lib/clip.rb

@@ -0,0 +1,6 @@
+# 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/seed_dump/perform.rb

@@ -1,4 +1,5 @@
 require "true"
+require "clip"
 
 module SeedDump
   class Perform
@@ -8,18 +9,20 @@ module SeedDump
       @ar_options = {} 
       @indent = ""
       @models = []
+      @last_record = []
       @seed_rb = "" 
       @id_set_string = ""
-      @verbose = true
       @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['without_protection'] = env['WITHOUT_PROTECTION'].true?
+      @opts['without_protection'] = env['WITHOUT_PROTECTION'].true? || (env['WITHOUT_PROTECTION'].nil? && @opts['timestamps'])
       @opts['skip_callbacks'] = env['SKIP_CALLBACKS'].true?
       @opts['models']  = env['MODELS'] || (env['MODEL'] ? env['MODEL'] : "")
       @opts['file']    = env['FILE'] || "#{Rails.root}/db/seeds.rb"
@@ -33,12 +36,17 @@ module SeedDump
 
     def loadModels
       puts "Searching in #{@opts['model_dir']} for models" if @opts['debug']
-      Dir[@opts['model_dir']].sort.each do |f|
+      Dir[Dir.pwd + '/' + @opts['model_dir']].sort.each do |f|
         puts "Processing file #{f}" if @opts['debug']
         # parse file name and path leading up to file name and assume the path is a module
         f =~ /models\/(.*).rb/
         # split path by /, camelize the constituents, and then reform as a formal class name
-        model = $1.split("/").map {|x| x.camelize}.join("::")
+        parts = $1.split("/").map {|x| x.camelize}
+        # Initialize nested model namespaces
+        parts.clip.inject(Object) { |x, y| x.const_set(y, Module.new) }
+        model = parts.join("::")
+        require f
+
         puts "Detected model #{model}" if @opts['debug']
         @models.push model if @opts['models'].include?(model) || @opts['models'].empty? 
       end
@@ -48,6 +56,10 @@ module SeedDump
       @models
     end
 
+    def last_record
+      @last_record
+    end
+
     def dumpAttribute(a_s,r,k,v)
       if v.is_a?(BigDecimal)
         v = v.to_s
@@ -56,13 +68,15 @@ module SeedDump
       end 
 
       unless k == 'id' && !@opts['with_id']
-        @opts['without_protection'] = true if ["id", "created_at", "updated_at"].include?(k)
-        a_s.push("#{k.to_sym.inspect} => #{v}")
+        if (!(k == 'created_at' || k == 'updated_at') || @opts['timestamps'])
+          a_s.push("#{k.to_sym.inspect} => #{v}")
+        end
       end 
     end
 
     def dumpModel(model)
       @id_set_string = ''
+      @last_record = []
       create_hash = ""
       options = ''
       rows = []
@@ -72,7 +86,12 @@ module SeedDump
 
       arr.each_with_index { |r,i| 
         attr_s = [];
-        r.attributes.each { |k,v| dumpAttribute(attr_s,r,k,v) }
+        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 
+          end
+        end
         rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
       } 
 
@@ -88,19 +107,18 @@ module SeedDump
       @models.sort.each do |model|
           m = model.constantize
           if m.ancestors.include?(ActiveRecord::Base)
-            puts "Adding #{model} seeds." if @verbose
+            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 @verbose
+              puts "Callbacks are disabled." if @opts['verbose']
             end
 
             @seed_rb << dumpModel(m) << "\n\n"
           else
-            puts "Skipping non-ActiveRecord model #{model}..." if @verbose
+            puts "Skipping non-ActiveRecord model #{model}..." if @opts['verbose']
           end
-          puts "Protection is disabled." if @verbose && @opts['without_protection']
       end
     end
 
@@ -134,6 +152,8 @@ module SeedDump
 
       setup env
 
+      puts "Protection is disabled." if @opts['verbose'] && @opts['without_protection']
+
       setSearchPath @opts['schema'] if @opts['schema']
 
       loadModels

data/seed_dump.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "seed_dump"
-  s.version = "0.4.0"
+  s.version = "0.4.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff"]
-  s.date = "2012-07-28"
+  s.date = "2012-08-01"
   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 = [
@@ -21,6 +21,7 @@ Gem::Specification.new do |s|
     "README.rdoc",
     "Rakefile",
     "VERSION",
+    "lib/clip.rb",
     "lib/seed_dump.rb",
     "lib/seed_dump/perform.rb",
     "lib/seed_dump/railtie.rb",

data/test/seed_dump_test.rb

@@ -7,24 +7,53 @@ class SeedDumpTest < ActiveSupport::TestCase
     @sd = SeedDump::Perform.new
     # universial options for every test
     @env = {
-     "MODEL_DIR" => Dir.pwd + '/test/models/**.rb',
+     "MODEL_DIR" => 'test/models/**.rb',
      "FILE" => Dir.pwd + '/test/db/seeds.rb',
-     "DEBUG" => true
+     "VERBOSE" => false,
+     "DEBUG" => false 
     }
   end
 
   test "load sample model" do
-    @env['MODEL_DIR'] = Dir.pwd + '/test/models/*.rb'
+    @env['MODEL_DIR'] = 'test/models/*.rb'
     @sd.setup @env 
     @sd.loadModels
     assert_equal ["Sample"], @sd.models
   end
 
   test "support nested models" do
-    @env['MODEL_DIR'] = Dir.pwd + '/test/models/**/*.rb'
+    @env['MODEL_DIR'] = 'test/models/**/*.rb'
     @sd.setup @env 
     @sd.loadModels
     assert_equal ["Nested::Sample", "Sample"], @sd.models
   end
 
+  test "without timestamps" do
+    @env['MODEL_DIR'] = 'test/models/*.rb'
+    @env['TIMESTAMPS'] = false
+    @sd.setup @env
+    @sd.loadModels
+    @sd.dumpModels
+    assert !@sd.last_record.include?("created_at"), "Should not include created_at if timestamps are off"
+  end
+
+  test "with timestamps" do
+    @env['MODEL_DIR'] = 'test/models/*.rb'
+    @env['TIMESTAMPS'] = true
+    @sd.setup @env
+    @sd.loadModels
+    @sd.dumpModels
+    assert @sd.last_record.include?("created_at"), "Must include created_at if timestamps are desired"
+  end
+
+  test "with id" do
+    @env['MODEL_DIR'] = 'test/models/*.rb'
+    @env['WITH_ID'] = true
+    @sd.setup @env
+    @sd.loadModels
+    @sd.dumpModels
+    assert @sd.last_record.include?("id"), "WITH_ID must include id"
+  end
+
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-28 00:00:00.000000000 Z
+date: 2012-08-01 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
@@ -24,6 +24,7 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
+- lib/clip.rb
 - lib/seed_dump.rb
 - lib/seed_dump/perform.rb
 - lib/seed_dump/railtie.rb