-
Notifications
You must be signed in to change notification settings - Fork 78
Home
Comma is a CSV (i.e. comma separated values) generation extension for Ruby objects, that lets you seamlessly define a CSV output format via a small DSL. Comma works well on pure Ruby objects with attributes, as well as complex ones such as ActiveRecord objects with associations, extensions, etc. It doesn't distinguish between attributes, methods, associations, extensions, etc. - they all are considered equal and invoked identically via the Comma DSL description. Multiple different CSV output descriptions can also be defined.
When multiple objects in an Array are converted to CSV, the output includes generation of a header row reflected from names of the properties requested, or specified via the DSL.
CSV can be a bit of a boring format - the motivation behind Comma was to have a CSV extension that was simple, flexible, and would treat attributes, methods, associations, etc., all the same without the need for any complex configuration, and also work on Ruby objects, not just ActiveRecord or other base class derivatives.
An example Comma CSV enabled ActiveRecord class:
class Book < ApplicationRecord
# ================
# = Associations =
# ================
has_many :pages
has_one :isbn
belongs_to :publisher
# ===============
# = CSV support =
# ===============
comma do
name
description
pages :size => 'Pages'
publisher :name
isbn number_10: 'ISBN-10', number_13: 'ISBN-13'
blurb 'Summary'
end
end
Annotated, the comma description is as follows:
# starts a Comma description block, generating 2 methods #to_comma and
# #to_comma_headers for this class.
comma do
# name, description are attributes of Book with the header being reflected as
# 'Name', 'Description'
name
description
# pages is an association returning an array, :size is called on the
# association results, with the header name specified as 'Pages'
pages size: 'Pages'
# publisher is an association returning an object, :name is called on the
# associated object, with the reflected header 'Name'
publisher :name
# isbn is an association returning an object, :number_10 and :number_13 are
# called on the object with the specified headers 'ISBN-10' and 'ISBN-13'
isbn :number_10 => 'ISBN-10', :number_13 => 'ISBN-13'
# blurb is an attribute of Book, with the header being specified directly
# as 'Summary'
blurb 'Summary'
end
In the above example, any of the declarations (name, description, pages, publisher, isbn, blurb, etc), could be methods, attributes, associations, etc - no distinction during configuration is required, as everything is invoked via Ruby's #send
method.
You can get the CSV representation of any object by calling the to_comma
method, optionally providing a CSV description name to use.
Object values are automatically converted to strings via to_s
allowing you to reuse any existing to_s
methods on your objects (instead of having to call particular properties or define CSV specific output methods). Header names are also automatically humanised when reflected (eg. Replacing _
characters with whitespace). The 'isbn' example above shows how multiple values can be added to the CSV output.
Multiple CSV descriptions can also be specified for the same class, eg:
class Book < ActiveRecord::Base
# ================
# = Associations =
# ================
has_many :pages
has_one :isbn
belongs_to :publisher
# ===============
# = CSV support =
# ===============
comma do
name
description
pages size: 'Pages'
publisher :name
isbn number_10: 'ISBN-10', number_13: 'ISBN-13'
blurb 'Summary'
end
comma :brief do
name
description
blurb 'Summary'
end
end
You can specify which output format you would like to use via an optional parameter to to_comma
:
Book.limit(10).to_comma(:brief)
Specifying no description name to to_comma is equivalent to specifying :default
as the description name.
You can pass all options for the CSV renderer (See: CSV.new
), e.g.
Book.limit(10).to_comma(style: :brief, col_sep: ';', force_quotes: true)
You can pass the :filename
option and have Comma writes the CSV output to this file:
Book.limit(10).to_comma(filename: 'books.csv')
You also can pass the :write_header
option to hide the header line (true is default):
Book.limit(10).to_comma(write_headers: false)
##Using blocks
For more complex relationships you can pass blocks for calculated values, or related values. Following the previous example here is a comma set using blocks (both with and without labels for your CSV headings):
class Publisher < ApplicationRecord
# ================
# = Associations =
# ================
has_one :primary_contact, class_name: 'User' #(basic user with a name)
has_many :users
end
class Book < ApplicationRecord
# ================
# = Associations =
# ================
has_many :pages
has_one :isbn
belongs_to :publisher
# ===============
# = CSV support =
# ===============
comma do
name
description
pages :size => 'Pages'
publisher :name
publisher { |publisher| publisher.primary_contact.name.to_s.titleize }
publisher 'Number of publisher users' do |publisher| publisher.users.size end
isbn :number_10 => 'ISBN-10', :number_13 => 'ISBN-13'
blurb 'Summary'
end
end
In the preceding example, the 2 new fields added (both based on the publisher relationship) mean that the following will be added:
- the first example 'publishers_contact' is loaded straight as a block. The value returned by the lambda is displayed with a header value of 'Publisher'
- the second example 'total_publishers_users' is sent via a hash and a custom label is set, if used in the first examples method the header would be 'Publisher', but sent as a hash the header is 'Number of publisher users'.
With __use__
field, you can reuse output formats that are defined in
the same class. In the example below, default format (:default
)
includes :minimum
format when #to_comma
is called.
class Book < ApplicationRecord
comma do
__use__ :minimum
description
isbn number_10: 'ISBN-10', number_13: 'ISBN-13'
end
comma :minimum do
name
end
end
When you want to have static value in your CSV output, you can use
__static__
field. You can provide values to output to blocks.
Without block, field will become empty.
class Book < ActiveRecord::Base
comma do
__static_column__ 'Check' do ' ' end
name
__static_column__ 'Spacer'
description
end
end
When used with Rails (ie. add 'comma' as a gem dependency), Comma automatically adds support for rendering CSV output in your controllers:
class BooksController < ApplicationController
def index
respond_to do |format|
format.csv { render csv: Book.limit(50) }
end
end
end
You can specify which output format you would like to use by specifying a style parameter or adding any available CSV option:
class BooksController < ApplicationController
def index
respond_to do |format|
format.csv { render csv: Book.limit(50), style: :brief }
end
end
end
You can also specify a different file extension ('csv' by default)
class BooksController < ApplicationController
def index
respond_to do |format|
format.csv { render csv: Book.limit(50), extension: 'txt' }
end
end
end
With the Rails renderer you can supply any of the regular parameters that you would use with to_comma such as :filename, :write_headers, :force_quotes, etc. The parameters just need to be supplied after you specify the collection for the csv as demonstrated above.
When used with Rails, comma also adds support for exporting scopes:
class Book < ActiveRecord::Base
scope :recent, => { { conditions: ['created_at > ?', 1.month.ago] } }
# ...
end
Calling the to_comma method on the scope will internally use Rails' find_each method, instantiating only 1,000 ActiveRecord objects at a time:
Book.recent.to_comma