Pastable version of Ruby Custom Sort

<pre class='prettyprint' language='ruby'>require 'csv'

def sort_csv(filename, fields = [&quot;ID&quot;], reverse = false)
  csv_data = load_csv_data(filename)

sort_fields = get_sort_indexes(fields, csv_data['headers'])
  sorted = custom_sort(csv_data['headers'], csv_data['rows'], sort_fields)
  sorted.reverse! if reverse
  
  write_csv_file(filename, sorted, csv_data['headers'])
end

def custom_sort(csv_headers, csv_data, sort_indexes)
  eyecolor_sort = [&quot;Red&quot;,&quot;Blue&quot;,&quot;Turquoise&quot;,&quot;Brown&quot;,&quot;Green&quot;]
  hair_colors = csv_data.map {|c| c[csv_headers.index(&quot;HAIRCOLOR&quot;)]}.sort.uniq
  if hair_colors.include?('Brown') and hair_colors.include?('Black')
    brown = hair_colors.delete('Brown')
    hair_colors.insert(hair_colors.index(&quot;Black&quot;) + 1, brown)
  end
  
  sorted = csv_data.sort do |a,b|
    indexes = sort_indexes.dup
    
    until indexes.empty? do
      idx = indexes.shift
      if (a[idx] != b[idx]) or indexes.empty?
        if csv_headers[idx] == &quot;EYECOLOR&quot;
          # do eyecolor sort
          comparison = &quot;eyecolor_sort.index(a[idx]) &lt;=&gt; eyecolor_sort.index(b[idx])&quot;
        elsif csv_headers[idx] == &quot;HAIRCOLOR&quot;
          # do hair sort
          comparison = &quot;hair_colors.index(a[idx]) &lt;=&gt; hair_colors.index(b[idx])&quot;
        else 
          # do alpha sort
          comparison = &quot;a[idx] &lt;=&gt; b[idx]&quot;
        end
      end
      break if comparison
    end
    eval(comparison)
  end
  
  sorted
end

def load_csv_data(file)
  data = {}
  csv = CSV.open(file, 'r')
  data['headers'] = csv.shift
  data['rows'] = []
  csv.each do |row|
    data['rows'] &lt;&lt; row
  end
  csv.close
  data
end

def write_csv_file(file, data, headers)
  output = File.open(&quot;#{file}.sorted&quot;, 'wb')
  CSV::Writer.generate(output) do |csv|
    csv &lt;&lt; headers
    data.each do |d|
      csv &lt;&lt; d
    end
  end
  output.close
end

def get_sort_indexes(fields, data)
  idx = []
  fields.each do |f|
    idx &lt;&lt; data.index(f)
  end
  idx
end</pre>
<a href="http://www.refactormycode.com/codes/418-ruby-custom-search" style="color:#fff" title="As seen on RefactorMyCode.com"><img alt="Small_logo" src="http://www.refactormycode.com/images/small_logo.gif" style="border:0" /></a>

Refactor :my => 'code'

Recent

Popular

Pastable version of Ruby Custom Sort