operator-csv-to-changeset.rb

###
# Takes in a CSV file with headers:
#   - operator_onestop_id
#   - country_iso
#   - state_iso
# and creates a changeset to set new country and state values.
# If country_iso or state_iso is "null", it sets value to an empty string.
###

require 'csv'
require 'json'

file_name = ARGV[0]

change_payloads = []
CSV.foreach(file_name, headers: true) do |row|
  country_code = (row['country_iso'] == 'null' ? "" : row['country_iso'])
  state_code = (row['state_iso'] == 'null' ? "" : row['state_iso'])

  change_payload = {
    action: "createUpdate",
    operator: {
      onestopId: row['operator_onestop_id'],
      country: country_code,
      state: state_code
    }
  }
  change_payloads << change_payload
end

changeset_json = {
  changes: change_payloads
}

puts JSON.pretty_generate(changeset_json)