У меня есть рабочий сценарий, который прекрасно преобразует файлы JSON в файлы CSV, однако я пытаюсь отредактировать сценарий, чтобы внести некоторые изменения в файл CSV перед сохранением, но в настоящее время безуспешно.
Вот мой текущий скрипт конвертации:
require 'csv'
require 'json'
require 'set'
def get_recursive_keys(hash, nested_key=nil)
hash.each_with_object([]) do |(k,v),keys|
k = "#{nested_key}.#{k}" unless nested_key.nil?
if v.is_a? Hash
keys.concat(get_recursive_keys(v, k))
else
keys << k
end
end
end
json = JSON.parse(File.open(ARGV[0]).read)
headings = Set.new
json.each do |hash|
headings.merge(get_recursive_keys(hash))
end
headings = headings.to_a
CSV.open(ARGV[0] + '.csv', 'w') do |csv|
csv << headings
json.each do |hash|
row = headings.map do |h|
v = hash.dig(*h.split('.'))
v.is_a?(Array) ? v.join(',') : v
end
csv << row
end
end
Который я запускаю с помощью этой команды:
for file in directory/*; do ruby json-to-csv.rb "$file"; done
Как я могу отредактировать этот скрипт, чтобы:
До сих пор все, что я пробовал, полностью ломало сценарий — с чего лучше всего начать вносить эти изменения?
@knh190 knh190 Я в основном работал с заголовками, но мой фильтр каждый раз разбивает массив. Я попытался использовать оператор switch для «h», чтобы отфильтровать столбцы, которые мне не нужны после «headings.map do |h|» - было бы лучше сосредоточиться на ключах json перед заголовками?





Вот код, который работает:
require 'csv'
require 'json'
require 'set'
def get_recursive_keys(hash, nested_key=nil)
hash.each_with_object([]) do |(k,v),keys|
# Col filter
next if ["score", "original_name"].include? k
k = "#{nested_key}.#{k}" unless nested_key.nil?
if v.is_a? Hash
keys.concat(get_recursive_keys(v, k))
else
keys << k
end
end
end
json = JSON.parse(File.open(ARGV[0]).read)
headings = Set.new
headings = get_recursive_keys(json)
headings = headings.to_a
# Header sorting
headings = headings.sort { |a, b| a <=> b }
CSV.open(ARGV[0] + '.csv', 'w') do |csv|
csv << headings
row = headings.map do |h|
v = (h.split('.').length > 1) ? json.dig(*h.split('.')) : h
v.is_a?(Array) ? v.join(',') : v
end
csv << row
end
Я протестировал эту небольшую строку json: {"score": "12", "name": "Obi", "original_name": "Wan Kenobi", "something": {"sub_key": "Wuhu"} }
Это здорово! Работает отлично, я также настроил это, чтобы теперь включать только указанные заголовки, а не удалять определенные заголовки.
Отфильтруйте ключи json, чтобы удалить определенные столбцы. И используйте отсортированный массив, чтобы сохранить порядок вывода. Что вы пробовали?