From ac8d1eb4bbadda6f4b29bf8c5f7dea0b2d308bda Mon Sep 17 00:00:00 2001
From: Adrian Short <adrian.short@gmail.com>
Date: Thu, 5 Dec 2013 15:37:08 +0000
Subject: [PATCH] Pass election date as command line option; prevent running on
 byelections; tidy output

---
 scripts/setpositions.rb | 49 +++++++++++++++++++++++++++++------------
 1 file changed, 35 insertions(+), 14 deletions(-)

diff --git a/scripts/setpositions.rb b/scripts/setpositions.rb
index dc91c3a..10033c1 100644
--- a/scripts/setpositions.rb
+++ b/scripts/setpositions.rb
@@ -2,20 +2,41 @@ require_relative '../models'
 require 'pp'
 
 # Set position and elected for each candidacy
+# WARNING - This will only work in full council elections not byelections (as the number of seats being elected in that eleciton is probably less than the total number of seats in the ward)
+# This will break all the byelection data currently in the database
 
-Election.all.each do |election|
-  election.body.districts.each do |district|
-    cands = Candidacy.all(:conditions => { :district_id => district.id, :election_id => election.id }, :order => [:votes.desc])
-    pp cands
-    puts
-    pos = 1
-    cands.each do |cand|
-      pos <= district.seats ? seats = 1 : seats = 0
-      print cand.candidate.surname, ' ', cand.votes, ' ', pos, ' ', seats, "\n"
-      cand.position = pos
-      cand.seats = seats
-      cand.save
-      pos += 1
+election = Election.first(:d => ARGV.shift)
+
+unless election
+  puts "Election not found. Usage example: $ %s 2006-05-04" % __FILE__
+  exit 1
+end
+
+puts "Setting candidacy positions for %s %s %s" % [ election.body.name, election.kind, election.d.to_s ]
+if election.candidacies.size < 30
+  puts "This script only works for full council elections. Your election looks like a byelection, so quitting."
+  exit 1
+end
+puts "%d candidacies in this election" % election.candidacies.size
+
+election.body.districts.each do |district|
+  cands = Candidacy.all(:conditions => { :district_id => district.id, :election_id => election.id }, :order => [:votes.desc])
+  # pp cands
+  puts
+  pos = 1
+  cands.each do |cand|
+    if pos == 1
+      puts "-" * 62
+      puts district.name
+      puts "-" * 62
     end
+
+    # If this candidate's position was in the top district.seats positions, they won a seat
+    pos <= district.seats ? seats = 1 : seats = 0
+    puts "%-25s %-25s %5d %2d %d" % [ cand.candidate.surname, cand.candidate.forenames, cand.votes, pos, seats ]
+    cand.position = pos
+    cand.seats = seats
+    cand.save
+    pos += 1
   end
-end
\ No newline at end of file
+end