Contents

class Quandl::Operation::QDFormat::Load
  
  class << self
  
    def from_file(path)
      from_string(File.read(path).strip)
    end
  
    def from_string(input)
      nodes = [{ attributes: '', data: '' }]
      input.each_line do |line|
        # strip whitespace
        line = line.strip.rstrip
        # ignore comments and blank lines
        next if line[0] == '#' || line.blank?
        # code_format denotes the start of a new node
        nodes << { attributes: '', data: '' } if line[0..2] == node_delimiter
        # attribute_format denotes an attribute
        if line =~ attribute_format
          # add the attribute to attributes
          nodes[-1][:attributes] += "#{line}\n"
          # otherwise it must be data
        else
          nodes[-1][:data] += "#{line}\n"
        end
      end
      # append the current node
      nodes = parse_nodes(nodes)
      nodes = initialize_nodes(nodes)
      nodes
    end
    
    
    protected
    
    def parse_nodes(nodes)
      nodes.collect do |node|
        # parse attrs as yaml
        node[:attributes] = YAML.load( node[:attributes] )
        # replace new line characters
        node[:attributes].each do |key, value|
          node[:attributes][key].gsub!('\n', "\n") if node[:attributes][key].respond_to?(:gsub!)
        end
        # parse data as csv
        node[:attributes][:data] = CSV.parse(node[:data])
        node
      end
    end
    
    def initialize_nodes(nodes)
      nodes.collect do |node|
        Quandl::Operation::QDFormat::Node.new(node[:attributes])
      end
    end
    
    def node_delimiter
      '---'
    end
  
    def attribute_format
      /^([a-z0-9_]+): (.+)/
    end
  
  end

end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
quandl_operation-0.1.21	lib/quandl/operation/qdformat/load.rb