vangogh 0.0.1 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 80adbf0ea988235079c45eadfedd8665363c1736
-  data.tar.gz: 58f370396da56f00f2f1c618831df14b6fbdd71c
+  metadata.gz: 975a2f5d4d0a1890a6ce46df6c4151829f88f0d4
+  data.tar.gz: f0ef17b4ced2a78632e6b24fd5e7facb3a8eab9f
 SHA512:
-  metadata.gz: 9922f54a815b3a48f3a88a6c693e18eac872e3eacaaa37efb70148348fd075ea3843db64fad6339f2691a554ba817f7ea5aa8a541557937e1933614808153595
-  data.tar.gz: b77ca744bde7b0fba3f889f37441cdd3a31fbcaf05ccdea6fb49c7a5a856f759234616d61d3de0660f18d49297e5dd536ccfeea2e2cd2f63b702bdfa6e535f36
+  metadata.gz: 12133477da376c8318233cc5b4e352251314311da60d20ca21489a32430c9cd4fcc1c8f0d55997fb26e89d62f951b5149491d44c8358634e9c11b3c27f2f0d71
+  data.tar.gz: 8881fefa3ef6e945d8c00cd44c5b70fba0a098ca5ed66967a09a7b2ad5c1ac2ac8c540f96503c7ea22e8e299a5f69b277e2a32dea5a2fa3a81e4e78a51100f48

data/.gitignore

@@ -12,3 +12,4 @@
 *.o
 *.a
 mkmf.log
+*.gem

data/README.md

@@ -13,7 +13,7 @@ Vangogh.abstract 12345
 => 12000
 
 # Abstract and abbreviate
-Vangogh.abbrevaite 1234
+Vangogh.abbreviate 1234
 => "1.2k"
 
 Vangogh.abbreviate 12345
@@ -40,7 +40,21 @@ $ vangogh abstract 1234
 
 $ vangogh abbreviate 12345
 12.3k
-``
+```
+
+## Abstracting CSVs
+
+Vangogh isn't just a single-medium artist. If you pass Vangogh the path to a CSV, it'll process all numeric-looking cells, and pipe the resulting CSV to STDOUT.
+
+```
+$ vangogh abstract csv.csv >> output.csv
+```
+
+If you want to explicitly include/exclude certain fields, you can do so with the `fields` argument.
+
+```
+$ vangogh abstract csv.csv --fields foo bar >> output.csv
+```
 
 ## Contributing
 

data/bin/vangogh

@@ -1,16 +1,62 @@
 #!/usr/bin/env ruby
 require_relative "../lib/vangogh"
 require 'thor'
+require 'csv'
 
 class VangoghCli < Thor
-  desc "abstract VALUE", "Abstract the given value (e.g., 1234 -> 1200)"
+
+  class_option :fields, :type => :array, :desc => "Fields to act on (default: all numeric fields)"
+
+  desc "abstract VALUE|CSV [--fields FIELDS]", "Abstract the given value (e.g., 1234 -> 1200)"
   def abstract(value)
-    puts Vangogh.abstract(value.to_f)
+    if csv?(value)
+      puts process_csv(value, :abstract)
+    else
+      puts Vangogh.abstract(value)
+    end
   end
 
-  desc "abbreviate VALUE", "Abbreviate the given value (e.g., 1234 -> 1.2k)"
+  desc "abbreviate VALUE|CSV [--fields FIELDS]", "Abbreviate the given value (e.g., 1234 -> 1.2k)"
   def abbreviate(value)
-    puts Vangogh.abbreviate(value.to_f)
+    if csv?(value)
+      puts process_csv(value, :abbreviate)
+    else
+      puts Vangogh.abbreviate(value)
+    end
+  end
+
+  private
+
+  no_commands do
+    def read_csv(csv)
+      CSV.read(csv, :headers => true)
+    end
+
+    def process_csv(path, process)
+      raise ArgumentError unless [:abstract, :abbreviate].include?(process)
+      fields = options["fields"] || []
+
+      csv = CSV.generate do |output|
+        input = read_csv(path)
+        input.each do |row|
+          output_row = []
+          output << input.headers
+          row.each do |key, value|
+            if (fields.empty? || fields.include?(key.strip)) && Vangogh.numeric?(value)
+              output_row.push Vangogh.send(process, value)
+            else
+              output_row.push value
+            end
+          end
+          output << output_row
+        end
+      end
+      csv
+    end
+
+    def csv?(value)
+      value =~ /\.csv$/ && File.exists?(value)
+    end
   end
 end
 

data/lib/vangogh/number.rb

@@ -4,7 +4,11 @@ module Vangogh
     attr_accessor :value
 
     def initialize(value)
-      @value = value
+      if value.is_a?(String)
+        @value = value.strip.to_f
+      else
+        @value = value
+      end
     end
 
     def abstract

data/lib/vangogh/version.rb

@@ -1,3 +1,3 @@
 module Vangogh
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/lib/vangogh.rb

@@ -11,5 +11,9 @@ module Vangogh
     def abbreviate(value)
       Number.new(value).abbreviate
     end
+
+    def numeric?(text)
+      !!(text.to_s.strip =~ /\A[\d\,.]+\z/)
+    end
   end
 end

data/spec/fixtures/csv.csv

@@ -0,0 +1,2 @@
+foo,bar
+baz,12345

data/spec/vangogh_bin_spec.rb

@@ -8,4 +8,18 @@ describe "Vangogh::Bin" do
   it "abbreviates values" do
     expect(`bundle exec vangogh abbreviate 1234`.strip).to eql("1.2k")
   end
+
+  it "abstracts CSVs" do
+    expect(`bundle exec vangogh abstract spec/fixtures/csv.csv`.strip).to match(/baz\,12000$/)
+  end
+
+  it "abbreviates CSVs" do
+    expect(`bundle exec vangogh abbreviate spec/fixtures/csv.csv`.strip).to match(/baz\,12k$/)
+  end
+
+  it "respects the field flag" do
+    expect(`bundle exec vangogh abbreviate spec/fixtures/csv.csv --fields foo`.strip).to match(/baz\,12345$/)
+
+    expect(`bundle exec vangogh abbreviate spec/fixtures/csv.csv --fields bar`.strip).to match(/baz\,12k$/)
+  end
 end

data/spec/vangogh_spec.rb

@@ -8,4 +8,15 @@ describe Vangogh do
   it "abbreviates numbers" do
     expect(subject.abbreviate(1234)).to eql("1.2k")
   end
+
+  it "knows when something is nummeric" do
+    expect(subject.numeric?(12345)).to eql(true)
+    expect(subject.numeric?("12345")).to eql(true)
+    expect(subject.numeric?("12,345")).to eql(true)
+    expect(subject.numeric?("123.45")).to eql(true)
+    
+    expect(subject.numeric?("123.45MB")).to eql(false)
+    expect(subject.numeric?("123.45 foo")).to eql(false)
+    expect(subject.numeric?("foo")).to eql(false)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vangogh
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
 platform: ruby
 authors:
 - Ben Balter
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-27 00:00:00.000000000 Z
+date: 2015-07-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: si
@@ -116,6 +116,7 @@ files:
 - script/cibuild
 - script/console
 - script/release
+- spec/fixtures/csv.csv
 - spec/helper.rb
 - spec/vangogh_bin_spec.rb
 - spec/vangogh_ext_spec.rb
@@ -147,6 +148,7 @@ signing_key:
 specification_version: 4
 summary: A Ruby Gem for abstracting numbers
 test_files:
+- spec/fixtures/csv.csv
 - spec/helper.rb
 - spec/vangogh_bin_spec.rb
 - spec/vangogh_ext_spec.rb