database-cloner-rails 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2894784ecfef1a9062058766e1c00c741a570daafa530b834d5b6107ca91919f
-  data.tar.gz: b5c02fde16a181427e684a4bb5075a54741789e95c14843890070f79be16acff
+  metadata.gz: 5b9746f0df92bf39ad438cdcd9b6b214a65dc5041bb259aa29358f2144176b69
+  data.tar.gz: 02a3b5b3e7e3d16a8be41c727690a15b591215cbd091972afebd8d664450f024
 SHA512:
-  metadata.gz: faf0e1434b290814d8cb10c2f46ccd3e50546edd97da67e0bb07274d0d726be2c2bb57392e271d904dea95781df8e6649fb8008bab681d2b768cd8955ddc64c8
-  data.tar.gz: 3d073850db2e76aec441781a9a0f33da8f0ff33dd426b5837fcebc9dd0fadca337bc0420a5c8e0bd3c99c33720cd6c8ec24580d894b2d00247390ce8a49833f9
+  metadata.gz: ec3b1c953a9b0b9be224ec786eddecfadbe2b40330b5327f3bc3e374f50983169c2423469b1110525718ebe30c6deadbddbcf86229109adf7dda34f5ed280e04
+  data.tar.gz: bda35012cba75780535f111cd962ce654c8025ce57fa651b3ca98e48b3a22932da6c29f2456161668efc151cc870aac4914f545be0dad19db7353fea958bdf76

data/lib/database_cloner_rails/downloader.rb

@@ -1,19 +1,26 @@
+require "fileutils"
+require "json"
+
 module DatabaseClonerRails
   class Downloader
-    def self.run(dump_dir: default_dump_dir)
-      new(dump_dir: dump_dir).run
+    def self.run(dump_dir: default_dump_dir, only: nil)
+      new(dump_dir: dump_dir, only: only).run
     end
 
     def self.default_dump_dir
       File.join(Rails.root, "lib", "tasks", "database_cloner", "db_dump")
     end
 
-    def initialize(dump_dir: self.class.default_dump_dir)
+    def initialize(dump_dir: self.class.default_dump_dir, only: nil)
       @dump_dir = dump_dir
+      @only     = Array(only).map(&:to_s).reject(&:empty?)
     end
 
     def run
-      ActiveRecord::Base.connection.tables.each do |table|
+      FileUtils.mkdir_p(@dump_dir)
+      tables = ActiveRecord::Base.connection.tables
+      tables = tables.select { |t| @only.include?(t) } if @only.any?
+      tables.each do |table|
         model = table.classify.safe_constantize
         next unless model.present?
         dump_table(table, model)

data/lib/database_cloner_rails/uploader.rb

@@ -1,11 +1,17 @@
 module DatabaseClonerRails
   class Uploader
-    def self.run
-      new.run
+    def self.run(only: nil)
+      new(only: only).run
+    end
+
+    def initialize(only: nil)
+      @only = Array(only).map(&:to_s).reject(&:empty?)
     end
 
     def run
-      ActiveRecord::Base.connection.tables.each do |table|
+      tables = ActiveRecord::Base.connection.tables
+      tables = tables.select { |t| @only.include?(t) } if @only.any?
+      tables.each do |table|
         next unless table.classify.safe_constantize.present?
         begin
           require "tasks/database_cloner/db_dump/#{table}"

data/lib/database_cloner_rails/version.rb

@@ -1,3 +1,3 @@
 module DatabaseClonerRails
-  VERSION = "1.0.1"
+  VERSION = "1.1.0"
 end

data/lib/tasks/database.rake

@@ -1,11 +1,13 @@
 namespace :database do
-  desc "Dump all model records to Ruby files in lib/tasks/database_cloner/db_dump/"
+  desc "Dump all model records to Ruby files in lib/tasks/database_cloner/db_dump/. Use MODELS=users,posts to filter."
   task download: :environment do
-    DatabaseClonerRails::Downloader.run
+    only = ENV["MODELS"]&.split(",")&.map(&:strip)
+    DatabaseClonerRails::Downloader.run(only: only)
   end
 
-  desc "Restore model records from dumped Ruby files"
+  desc "Restore model records from dumped Ruby files. Use MODELS=users,posts to filter."
   task upload: :environment do
-    DatabaseClonerRails::Uploader.run
+    only = ENV["MODELS"]&.split(",")&.map(&:strip)
+    DatabaseClonerRails::Uploader.run(only: only)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: database-cloner-rails
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Abhishek Sharma