Class: RubyCharts::Loaders::YAMLLoader

Inherits:
Object
  • Object
show all
Defined in:
lib/ruby_charts/loaders/yaml_loader.rb

Class Method Summary collapse

Class Method Details

.load(filepath, options = {}) ⇒ Object



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# File 'lib/ruby_charts/loaders/yaml_loader.rb', line 6

def self.load(filepath, options = {})
  raise "File not found: #{filepath}" unless File.exist?(filepath)
  
  data = YAML.safe_load_file(filepath)
  
  if data.is_a?(Hash) && data['data']
    data = data['data']
  end
  
  labels = []
  values = []
  
  if data.is_a?(Array)
    data.each do |item|
      if item.is_a?(Hash)
        labels << item['name'] || item['label']
        values << (item['value'] || item['amount']).to_f
      end
    end
  elsif data.is_a?(Hash)
    data.each do |key, value|
      labels << key
      values << value.to_f
    end
  end
  
  {
    labels: labels,
    values: values,
    type: options[:type] || :bar
  }
end