From 4b533e9e3c861a98c9053d5f25e83e86851ac217 Mon Sep 17 00:00:00 2001
From: Adrian Short <adrian.short@gmail.com>
Date: Fri, 2 May 2014 13:54:10 +0100
Subject: [PATCH] Add candidate importer

---
 scripts/import-candidates-csv.rb | 65 ++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 scripts/import-candidates-csv.rb

diff --git a/scripts/import-candidates-csv.rb b/scripts/import-candidates-csv.rb
new file mode 100644
index 0000000..0c4561c
--- /dev/null
+++ b/scripts/import-candidates-csv.rb
@@ -0,0 +1,65 @@
+require_relative '../models'
+require 'csv'
+require 'pp'
+# $ ruby scripts/import-candidates-csv.rb [ELECTION ID] [CSV FILENAME]
+
+# Import a CSV file of election candidates for a specified election
+# The election for which you're importing data must already exist in the elections table
+
+@election = Election.get(ARGV.shift)
+pp @election
+
+candidates_matched = 0
+
+CSV.foreach(ARGV.shift) do |line|
+  # pp line
+
+  district_name, candidate_surname, candidate_forenames, party = line
+
+  # District
+  @district = District.first(:name => district_name)
+
+  # pp @district
+
+  # Candidate
+  @candidate = Candidate.first_or_create(:forenames => candidate_forenames, :surname => candidate_surname)
+  
+  if @candidate
+    pp @candidate
+    candidates_matched += 1
+  end
+
+  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 =>     nil
+  )
+
+  unless @candidacy.saved?
+    $stderr.puts "Couldn't save candidacy #{@candidacy}"
+    exit 1
+  end
+  
+end
+
+puts "Candidates: #{Candidate.count}"
+puts "Candidacies: #{Candidacy.count}"
+puts candidates_matched
+