Sha256: 77668c228ffa304bddedad528c3e5e1184c506baefd96950fc8f8a7b643b4ed3
Contents?: true
Size: 1.72 KB
Versions: 10
Compression:
Stored size: 1.72 KB
Contents
module CsvRowModel
module Export
class File
attr_reader :export_model_class, :csv, :file, :context
# @param [Export] export_model export model class
def initialize(export_model_class, context={})
@export_model_class = export_model_class
@context = context.to_h.symbolize_keys
end
def headers
export_model_class.headers(self.context)
end
# Add a row_model to the
# @param [] source_model the source model of the export row model
# @param [Hash] context the extra context given to the instance of the row model
# @return [CsvRowModel::Export] the row model appended
def append_model(source_model, context={})
row_model = export_model_class.new(source_model, context.reverse_merge(self.context))
row_model.to_rows.each do |row|
csv << row
end
row_model
end
alias_method :<<, :append_model
# @return [Boolean] true, if a csv file is generated
def generated?
!!file
end
# Open a block to generate a file
# @param [Boolean] with_headers adds the header to the file if true
def generate(with_headers: true)
@file = Tempfile.new([export_model_class.name, ".csv"])
CSV.open(file.path, "wb") do |csv|
@csv = csv
export_model_class.setup(csv, context, with_headers: with_headers)
yield Proxy.new(self)
end
ensure
@csv = nil
end
def to_s
file.read
end
class Proxy
def initialize(file)
@file = file
end
def append_model(*args)
@file.append_model(*args)
end
alias_method :<<, :append_model
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems