seed_dump 0.3.4 → 0.4.0

data/CHANGELOG.rdoc

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

data/README.rdoc

@@ -14,7 +14,7 @@ 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 10 records:
+Dump only data from the users and products table and dump a maximum amount of 2 records:
 
   $ rake db:seed:dump MODELS=User,Product LIMIT=2
 
@@ -78,5 +78,17 @@ NO_DATA:
 WITH_ID:
  Inlcude the +:id+ in the create options
 
+SKIP_CALLBACKS:
+ Deactivate callbacks while importing.
+
+PG_SCHEMA:
+ Postgres schema support
+
+WITHOUT_PROTECTION:
+ Skip protection for columns that are protected by default. (Automatically activated if such columns are present)
+
+MODEL_DIR:
+ Specify an alternative model dir
+
 
 Copyright (c) 2010 Rob Halff, released under the MIT license

data/Rakefile

@@ -40,7 +40,7 @@ task :test => :check_dependencies
 
 task :default => :test
 
-require 'rake/rdoctask'
+require 'rdoc/task'
 Rake::RDocTask.new do |rdoc|
   version = File.exist?('VERSION') ? File.read('VERSION') : ""
 

data/VERSION

@@ -1 +1 @@
-0.3.4
+0.4.0

data/lib/seed_dump/perform.rb

@@ -1,3 +1,5 @@
+require "true"
+
 module SeedDump
   class Perform
 
@@ -9,58 +11,76 @@ module SeedDump
       @seed_rb = "" 
       @id_set_string = ""
       @verbose = true
-      @model_dir = 'app/models/*.rb'
+      @model_dir = 'app/models/**/*.rb'
     end
 
     def setup(env)
       # config
-      @opts['with_id'] = !env["WITH_ID"].nil?
-      @opts['no-data'] = !env['NO_DATA'].nil?
+      @opts['debug'] = env["DEBUG"].true?
+      @opts['with_id'] = env["WITH_ID"].true?
+      @opts['no-data'] = env['NO_DATA'].true?
+      @opts['without_protection'] = env['WITHOUT_PROTECTION'].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'].nil? && File.exists?(@opts['file']) )
+      @opts['append']  = (env['APPEND'].true? && 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 }
+      @opts['schema']  = env['PG_SCHEMA']
+      @opts['model_dir']  = env['MODEL_DIR'] || @model_dir
     end
 
     def loadModels
-      Dir[@model_dir].sort.each do |f|
-        model = File.basename(f, '.*').camelize
+      puts "Searching in #{@opts['model_dir']} for models" if @opts['debug']
+      Dir[@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("::")
+        puts "Detected model #{model}" if @opts['debug']
         @models.push model if @opts['models'].include?(model) || @opts['models'].empty? 
       end
     end
 
+    def models
+      @models
+    end
+
     def dumpAttribute(a_s,r,k,v)
-      v = attribute_for_inspect(r,k)
-      if k == 'id' && @opts['with_id']
-        @id_set_string = "{ |c| c.#{k} = #{v} }.save"
+      if v.is_a?(BigDecimal)
+        v = v.to_s
       else
-        a_s.push("#{k.to_sym.inspect} => #{v}") unless k == 'id' && !@opts['with_id']
+        v = attribute_for_inspect(r,k)
+      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}")
       end 
     end
 
     def dumpModel(model)
       @id_set_string = ''
       create_hash = ""
+      options = ''
       rows = []
       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) }
-        if @id_set_string.empty?
-          rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
-        else
-          create_hash << "\n#{model}.create" << '( ' << attr_s.join(', ') << ' )' << @id_set_string
-        end
+        rows.push "#{@indent}{ " << attr_s.join(', ') << " }"
       } 
-      if @id_set_string.empty?
-        "\n#{model}.create([\n" << rows.join(",\n") << "\n])\n"
-      else
-        create_hash
+
+      if @opts['without_protection']
+        options = ', :without_protection => true '
       end
+
+      "\n#{model}.create([\n" << rows.join(",\n") << "\n]#{options})\n"
     end
 
     def dumpModels
@@ -69,15 +89,24 @@ module SeedDump
           m = model.constantize
           if m.ancestors.include?(ActiveRecord::Base)
             puts "Adding #{model} seeds." if @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
+            end
+
             @seed_rb << dumpModel(m) << "\n\n"
           else
             puts "Skipping non-ActiveRecord model #{model}..." if @verbose
           end
+          puts "Protection is disabled." if @verbose && @opts['without_protection']
       end
     end
 
     def writeFile
       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}"
       }
@@ -96,10 +125,17 @@ module SeedDump
       end
     end
 
+    def setSearchPath(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
 
+      setSearchPath @opts['schema'] if @opts['schema']
+
       loadModels
 
       puts "Appending seeds to #{@opts['file']}." if @opts['append']

data/lib/true.rb

@@ -0,0 +1,9 @@
+# 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/seed_dump.gemspec

@@ -4,14 +4,14 @@
 # -*- encoding: utf-8 -*-
 
 Gem::Specification.new do |s|
-  s.name = %q{seed_dump}
-  s.version = "0.3.4"
+  s.name = "seed_dump"
+  s.version = "0.4.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Rob Halff"]
-  s.date = %q{2011-07-14}
-  s.description = %q{Dump (parts) of your database to db/seeds.rb to get a headstart creating a meaningful seeds.rb file}
-  s.email = %q{rob.halff@gmail.com}
+  s.date = "2012-07-28"
+  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 = [
     "README.rdoc"
   ]
@@ -21,20 +21,22 @@ Gem::Specification.new do |s|
     "README.rdoc",
     "Rakefile",
     "VERSION",
-    "contents.yml",
     "lib/seed_dump.rb",
     "lib/seed_dump/perform.rb",
     "lib/seed_dump/railtie.rb",
     "lib/tasks/seed_dump.rake",
+    "lib/true.rb",
     "seed_dump.gemspec",
     "test/fixtures/samples.yml",
+    "test/models/nested/sample.rb",
+    "test/models/sample.rb",
     "test/seed_dump_test.rb",
     "test/test_helper.rb"
   ]
-  s.homepage = %q{http://github.com/rhalff/seed_dump}
+  s.homepage = "http://github.com/rhalff/seed_dump"
   s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.6.2}
-  s.summary = %q{{Seed Dumper for Rails}}
+  s.rubygems_version = "1.8.24"
+  s.summary = "{Seed Dumper for Rails}"
 
   if s.respond_to? :specification_version then
     s.specification_version = 3

data/test/fixtures/samples.yml

@@ -1,10 +1,54 @@
-# lo & behold!  I am a YAML comment!
-david:
- name: David Heinemeier Hansson
-  birthday: 1979-10-15
-   profession: Systems development
-    
-steve:
- name: Steve Ross Kellock
- birthday: 1974-09-27
- profession: guy with keyboard
+# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
+
+bharathiyar:
+  string: Poetry of Subramaniya Bharathiyar 
+  text: யாமறிந்த மொழிகளிலே தமிழ்மொழி போல் இனிதாவது எங்கும் காணோம்,
+பாமரராய் விலங்குகளாய், உலகனைத்தும் இகழ்ச்சிசொலப் பான்மை கெட்டு,
+நாமமது தமிழரெனக் கொண்டு இங்கு வாழ்ந்திடுதல் நன்றோ? சொல்லீர்!
+தேமதுரத் தமிழோசை உலகமெலாம் பரவும்வகை செய்தல் வேண்டும்.  
+  integer: 1
+  float: 3.3333
+  decimal: 9.99
+  datetime: 1902-07-21 17:01:31
+  timestamp: 1900-07-21 17:01:31
+  time: 1903-07-21 17:01:31
+  date: 1904-07-21
+  binary: 
+  boolean: false
+
+pushkin:
+  string: Pushkin's Bronze Horseman 
+  text: На берегу пустынных волн
+Стоял он, дум великих полн,
+И вдаль глядел. Пред ним широко
+Река неслася; бедный чёлн
+По ней стремился одиноко.
+По мшистым, топким берегам
+Чернели избы здесь и там,
+Приют убогого чухонца;
+И лес, неведомый лучам
+В тумане спрятанного солнца,
+Кругом шумел. 
+  integer: 1
+  float: 1.5
+  decimal: 9.99
+  datetime: 2012-07-21 17:01:31
+  timestamp: 2012-07-21 17:01:31
+  time: 2012-07-21 17:01:31
+  date: 2012-07-21
+  binary: 
+  boolean: false
+poem:
+  string: Anglo-Saxon Rune Poem 
+  text: ᚠᛇᚻ᛫ᛒᛦᚦ᛫ᚠᚱᚩᚠᚢᚱ᛫ᚠᛁᚱᚪ᛫ᚷᛖᚻᚹᛦᛚᚳᚢᛗ
+ᛋᚳᛖᚪᛚ᛫ᚦᛖᚪᚻ᛫ᛗᚪᚾᚾᚪ᛫ᚷᛖᚻᚹᛦᛚᚳ᛫ᛗᛁᚳᛚᚢᚾ᛫ᚻᛦᛏ᛫ᛞᚫᛚᚪᚾ
+ᚷᛁᚠ᛫ᚻᛖ᛫ᚹᛁᛚᛖ᛫ᚠᚩᚱ᛫ᛞᚱᛁᚻᛏᚾᛖ᛫ᛞᚩᛗᛖᛋ᛫ᚻᛚᛇᛏᚪᚾ 
+  integer: 1
+  float: 1.9
+  decimal: 4.55
+  datetime: 2012-07-21 17:01:31
+  timestamp: 2012-07-21 17:01:31
+  time: 2012-07-21 17:01:31
+  date: 2012-07-21
+  binary: 
+  boolean: false

data/test/models/nested/sample.rb

@@ -0,0 +1,3 @@
+class Nested::Sample < ActiveRecord::Base
+  attr_accessible :name
+end

data/test/models/sample.rb

@@ -0,0 +1,3 @@
+class Sample < ActiveRecord::Base
+  attr_accessible :binary, :boolean, :date, :datetime, :decimal, :float, :integer, :string, :text, :time, :timestamp
+end

data/test/seed_dump_test.rb

@@ -1,8 +1,30 @@
 require 'test_helper'
+require "seed_dump/perform"
 
 class SeedDumpTest < ActiveSupport::TestCase
-  # Replace this with your real tests.
-  test "test WITH_ID" do
-    assert false 
+
+  setup do
+    @sd = SeedDump::Perform.new
+    # universial options for every test
+    @env = {
+     "MODEL_DIR" => Dir.pwd + '/test/models/**.rb',
+     "FILE" => Dir.pwd + '/test/db/seeds.rb',
+     "DEBUG" => true
+    }
+  end
+
+  test "load sample model" do
+    @env['MODEL_DIR'] = Dir.pwd + '/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'
+    @sd.setup @env 
+    @sd.loadModels
+    assert_equal ["Nested::Sample", "Sample"], @sd.models
+  end
+
 end

data/test/test_helper.rb

@@ -1,3 +1,30 @@
+$LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'lib')
 require 'rubygems'
 require 'test/unit'
 require 'active_support'
+require 'active_record'
+
+#require 'your_thing'
+
+ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+
+ActiveRecord::Schema.define(:version => 1) do
+ 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
+
+#class Sample < ActiveRecord::Base
+#end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seed_dump
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.4.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,8 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-07-14 00:00:00.000000000 +02:00
-default_executable: 
+date: 2012-07-28 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
@@ -25,16 +24,17 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
-- contents.yml
 - lib/seed_dump.rb
 - lib/seed_dump/perform.rb
 - lib/seed_dump/railtie.rb
 - lib/tasks/seed_dump.rake
+- lib/true.rb
 - seed_dump.gemspec
 - test/fixtures/samples.yml
+- test/models/nested/sample.rb
+- test/models/sample.rb
 - test/seed_dump_test.rb
 - test/test_helper.rb
-has_rdoc: true
 homepage: http://github.com/rhalff/seed_dump
 licenses: []
 post_install_message: 
@@ -55,7 +55,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.6.2
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: ! '{Seed Dumper for Rails}'

data/contents.yml

@@ -1,22 +0,0 @@
-# Read about fixtures at http://ar.rubyonrails.org/classes/Fixtures.html
-article1:
-  id: 1
-  title: Article one 
-  body: body
-  extended: extended content
-  created_at: 2011-06-01 00:00:01
-  updated_at: 2011-06-01 00:00:02
-  some_id: 1
-  author: Johnny Johnson 
-  published: true
-
-article2:
-  id: 2
-  title: This is a very long title and it should not be truncated
-  body: body
-  extended: extended content
-  created_at: 2010-06-01 20:00:01
-  updated_at: 2010-06-01 20:00:02
-  some_id: 2
-  author: Bob
-  published: true