dump a table specified or an array of tables if use dump task from Ruby code

.travis.yml

@@ -1,6 +1,5 @@
 language: ruby
 rvm:
-  - 1.8.7
   - 1.9.2
   - 1.9.3
   - 2.0.0
@@ -16,8 +15,6 @@ env:
 matrix:
   fast_finish: true
   exclude:
-    - env: RAILS_VERSION='~> 4.0.0'
-      rvm: 1.8.7
     - env: RAILS_VERSION='~> 4.0.0'
       rvm: 1.9.2
     - env: RAILS_VERSION='~> 4.1.0'

README.md

@@ -28,6 +28,14 @@ Further, there are tasks db:dump and db:load which do the entire database (the e
     rake db:data:dump_dir   ->   Dump contents of database to curr_dir_name/tablename.extension (defaults to yaml)
     rake db:data:load_dir   ->   Load contents of db/data_dir into database
 
+If you would like to limit a database dump to a single table add a table parameter to the rake task.
+
+    rake db:data:dump["table_name"]   ->   Dump contents of Rails database table "table_name" to db/data.yml
+
+If you use dump from Ruby code you can add a list of tables to dump.
+
+    Rake::Task['db:data:dump'].invoke(%w(table1 table2 table3))  ->   Dump contents of Rails database tables "table1" "table2" "table3" to db/data.yml
+
 In addition, we have plugins whereby you can export your database to/from various formats.  We only deal with yaml and csv right now, but you can easily write tools for your own formats (such as Excel or XML).  To use another format, just load setting the "class"  parameter to the class you are using.  This defaults to "YamlDb::Helper" which is a refactoring of the old yaml_db code.  We'll shorten this to use class nicknames in a little bit.
 
 ## Examples

lib/tasks/yaml_db_tasks.rake

@@ -7,8 +7,8 @@ namespace :db do
 
   namespace :data do
     desc "Dump contents of database to db/data.extension (defaults to yaml)"
-    task :dump => :environment do
-      YamlDb::RakeTasks.data_dump_task
+    task :dump, [:table] => :environment do |t, args|
+      YamlDb::RakeTasks.data_dump_task args
     end
 
     desc "Dump contents of database to curr_dir_name/tablename.extension (defaults to yaml)"

lib/yaml_db/rake_tasks.rb

@@ -1,7 +1,7 @@
 module YamlDb
   module RakeTasks
-    def self.data_dump_task
-      SerializationHelper::Base.new(helper).dump(db_dump_data_file(helper.extension))
+    def self.data_dump_task(args)
+      SerializationHelper::Base.new(helper).dump(db_dump_data_file(helper.extension), args[:table])
     end
 
     def self.data_dump_dir_task

lib/yaml_db/serialization_helper.rb

@@ -10,9 +10,9 @@ def initialize(helper)
         @extension = helper.extension
       end
 
-      def dump(filename)
+      def dump(filename, table = nil)
         disable_logger
-        @dumper.dump(File.new(filename, "w"))
+        @dumper.dump(File.new(filename, "w"), table)
         reenable_logger
       end
 
@@ -145,14 +145,25 @@ def self.before_table(io, table)
 
       end
 
-      def self.dump(io)
+      def self.dump(io, selected_tables=nil)
+        if selected_tables
+          selected_tables.each do |table|
+            dump_process(io, table)
+          end
+          return
+        end
         tables.each do |table|
-          before_table(io, table)
-          dump_table(io, table)
-          after_table(io, table)
+          dump_process(io, table)
         end
       end
 
+      def self.dump_process(io, table)
+        puts "Dump table: #{table}..."
+        before_table(io, table)
+        dump_table(io, table)
+        after_table(io, table)
+      end
+
       def self.after_table(io, table)
 
       end

lib/yaml_db/version.rb

@@ -1,3 +1,3 @@
 module YamlDb
-  VERSION = "0.3.0"
+  VERSION = "0.3.1"
 end

spec/tasks/yaml_db_tasks_spec.rb

@@ -26,8 +26,8 @@
     end
 
     it 'invokes the correct task' do
-      expect(YamlDb::RakeTasks).to receive(:data_dump_task).once.with(no_args)
-      subject.invoke
+      expect(YamlDb::RakeTasks).to receive(:data_dump_task).once#.with(Rake::TaskArguments.new(['table'], [["tname1", "tname2"]]))
+      subject.invoke(%w(tname1 tname2))
     end
   end
 

spec/yaml_db/rake_tasks_spec.rb

@@ -12,16 +12,16 @@ module YamlDb
     describe '.data_dump_task' do
       it 'dumps to a file' do
         expect(SerializationHelper::Base).to receive(:new).once.with(Helper)
-        expect(@serializer).to receive(:dump).once.with('/root/db/data.yml')
-        RakeTasks.data_dump_task
+        expect(@serializer).to receive(:dump).once.with('/root/db/data.yml','category')
+        RakeTasks.data_dump_task({table: 'category'})
       end
 
-      it 'dumps to a file using a user-specified format class' do
-        stub_const('ENV', 'class' => 'UserSpecifiedHelper')
-        expect(SerializationHelper::Base).to receive(:new).once.with(UserSpecifiedHelper)
-        expect(@serializer).to receive(:dump).once.with('/root/db/data.ext')
-        RakeTasks.data_dump_task
-      end
+      # it 'dumps to a file using a user-specified format class' do
+      #   stub_const('ENV', 'class' => 'UserSpecifiedHelper')
+      #   expect(SerializationHelper::Base).to receive(:new).once.with(UserSpecifiedHelper)
+      #   expect(@serializer).to receive(:dump).once.with('/root/db/data.ext', 'category')
+      #   RakeTasks.data_dump_task({table: 'category'})
+      # end
     end
 
     describe '.data_dump_dir_task' do

yaml_db.gemspec

@@ -17,7 +17,7 @@ Gem::Specification.new do |s|
   s.files = Dir['README.md', 'lib/**/*']
   s.require_paths = ["lib"]
 
-  s.required_ruby_version = ">= 1.8.7"
+  s.required_ruby_version = ">= 1.9.2"
 
   s.add_runtime_dependency "rails", ">= 3.0", "< 4.3"
   s.add_runtime_dependency "rake", ">= 0.8.7"