rcsv 0.0.8 → 0.0.9

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rcsv (0.0.8)
+    rcsv (0.0.9)
 
 GEM
   remote: https://rubygems.org/

data/RELNOTES

@@ -1,3 +1,6 @@
+Version 0.0.9
+  * preliminary write support
+
 Version 0.0.8
   * libcsv upgraded to 3.0.3
   * :parse_empty_fields_as option added

data/lib/rcsv.rb

@@ -4,6 +4,11 @@ require "rcsv/version"
 require "stringio"
 
 class Rcsv
+
+  attr_reader :write_options
+
+  BOOLEAN_FALSE = [nil, false, 0, 'f', 'false']
+
   def self.parse(csv_data, options = {}, &block)
     #options = {
       #:column_separator => "\t",
@@ -102,4 +107,56 @@ class Rcsv
     csv_data.pos = initial_position
     return self.raw_parse(csv_data, raw_options, &block)
   end
+
+  def initialize(write_options = {})
+    @write_options = write_options
+    @write_options[:column_separator] ||= ','
+    @write_options[:newline_delimiter] ||= "\r\n" # Making Excel happy...
+    @write_options[:header] ||= false
+  end
+
+  def write(io, &block)
+    io.write generate_header if @write_options[:header]
+    while row = yield
+      io.write generate_row(row)
+    end
+  end
+
+  def generate_header
+    return @write_options[:columns].map { |c|
+      c[:name].to_s
+    }.join(@write_options[:column_separator]) << @write_options[:newline_delimiter]
+  end
+
+  def generate_row(row)
+    column_separator = @write_options[:column_separator]
+    csv_row = ''
+    max_index = row.size - 1
+
+    row.each_with_index do |field, index|
+      unquoted_field = process(field, @write_options[:columns][index])
+      # TODO: a better quoting
+      csv_row << (unquoted_field.match(/,/) ? "\"#{unquoted_field}\"" : unquoted_field)
+      csv_row << column_separator unless index == max_index
+    end
+
+    return csv_row << @write_options[:newline_delimiter]
+  end
+
+  protected
+
+  def process(field, column_options)
+    return case column_options[:formatter]
+    when :strftime
+      format = column_options[:format] || "%Y-%m-%d %H:%M:%S %z"
+      field.strftime(format)
+    when :printf
+      format = column_options[:format] || "%s"
+      sprintf(format, field)
+    when :boolean
+      BOOLEAN_FALSE.include?(field.respond_to?(:downcase) ? field.downcase : field) ? 'false' : 'true'
+    else
+      field.to_s
+    end
+  end
 end

data/lib/rcsv/version.rb

@@ -1,3 +1,3 @@
 class Rcsv
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/test/test_rcsv_raw_parse.rb

@@ -1,7 +1,7 @@
 require 'test/unit'
 require 'rcsv'
 
-class RcsvTest < Test::Unit::TestCase
+class RcsvRawParseTest < Test::Unit::TestCase
   def setup
     @csv_data = File.open('test/test_rcsv.csv')
   end

data/test/test_rcsv_write.rb

@@ -0,0 +1,79 @@
+require 'test/unit'
+require 'rcsv'
+
+class RcsvWriteTest < Test::Unit::TestCase
+  def setup
+    @options = {
+      :header => true,
+      :columns => [
+        {
+          :name => 'ID'
+        },
+        {
+          :name => 'Date',
+          :formatter => :strftime,
+          :format => '%Y-%m-%d'
+        },
+        {
+          :name => 'Money',
+          :formatter => :printf,
+          :format => '$%2.2f'
+        },
+        {
+          :name => 'Banana IDDQD'
+        },
+        {
+          :name => nil,
+          :formatter => :boolean
+        }
+      ]
+    }
+
+    @data = [
+      [1, Date.parse('2012-11-11'), 100.234, true, nil],
+      [nil, Date.parse('1970-01-02'), -0.1, :nyancat, 0],
+      [3, Date.parse('2012-12-12'), 0, 'sepulka', 'zoop']
+    ]
+
+    @writer = Rcsv.new(@options)
+  end
+
+  def test_rcsv_generate_header
+    assert_equal(
+      "ID,Date,Money,Banana IDDQD,\r\n", @writer.generate_header
+    )
+  end
+
+  def test_rscv_generate_row
+    assert_equal("1,2012-11-11,$100.23,true,false\r\n", @writer.generate_row(@data.first))
+  end
+
+  def test_rcsv_write
+    io = StringIO.new
+
+    @writer.write(io) do
+      @data.shift
+    end
+
+    io.rewind
+
+    assert_equal(
+      "ID,Date,Money,Banana IDDQD,\r\n1,2012-11-11,$100.23,true,false\r\n,1970-01-02,$-0.10,nyancat,false\r\n3,2012-12-12,$0.00,sepulka,true\r\n", io.read
+    )
+  end
+
+  def test_rcsv_write_no_headers
+    io = StringIO.new
+    @writer.write_options[:header] = false
+
+    @writer.write(io) do
+      @data.shift
+    end
+
+    io.rewind
+
+    assert_equal(
+      "1,2012-11-11,$100.23,true,false\r\n,1970-01-02,$-0.10,nyancat,false\r\n3,2012-12-12,$0.00,sepulka,true\r\n", io.read
+    )
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rcsv
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-11 00:00:00.000000000 Z
+date: 2013-01-22 00:00:00.000000000 Z
 dependencies: []
 description: A libcsv-based CSV parser for Ruby
 email:
@@ -40,6 +40,7 @@ files:
 - rcsv.gemspec
 - test/test_rcsv.csv
 - test/test_rcsv_raw_parse.rb
+- test/test_rcsv_write.rb
 homepage: http://github.com/fiksu/rcsv
 licenses: []
 post_install_message: 
@@ -69,4 +70,5 @@ summary: Fast CSV parsing library for MRI based on libcsv. Supports type convers
 test_files:
 - test/test_rcsv.csv
 - test/test_rcsv_raw_parse.rb
+- test/test_rcsv_write.rb
 has_rdoc: