Class: Ibge::Reader::Base

Inherits:
Object
  • Object
show all
Extended by:
Helpers
Defined in:
lib/ibge/reader/base.rb

Constant Summary

Constants included from Helpers

Helpers::SOURCES

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Helpers

columns, source, valid_informed_source

Constructor Details

#initialize(options = {}) ⇒ Base

Returns a new instance of Base.



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/ibge/reader/base.rb', line 14

def initialize(options = {})
  @spreadsheet = {}
  @rows = {}
  @selected_sheet = {}

  self.selected_source = self.class.selected_source

  unless options.fetch(:filename, nil).nil?
    filename = options.fetch(:filename)

    if filename.is_a?(Hash)
      sources_filename = options.fetch(:filename).keys

      sources_filename.each do |source_filename|
        self.class.valid_informed_source(source_filename)
      end
    else
      raise ArgumentError,
            "Filename must be a hash with the source as key"
    end
  end

  default_filename = default_filename_to(selected_source)

  @filename = options.fetch(:filename, "#{DATA_PATH}/#{default_filename}")
end

Instance Attribute Details

#filename ⇒ Object (readonly)

Returns the value of attribute filename.



8
9
10
 
# File 'lib/ibge/reader/base.rb', line 8

def filename
  @filename
end

Class Method Details

.read(options = {}) ⇒ Object 



10
11
12
 
# File 'lib/ibge/reader/base.rb', line 10

def self.read(options = {})
  new(options).read
end

Instance Method Details

#read ⇒ Object 



41
42
43
44
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/ibge/reader/base.rb', line 41

def read
  transformed_rows = []
  previous_column_range = nil

  if multiple_sources?
    loaded_sources = defined_columns.keys.sort

    loaded_sources.each do |loaded_source|
      self.selected_source = loaded_source
      @filename = default_filename_to(loaded_source)

      if transformed_rows == []
        transformed_rows = rows[1..-1].map do |row|
          transform(row.compact.values_at(*column_range), column_range)
        end

        transformed_rows = transformed_rows.uniq

        previous_column_range = column_range
      else
        transformed_rows.each do |transformed_row|
          selected_rows = rows[1..-1].select do |r|
            r[previous_column_range.first] == transformed_row.values.first
          end

          new_row = selected_rows.compact.uniq { |r| r[0] }.map do |row|
            transform(row.compact.values_at(*column_range), column_range)
          end

          if new_row.empty?
            columns = self.class.defined_columns[selected_source]

            columns.each do |column|
              new_row << { column[1] => "" }
            end
          end

          transformed_row.merge!(*new_row)
        end
      end
    end
  else
    transformed_rows = rows[1..-1].map do |row|
      transform(row.compact.values_at(*column_range), column_range)
    end
  end

  transformed_rows.uniq.sort_by { |h| data_sort(h) }
end