Как пройти все узлы в дереве YAML в Ruby?

Question

Я загружаю произвольный документ YAML и хочу пройтись по каждому узлу в дереве. Я не знаю заранее, как вложено дерево, поэтому я не могу просто использовать простой оператор each для обхода всех узлов.

Вот как я загружаю документ:

tree = File.open( "#{RAILS_ROOT}/config/locales/es.yml" ){ |yf| YAML::load (yf)}

Answer 1

def traverse(obj, &blk)
  case obj
  when Hash
    # Forget keys because I don't know what to do with them
    obj.each {|k,v| traverse(v, &blk) }
  when Array
    obj.each {|v| traverse(v, &blk) }
  else
    blk.call(obj)
  end
end

traverse( YAML.load_file(filename) ) do |node|
  puts node
end

Edit:

Обратите внимание, что это дает только листовые узлы. Вопрос был не очень ясен относительно того, что именно было разыскивается.

Answer 2

Вот как проанализировать древовидную структуру YAML и действовать на каждый узел, ветвь и лист. Он также предоставляет вам родительский элемент для каждого узла, который вам нужен, например, при заполнении древовидной структуры базы данных из файла YAML. Это небольшое дополнение к @ sepp2k отличный ответ:

require 'yaml'

def traverse(obj,parent, &blk)
 case obj
 when Hash
   obj.each do |k,v| 
     blk.call(k,parent)
     # Pass hash key as parent
     traverse(v,k, &blk) 
   end
 when Array
   obj.each {|v| traverse(v, parent, &blk) }
 else
   blk.call(obj,parent)
 end
end


# Example, creating a database tree structure, from your YAML file.
# Passing nil to root node(s) as parent

tree_str =<<EOF
  Regions:
    - Asia
    - Europe
    - Americas
  Other:
    - Foo:
      - bar
      - buz
EOF

traverse( YAML.load(tree_str), nil ) do |node,parent|
  puts "Creating node '#{node}' with parent '#{ parent || 'nil' }'"
end