dino_utils 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8872337516c4a023f16c68fa89a08a97cced682e
-  data.tar.gz: d8364252a4bdc7614c47f32d93d60d7baf9d751f
+  metadata.gz: 55c0be58943aed51e6dfd6febd331afe5fe72d20
+  data.tar.gz: 2e5b247b3fd9c8afb603082f4fb17958ed337caa
 SHA512:
-  metadata.gz: f9a202ee692e138e040f0f6522a9c8f106e2b207f088fa0e543eb78481d9981f8bfea20d66c427c5f372c5f631065abb5772b60c0f8440fc24f19094fe45727d
-  data.tar.gz: 039ac434917e43bfab694c301b9e826e5e6528d0e60e7c575465e2da81c2944379bf8bc2fe9b0ab5e75deafe5b66b37d38a42d84012edbce70fa8efe78ed0d93
+  metadata.gz: 1e776cb5141a8afa54aa33922072c6057eba5198f90c1782ed2ff518cea92a47f57a26c05771794afe71ac093693cedbba6dac5e9d995ef169e48478f4355be9
+  data.tar.gz: ce1da0d563c48f6292d6a56b93468af12ad033b7f7ff4b86303811b9a50b85106c76d7cd6bd04be2c8a2c73caa3bdbcac395bfebdeb664dc4228acb5bdaacb3d

data/lib/dino_utils/version.rb

@@ -1,3 +1,3 @@
 module DinoUtils
-  VERSION = "0.1.10"
+  VERSION = "0.1.11"
 end

data/lib/dino_utils.rb

@@ -4,3 +4,5 @@ require_relative 'string_monkeypatch'
 require_relative 'tn'
 require_relative 'tn/http'
 require_relative 'tn/advisory_lock'
+require_relative 'tn/csv_builder'
+require_relative 'tn/temp_file'

data/lib/tn/csv_builder.rb

@@ -0,0 +1,51 @@
+require 'csv'
+
+module TN::CSVBuilder
+  module ClassMethods
+    def csv_header(attribute, name: attribute)
+      @csv_headers ||= {}
+      @csv_headers[attribute] = name
+    end
+
+    attr_reader :csv_headers
+  end
+
+  module InstanceMethods
+    def csv(io, flush_after_each_row: false)
+      csv_header_row ||= CSV::Row.new(csv_headers.keys, csv_headers.values, true)
+      csv = CSV.new(io)
+      csv << csv_header_row
+      csv_data do |row|
+        row_data = []
+        csv_headers.each do |key, _|
+          row_data << row[key]
+        end
+        row.each do |key, _|
+          fail "#{key} not a valid csv header!" unless csv_headers.has_key?(key)
+        end
+        csv << CSV::Row.new(csv_headers, row_data)
+        csv.flush if flush_after_each_row
+      end
+      csv
+    end
+
+    def generate_csv_tempfile(flush_after_each_row: false)
+      file = TN::TempFile.new
+      csv(file, flush_after_each_row: flush_after_each_row)
+      file.close
+      if block_given?
+        yield(file)
+      end
+      file
+    end
+
+    def csv_headers
+      self.class.csv_headers
+    end
+  end
+
+  def self.included(klass)
+    klass.extend(ClassMethods)
+    klass.send(:include, InstanceMethods)
+  end
+end

data/lib/tn/temp_file.rb

@@ -0,0 +1,33 @@
+module TN
+  class TempFile
+
+    def initialize(content:nil, file:nil, name: 'tanga')
+      extension = File.extname(name)
+      file_name = SecureRandom.hex
+      @file = file || ::Tempfile.new([file_name, extension])
+      @file.binmode
+      if content
+        @file.write(content) if content
+        @file.close
+      end
+
+      if block_given?
+        yield(self)
+        done
+      end
+    end
+
+    def method_missing(method, *args, &block)
+      @file.send(method, *args, &block)
+    end
+
+    def path
+      File.absolute_path(@file.path)
+    end
+
+    def done
+      @file.close
+      @file.unlink
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dino_utils
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.1.11
 platform: ruby
 authors:
 - Joe Van Dyk
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-05-02 00:00:00.000000000 Z
+date: 2016-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -145,7 +145,9 @@ files:
 - lib/string_monkeypatch.rb
 - lib/tn.rb
 - lib/tn/advisory_lock.rb
+- lib/tn/csv_builder.rb
 - lib/tn/http.rb
+- lib/tn/temp_file.rb
 homepage: https://github.com/tanga/dino_utils
 licenses:
 - MIT