TSV_to_YAML.rb

TSV_to_YAML.rb

#!/usr/bin/env ruby
require 'yaml'

# Only runs if and argument is provided at the command prompt
if ARGV[0]
  file = File.open(ARGV[0], 'r')
  linelist = []
  keyarr = []
  valuearr = []
  flag = false
  temphash = {}

  file.each_line do |line|
    # This reads the hash keys and stores them in a separate array.
    #  This block of code runs only once
    if flag == false
      keyarr = line.gsub("\n", '').split("\t")
      flag = true
    else
      valuearr = line.gsub("\n", '').split("\t")
      temphash = Hash[keyarr.zip valuearr]
      linelist << temphash
    end
  end
  file.close

  if ARGV[1]
    File.open(ARGV[1], 'w') do |files|
      files.puts linelist.to_yaml
    end
  else
    puts linelist.to_yaml
  end

# Error output message if commandline argument is not provided when code is run.
else
  puts 'Please enter input .tsv file'
end