Total Complexity | 4 |
Total Lines | 27 |
Duplicated Lines | 0 % |
Changes | 1 | ||
Bugs | 0 | Features | 0 |
1 | require "csv" |
||
5 | class MappedColumnsParser < CsvParser |
||
6 | attr_reader :column_names |
||
7 | |||
8 | def initialize(options = {}) |
||
9 | @column_names = options.fetch(:column_names) |
||
10 | options.delete(:column_names) |
||
11 | super(options) |
||
12 | end |
||
13 | |||
14 | def parse(contents, &block) |
||
15 | mapped = [] |
||
16 | CSV.parse(contents, parser_options) do |row| |
||
17 | new_row = {} |
||
18 | row.each_with_index do |value, index| |
||
19 | new_row[column_name(index)] = value |
||
20 | end |
||
21 | mapped << new_row |
||
22 | end |
||
23 | block_given? ? mapped.each(&block) : mapped |
||
24 | end |
||
25 | |||
26 | private |
||
27 | |||
28 | def column_name(index) |
||
29 | column_names.fetch(index) |
||
30 | end |
||
31 | end |
||
32 | end |
||
34 |