Fixed minor bug in the scruffy code.

[selectricity-live] / app / controllers / site_controller.rb

diff --git a/app/controllers/site_controller.rb b/app/controllers/site_controller.rb
index dd049f7311d2bf0d95e03b7212bacc2b3565f12e..a33a1fa16707403589ce4caebd0bb36161bb30d9 100644 (file)
--- a/app/controllers/site_controller.rb
+++ b/app/controllers/site_controller.rb
@@ -1,23 +1,25 @@
 class SiteController < ApplicationController
 class SiteController < ApplicationController

-  layout 'hc'
+  layout 'main'

   model :user, :election, :account
 
   def index
   model :user, :election, :account
 
   def index

-    @quickvotes = QuickVote.find_all(["quickvote = 1"]).sort {|a,b| b.enddate <=> a.enddate}[0..1]
+    @quickvotes = QuickVote.find(:all).sort {|a,b| b.enddate <=> a.enddate}[0..1]

     # if the person claims they are logged in
     # if the person claims they are logged in

-    if session[:user]
-
+   
+    if logged_in?

       # check to see that we actually have record of them
       # check to see that we actually have record of them

-      if User.find_all(["id = ?", session[:user].id]).length == 1
+      if User.exists?(session[:user])
+        username = User.find(session[:user]).login

         # if we have record of them, grab the list of their elections
         session[:user] = User.find(session[:user])
         @current_elections = session[:user].elections.sort do |a,b|
           b.enddate <=> a.enddate
         end
       else
         # if we have record of them, grab the list of their elections
         session[:user] = User.find(session[:user])
         @current_elections = session[:user].elections.sort do |a,b|
           b.enddate <=> a.enddate
         end
       else

-        # if we have no record of them, set the session id back to
-       # nothing and start again
+        # if we have no record of them, set the user back to
+             # nothing and start again

         session[:user] = nil
         session[:user] = nil

+        

       end
     end
   end
       end
     end
   end