molo 0.7.4 → 0.7.5

data/Molo.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{molo}
-  s.version = "0.7.4"
+  s.version = "0.7.5"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Joel Moss", "Todd Huss", "Michael Grosser"]
-  s.date = %q{2010-12-20}
+  s.date = %q{2010-12-21}
   s.email = %q{joel@developwithstyle.com}
   s.extra_rdoc_files = [
     "README.markdown"

data/VERSION

@@ -1 +1 @@
-0.7.4
+0.7.5

data/lib/tasks/molo.rb

@@ -244,14 +244,14 @@ class MigratorTasks < ::Rake::TaskLib
     		task :load => :ar_init do
           format_class = ENV['class'] || "YamlDb::Helper"
           helper = format_class.constantize
-    			SerializationHelper::Base.new(helper).load(db_dump_data_file helper.extension)
+    			SerializationHelper::Base.new(helper).load db_dump_data_file(helper.extension), table_options
     		end
 
     		desc "Load contents of db/data/* into database"
     		task :load_dir  => :ar_init do
           dir = ENV['dir'] || "data"
           format_class = ENV['class'] || "YamlDb::Helper"
-    	    SerializationHelper::Base.new(format_class.constantize).load_from_dir dump_dir("/#{dir}")
+    	    SerializationHelper::Base.new(format_class.constantize).load_from_dir dump_dir("/#{dir}"), table_options
     		end
     	end
 

data/vendor/yaml_db/lib/serialization_helper.rb

@@ -49,9 +49,26 @@ module SerializationHelper
       reenable_logger
     end
 
-    def load_from_dir(dirname, truncate = true)
+    def load_from_dir(dirname, *opts)
+      options = opts.extract_options!
+      
+      truncate = options[:truncate] || true
+      except = options[:except].is_a?(String) ? options[:except].split(',') : options[:except]
+      only = options[:only].is_a?(String) ? options[:only].split(',') : options[:only]
+      
       Dir.entries(dirname).each do |filename|
         next if filename =~ /^[.]/
+        table = filename.gsub(/\.[a-z]+/, "")
+        
+        if except.present? and except.include? table
+          puts "    excluding #{filename}"
+          next
+        end
+        
+        if only.present? and !only.include? table
+          puts "    excluding #{filename}"
+          next
+        end
         
         puts " -- loading data: #{filename}"
         

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: molo
 version: !ruby/object:Gem::Version 
-  hash: 11
+  hash: 9
   prerelease: false
   segments: 
   - 0
   - 7
-  - 4
-  version: 0.7.4
+  - 5
+  version: 0.7.5
 platform: ruby
 authors: 
 - Joel Moss
@@ -17,7 +17,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-20 00:00:00 +00:00
+date: 2010-12-21 00:00:00 +00:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency