suttonelections/scripts/import-results.rb

require_relative '../models'
require 'pp'
# $ ruby scripts/import-results.rb [ELECTION ID] [CSV FILENAME]

# Import a CSV file of election results for a specified election
# The election for which you're importing data must already exist in the elections table

# Run setpositions.rb after this importer to set candidacies.position and candidacies.elected

@election = Election.get(ARGV.shift)

CSV.foreach(ARGV.shift) do |line|
  district_name, candidate_name, party, votes, seats = line
  
  pp line
  
  # District
  @district = District.first(:name => district_name)

  puts @district.name

  # Candidate
  # Assumes that the candidate name is written "forename surname"
  bits = candidate_name.split(" ")
  candidate_forenames = bits[0..-2].join(" ")
  candidate_surname = bits.last

  # Assumes that the candidate name is written "surname, forename(s)"
  # bits = candidate_name.split(", ")
  # candidate_forenames = bits[1]
  # candidate_surname = bits[0]
  
  @candidate = Candidate.first_or_create(:forenames => candidate_forenames, :surname => candidate_surname)

  pp @candidate

  unless @candidate.saved?
    $stderr.puts "Couldn't save candidate #{@candidate}"
    exit 1
  end
  
  
  # Party
  @party = Party.first_or_create(:name => party)
  
  puts @party.name
  unless @party.saved?
    $stderr.puts "Couldn't save party #{@party}"
    exit 1
  end
  
  # Candidacy
  @candidacy = Candidacy.create(
    :election =>  @election,
    :candidate => @candidate,
    :party =>     @party,
    :district =>  @district,
    :votes =>     votes,
    :seats =>     seats
  )

  unless @candidacy.saved?
    $stderr.puts "Couldn't save candidacy #{@candidacy}"
    exit 1
  end
  
end