ruby CSV到JSON ruby​​脚本

Posted 2021-05-16

require 'csv'
require 'json'
require 'open-uri'


if ARGV[0].nil?
  raise "wrong number of arguments (0 for 1..2)"
end

file = open(ARGV[0])
file.set_encoding("UTF-8")

header = nil
rows = []

CSV.parse(file.read).each do |line|
  if header.nil?
    header = line.map(&:strip)
    next
  end

  rows << Hash[header.zip(line.map(&:strip))]
end

File.open(ARGV[1] || 'to-json.js', 'w') {|f| f << JSON.pretty_generate(rows) }