Sha256: 3118d010309703a8822f740a065b3a48cc96c39f4c7b225aaf9c706348deba50
Contents?: true
Size: 1.72 KB
Versions: 6
Compression:
Stored size: 1.72 KB
Contents
require 'csv'
if CSV.const_defined? :Reader
# Ruby 1.8 compatible
require 'fastercsv'
Object.send(:remove_const, :CSV)
CSV = FasterCSV
else
# CSV is now FasterCSV in ruby 1.9
end
module HammerCLI::Output::Adapter
class CSValues < Abstract
def tags
[:flat]
end
def print_record(fields, record)
print_collection(fields, [record].flatten(1))
end
def print_collection(fields, collection)
csv_string = generate do |csv|
# labels
csv << fields.select{ |f| !(f.class <= Fields::Id) || @context[:show_ids] }.map { |f| f.label }
# data
collection.each do |d|
csv << fields.inject([]) do |row, f|
unless f.class <= Fields::Id && !@context[:show_ids]
value = (f.get_value(d) || '')
formatter = @formatters.formatter_for_type(f.class)
row << (formatter ? formatter.format(value) : value)
end
row
end
end
end
puts csv_string
end
def print_message(msg, msg_params={})
csv_string = generate do |csv|
id = msg_params["id"] || msg_params[:id]
name = msg_params["name"] || msg_params[:name]
labels = ["Message"]
data = [msg.format(msg_params)]
if id
labels << "Id"
data << id
end
if name
labels << "Name"
data << name
end
csv << labels
csv << data
end
puts csv_string
end
private
def generate(&block)
CSV.generate(
:col_sep => @context[:csv_separator] || ',',
:encoding => 'utf-8',
&block
)
end
end
HammerCLI::Output::Output.register_adapter(:csv, CSValues)
end
Version data entries
6 entries across 6 versions & 1 rubygems