From c7221ef377135142f5af0683d13cd18f25e54337 Mon Sep 17 00:00:00 2001
From: nilsding <nilsding@nilsding.org>
Date: Sun, 30 Nov 2014 19:47:39 +0100
Subject: [PATCH] justask;recount now recounts the smile_counts too.

---
 Rakefile | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/Rakefile b/Rakefile
index 1afced0b..b52b8d7e 100644
--- a/Rakefile
+++ b/Rakefile
@@ -6,24 +6,37 @@ require File.expand_path('../config/application', __FILE__)
 Rails.application.load_tasks
 
 namespace :justask do
-  desc "Recount every user's answer/question count."
+  desc "Recount everything!"
   task recount: :environment do
+    format = '%t (%c/%C) [%b>%i] %e'
     total = User.count
-    progress = ProgressBar.create title: 'Processing user', format: "%t (%c/%C) [%b>%i] %e", starting_at: 0, total: total
+    progress = ProgressBar.create title: 'Processing users', format: format, starting_at: 0, total: total
     User.all.each do |user|
       begin
         answered = Answer.where(user: user).count
         asked = Question.where(user: user).where(author_is_anonymous: false).count
         commented = Comment.where(user: user).count
+        smiled = Smile.where(user: user).count
         user.friend_count = user.friends.count
         user.follower_count = user.followers.count
         user.answered_count = answered
         user.asked_count = asked
         user.commented_count = commented
+        user.smiled_count = smiled
         user.save!
       end
       progress.increment
     end
+    total = Answer.count
+    progress = ProgressBar.create title: 'Processing answers', format: format, starting_at: 0, total: total
+    Answer.all.each do |answer|
+      begin
+        smiles = Smile.where(answer: answer).count
+        answer.smile_count = smiles
+        answer.save!
+      end
+      progress.increment
+    end
   end
 
   desc "Gives admin status to an user."