toku 0.1.0 → 0.1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a28d71264c08bd9a4e7fb28ba09b15c31ba48a57
-  data.tar.gz: 2818fa86da6e86ce6763dbda685e02d46288036e
+  metadata.gz: 67d6e00338c1d49735debcaf61650c850b838994
+  data.tar.gz: 45d23c0bca9a5403ab90e5c1f3c10d0886efb00a
 SHA512:
-  metadata.gz: 5e7c7968f76acccab091e5909fac96205e69cbaefc2dcdd735a56d5dd47987027e1e0f086ed550e862a80628e214ea80198b26d429fb36cc81c5cd9026dbec4e
-  data.tar.gz: ba307d99a54bf305eac18232f14f2f7fa3f80b67e5466f287e7d89c8d0c2f9a2bd83b4c6692cac4f3536cdf861ef343a5f1b74e025cafb6a885301e62bfb1fd7
+  metadata.gz: 01227a3efdcbd534c1efa543a9981da54d746e08afa543df5d1a4b84cb8f5341bd7a949286b849d14d736b7855f868b115e171bc8facb36d0fa4f324e94d34dc
+  data.tar.gz: 0b1b22fa0273d02f3270cdc7700871ad36ded10f9b3fd2b47ec607ec99063aa1a25a1596ffeb56e6dc04f6f94a7ecce739c916ec94eac99e0e837e2d0a77eef0

data/lib/toku.rb

@@ -37,21 +37,20 @@ module Toku
       source_db.extension(:pg_streaming)
 
       source_db.tables.each do |table|
-        if @config[table.to_s]['columns'].count < source_db.from(table).columns.count
+        if !row_filters?(table) && @config[table.to_s]['columns'].count < source_db.from(table).columns.count
           raise Toku::ColumnFilterMissingError
         end
 
         row_enumerator = source_db[table].stream.to_enum
 
         @config[table.to_s]['rows'].each do |f|
-          row_filter = self.row_filters[f.keys.first.to_sym]
           if f.is_a? String
-            options = nil
+            row_filter = self.row_filters[f.to_sym].new
           elsif f.is_a? Hash
-            options = f.values.first
+            row_filter = self.row_filters[f.keys.first.to_sym].new(f.values.first)
           end
 
-          row_enumerator = row_filter.new(options).call(row_enumerator)
+          row_enumerator = row_filter.call(row_enumerator)
         end
 
         row_enumerator = row_enumerator.map { |row| transform(row, table) }
@@ -71,5 +70,9 @@ module Toku
         source_db.schema(table) == destination_db.schema(table)
       end
     end
+
+    def row_filters?(table)
+      !@config[table.to_s]['rows'].nil? && @config[table.to_s]['rows'].any?
+    end
   end
 end

data/lib/toku/filters/row/drop.rb

@@ -2,7 +2,7 @@ module Toku
   class RowFilter
     class Drop < Toku::RowFilter
       def call(stream)
-        stream.each { |row| nil }
+        stream.select { |row| false }
       end
     end
   end

data/lib/toku/version.rb

@@ -1,3 +1,3 @@
 module Toku
-  VERSION = "0.1.0"
+  VERSION = "0.1.0.1"
 end

data/toku.gemspec

@@ -6,7 +6,7 @@ require 'toku/version'
 Gem::Specification.new do |spec|
   spec.name          = "toku"
   spec.version       = Toku::VERSION
-  spec.authors       = ["PSKL"]
+  spec.authors       = ["PSKL", "lordofthelake"]
   spec.email         = ["hello@pascal.cc"]
 
   spec.summary       = %q{Anonymize a database, fast}

metadata

@@ -1,14 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: toku
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.0.1
 platform: ruby
 authors:
 - PSKL
+- lordofthelake
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-12-28 00:00:00.000000000 Z
+date: 2017-12-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler