db-purger 0.2.0 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f85bace783d6c2c4bf0526a608326a9d15ee3dc0babd439a2dd86c22ab304343
-  data.tar.gz: fa1f889ecf967b73357d2764764e01a3ba7a52a9dea5737b7008e9c87edf953b
+  metadata.gz: 3d0ca2166438859558fecf9ba8a9e0a2fb4b7763467360551d428269e4cf7aa5
+  data.tar.gz: 50a53cebe730dc8240523d557997ca6186fd7ad87a2b2d6da39ac427e217204b
 SHA512:
-  metadata.gz: cc7a13db98536483613543736ed8a0bd03dc5dd1076060a4b1735e7b5499634d1e80e955cd714ce67785dcf42b0841b47ebefbaecae9143b4366c7d27be80786
-  data.tar.gz: bf6d5fa8f36eaf89087056673d1e49bfa74d4141321c078c5a0b0414fe58b8d64f1409b80838a2d05a7ae163cf11f0cf345dd2d812020a8adb4a1e5020ed1202
+  metadata.gz: 73f96ec383727353572583c2d182ac42e191fe50a7bd023d924255fba677ddb7fb806002c4417503a268cd62c1db61b23c3d4edf60ae14889d6c9f3fa3aff948
+  data.tar.gz: '09b8e2e7f5067ad8c92bf05211d5ddad143807065389d29f4848135534db8c5155418eb75e006f65819b0a47231b2fb4a91fc8b42effa0a65918018f59dd5528'

data/lib/db-purger/dynamic_plan_builder.rb

@@ -11,12 +11,13 @@ module DBPurger
       @database = database
       @output = ''.dup
       @indent_depth = 0
-      write('DBPurger::PlanBuilder.build do')
-      @indent_depth += 1
+      @tables = []
     end
 
     def build(base_table_name, field)
-      write("base_table(#{base_table_name.to_sym.inspect}, #{field.to_sym.inspect})")
+      write('DBPurger::PlanBuilder.build do')
+      @indent_depth += 1
+      write_table('base', base_table_name.to_s, field, [], nil)
       line_break
       model = find_model_for_table(base_table_name)
       foreign_key = foreign_key_name(model)
@@ -25,6 +26,7 @@ module DBPurger
         line_break
         add_child_tables(child_models, foreign_key, 0)
       end
+      ignore_missing_tables
       @indent_depth -= 1
       write('end')
       @output
@@ -86,6 +88,7 @@ module DBPurger
     end
 
     def write_table(table_type, table_name, field, child_models, foreign_key)
+      @tables << table_name
       if child_models.empty?
         write("#{table_type}_table(#{table_name.to_sym.inspect}, #{field.to_sym.inspect})")
       else
@@ -94,5 +97,15 @@ module DBPurger
         write('end')
       end
     end
+
+    def ignore_missing_tables
+      missing_tables = @database.models.map(&:table_name) - @tables
+      return if missing_tables.empty?
+
+      line_break
+      missing_tables.each do |table_name|
+        write("ignore_table #{table_name.to_sym.inspect}")
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: db-purger
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
 platform: ruby
 authors:
 - Doug Youch