Class: Comma::Generator

Inherits:
Object show all
Defined in:
lib/comma/generator.rb

Instance Method Summary collapse

Constructor Details

#initialize(instance, style) ⇒ Generator

Returns a new instance of Generator.



5
6
7
8
9
10
11
# File 'lib/comma/generator.rb', line 5

def initialize(instance, style)
  @instance = instance
  parsed = Comma::Options.parse(style)
  @style = parsed[:style]
  @filename = parsed[:filename]
  @options = parsed[:csv]
end

Instance Method Details

#run(iterator_method) ⇒ Object



13
14
15
16
17
18
19
# File 'lib/comma/generator.rb', line 13

def run(iterator_method)
  if @filename
    CSV_HANDLER.open(@filename, 'w', **@options) { |csv| append_csv(csv, iterator_method) } && (return true)
  else
    CSV_HANDLER.generate(**@options) { |csv| append_csv(csv, iterator_method) }
  end
end