Class: SmarterCSV::Writer

Inherits:
Object
  • Object
show all
Defined in:
lib/smarter_csv/writer.rb

Instance Method Summary collapse

Constructor Details

#initialize(file_path, options = {}) ⇒ Writer

Returns a new instance of Writer.



45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/smarter_csv/writer.rb', line 45

def initialize(file_path, options = {})
  @options = options

  @row_sep = options[:row_sep] || $/
  @col_sep = options[:col_sep] || ','
  @quote_char = options[:quote_char] || '"'
  @force_quotes = options[:force_quotes] == true
  @quote_headers = options[:quote_headers] == true
  @disable_auto_quoting = options[:disable_auto_quoting] == true
  @value_converters = options[:value_converters] || {}
  @map_all_keys = @value_converters.has_key?(:_all)
  @mapped_keys = @value_converters.keys - [:_all]
  @header_converter = options[:header_converter]

  @discover_headers = true
  if options.has_key?(:discover_headers)
    @discover_headers = options[:discover_headers] == true
  else
    @discover_headers = !(options.has_key?(:map_headers) || options.has_key?(:headers))
  end

  @headers = []
  @headers = options[:headers] if options.has_key?(:headers)
  @headers = options[:map_headers].keys if options.has_key?(:map_headers) && !options.has_key?(:headers)
  @map_headers = options[:map_headers] || {}

  @output_file = File.open(file_path, 'w+')
  @temp_file = Tempfile.new('tempfile', '/tmp')
  @quote_regex = Regexp.union(@col_sep, @row_sep, @quote_char)
end

Instance Method Details

#<<(data) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/smarter_csv/writer.rb', line 76

def <<(data)
  case data
  when Hash
    process_hash(data)
  when Array
    data.each { |item| self << item }
  when NilClass
    # ignore
  else
    # :nocov:
    raise InvalidInputData, "Invalid data type: #{data.class}. Must be a Hash or an Array."
    # :nocov:
  end
end

#finalize ⇒ Object 



91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
 
# File 'lib/smarter_csv/writer.rb', line 91

def finalize
  mapped_headers = @headers.map { |header| @map_headers[header] || header }

  mapped_headers = @headers.map { |header| @header_converter.call(header) } if @header_converter

  force_quotes = @quote_headers || @force_quotes
  mapped_headers = mapped_headers.map { |x| escape_csv_field(x, force_quotes) }

  @temp_file.rewind
  @output_file.write(mapped_headers.join(@col_sep) + @row_sep) unless mapped_headers.empty?
  @output_file.write(@temp_file.read)
  @output_file.flush
  @output_file.close
  @temp_file.delete
end