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
 
# 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]

  @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 



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

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 



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

def finalize
  mapped_headers = @headers.map { |header| @map_headers[header] || header }
  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