obfuscator 0.0.1 → 0.0.2

data/README.md

@@ -29,7 +29,6 @@ As a Ruby program:
 # Without a format, Obfuscator will fill all of the given columns
 # with dummy data based on the column's SQL type
 Obfuscator.scrub! "Message" do
-  # Would generate dummy sentences, paragraphs, and timestamps accordingly
   overwrite :title, :body, :created_at
 end
 
@@ -40,18 +39,16 @@ Obfuscator.scrub! "User" do
   overwrite :login do
     format :user_name
   end
-end
 
-Obfuscator.scrub! "Subscriber" do
   overwrite :email_address do
     format :email
   end
 end
 ```
 
-Or as a Rake task (TODO: Not finished yet):
+Or as a Rake task:
 
-    rake obfuscator:scrub["User", ["login","email"]]
+    rake obfuscator:scrub[User] COLUMNS=login,email
 
 ## Contributing
 
@@ -63,9 +60,8 @@ Or as a Rake task (TODO: Not finished yet):
 
 ## TODO
 
-* Get Rake task fully working
-* Move away from iterating through all records and calling
-  update_attributes. Better: finding in batches. Best: Generating a
-  giant update_all.
-* Definite jankiness with the Dsl (especially that you can't pass
-  multiple overwrite/format blocks in one parent block currently)
+* Reflect on model validations to ensure generated data is valid
+* Make DSL more flexible about taking both arrays and hashes of columns
+  with options
+* Move away from iterating through batches of records and calling
+  update_attributes by constructing an update_all

data/lib/obfuscator/dsl.rb

@@ -32,7 +32,12 @@ module Obfuscator
       end
 
       def format(format)
-        @columns = { columns.first => format }
+        if @columns_hash.present?
+          @columns = @columns_hash.merge(columns.first => format)
+        else
+          @columns = { columns.first => format }
+          @columns_hash = @columns
+        end
       end
     end
 

data/lib/obfuscator/generic.rb

@@ -74,11 +74,13 @@ module Obfuscator
     end
 
     def scrub_all_records!
-      @model.all.each do |m|
-        attributes = model_columns_contain_given? ?
-          columns_with_obfuscated_values_hash : {}
+      @model.find_in_batches do |group|
+        group.each do |record|
+          attributes = model_columns_contain_given? ?
+            columns_with_obfuscated_values_hash : {}
 
-        m.update_attributes(attributes)
+          record.update_attributes(attributes)
+        end
       end
     end
   end

data/lib/obfuscator/version.rb

@@ -1,3 +1,3 @@
 module Obfuscator
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/tasks/obfuscator.rake

@@ -1,12 +1,11 @@
 namespace :obfuscator do
-  desc "Scrub sensitive data in provided model's columns"
-
-  task :scrub, [:model, :columns] => :environment do |task, arguments|
+  desc "Scrub sensitive data for a given model's column"
+  task :scrub, [:model] => :environment do |task, arguments|
     model   = arguments[:model]
-    columns = arguments[:columns]
-
-    obfuscator = Obfuscator::Generic.new
+    columns = ENV["COLUMNS"].split(",")
 
-    obfuscator.scrub!(model, columns)
+    Obfuscator.scrub! model do
+      overwrite columns
+    end
   end
 end

data/spec/obfuscator/dsl_spec.rb

@@ -53,12 +53,16 @@ describe Obfuscator::Dsl do
   describe ".format" do
     it "sets the type of the column to the given format" do
       Obfuscator::Generic.any_instance.should_receive(:scrub!).
-        with("User", { login: :user_name })
+        with("User", { login: :user_name, email: :email })
 
       Obfuscator.scrub!("User") do
         overwrite :login do
           format :user_name
         end
+
+        overwrite :email do
+          format :email
+        end
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: obfuscator
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-03 00:00:00.000000000Z
+date: 2012-06-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
-  requirement: &70183958259900 !ruby/object:Gem::Requirement
+  requirement: &70194697721360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70183958259900
+  version_requirements: *70194697721360
 - !ruby/object:Gem::Dependency
   name: faker
-  requirement: &70183958259480 !ruby/object:Gem::Requirement
+  requirement: &70194697718000 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70183958259480
+  version_requirements: *70194697718000
 description: A clean, friendly API for obfuscating slightly sensitive data from a
   Rails application
 email:
@@ -80,7 +80,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.16
+rubygems_version: 1.8.15
 signing_key: 
 specification_version: 3
 summary: Obfuscate provides an easy way to quickly overwrite slightly sensitive data