active_record_to_csv 0.1.3 → 0.1.4

data/README.rdoc

@@ -29,13 +29,23 @@ Note that #to_csv is called like a scope or query. The following will NOT give y
 
 This will use Ruby's Array#to_csv method.
 
+=== Attribute#to_csv
+
+After a model object's attributes are collected, to_csv is called on the resulting array. However, this poses a problem because it will blindly convert the attributes to a string -- i.e. call to_s on them. If one of your attributes is a Date, then calling to_s may produce unwanted output. For example, if you have Date::DATE_FORMATS[:default] = '%d %B, %Y' your dates will have the month written out like 'January', 'February', etc. To counter this, this gem will make an attempt to call to_csv() on each attribute. To get YYYY-MM-DD output, you could do something like:
+
+  class Date
+    def to_csv
+      strftime('%Y/%m/%d')
+    end
+  end
+
 == TODO
 
 Options to specify columns to be included (currently, id and timestamp columns are excluded).
 
 == Compatibility
 
-Tested with ActiveRecord v3.0.4
+Tested with ActiveRecord v3.0.5
 
 http://gem-testers.org/gems/active_record_to_csv
 

data/lib/active_record_to_csv.rb

@@ -9,7 +9,9 @@ module ActiveRecordToCSV
     header_row = csv_columns.to_csv
     records_rows = all.map do |record|
       csv_columns.map do |column|
-        record[column]
+        value = record[column]
+        value = value.to_csv if value.respond_to?(:to_csv)
+        value
       end.to_csv
     end.join
     header_row + records_rows
@@ -27,4 +29,3 @@ class ActiveRecord::Relation
     end
   end
 end
-

data/lib/active_record_to_csv/version.rb

@@ -1,3 +1,3 @@
 module ActiveRecordToCsv
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

data/spec/active_record_to_csv_spec.rb

@@ -10,7 +10,11 @@ describe ActiveRecordToCSV do
       "The King's Speech",
       'The Kids Are All Right'
     ].each_with_index do |title, i|
-      Movie.create! :title => title, :director_id => i
+      Movie.create!(
+        :title => title,
+        :director_id => i + 1,
+        :released_on => Date.new(2010, 1, i + 1)
+      )
     end
   end
 
@@ -23,7 +27,15 @@ describe ActiveRecordToCSV do
   end
 
   it 'should not include id and timestamp fields' do
-    Movie.to_csv.lines.first.should eq("title,director_id\n")
+    header = Movie.to_csv.lines.first
+    header.should_not =~ /^id|created_at|updated_at/
+  end
+
+  it 'should call to_csv on each attribute if it responds to it' do
+    date = Movie.limit(1).first.released_on
+    csv_string = Movie.limit(1).to_csv
+    csv_string.should include(date.to_csv)
+    csv_string.should_not include(date.to_s)
   end
 
 end

data/spec/schema.rb

@@ -6,6 +6,7 @@ ActiveRecord::Schema.define do
   create_table :movies, :force => true do |t|
     t.string :title
     t.integer :director_id
+    t.date :released_on
     t.timestamps
   end
 end

data/spec/support/fixtures/movies.csv

@@ -1,6 +1,6 @@
-title,director_id
-Black Swan,0
-Inception,1
-The Fighter,2
-The King's Speech,3
-The Kids Are All Right,4
+title,director_id,released_on
+Black Swan,1,2010/01/01
+Inception,2,2010/01/02
+The Fighter,3,2010/01/03
+The King's Speech,4,2010/01/04
+The Kids Are All Right,5,2010/01/05

data/spec/support/fixtures/movies_bad.csv

@@ -1,2 +1,2 @@
-title,director_id
-The Kids Are All Right,4
+title,director_id,released_on
+The Kids Are All Right,5,2010/01/05

data/spec/support/models/date.rb

@@ -0,0 +1,5 @@
+class Date
+  def to_csv
+    strftime('%Y/%m/%d')
+  end
+end

metadata

@@ -2,7 +2,7 @@
 name: active_record_to_csv
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Jared Ning
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-03-01 00:00:00 -06:00
+date: 2011-03-02 00:00:00 -06:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -71,6 +71,7 @@ files:
 - spec/support/fixtures/movies.csv
 - spec/support/fixtures/movies_bad.csv
 - spec/support/macros.rb
+- spec/support/models/date.rb
 - spec/support/models/movie.rb
 has_rdoc: true
 homepage: https://github.com/ordinaryzelig/active_record_to_csv
@@ -107,4 +108,5 @@ test_files:
 - spec/support/fixtures/movies.csv
 - spec/support/fixtures/movies_bad.csv
 - spec/support/macros.rb
+- spec/support/models/date.rb
 - spec/support/models/movie.rb