From f085770ed415dc0f89bb042f6f6a08faefa4897d Mon Sep 17 00:00:00 2001
From: Neil Lalonde <neillalonde@gmail.com>
Date: Thu, 5 Jun 2014 15:30:29 -0400
Subject: [PATCH] Import scripts: add update_feature_topic_users

---
 script/import_scripts/base.rb   | 16 ++++++++++++++++
 script/import_scripts/drupal.rb |  2 --
 script/import_scripts/kunena.rb |  2 --
 3 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/script/import_scripts/base.rb b/script/import_scripts/base.rb
index 79938957c46..a84c0ef17f2 100644
--- a/script/import_scripts/base.rb
+++ b/script/import_scripts/base.rb
@@ -43,6 +43,9 @@ class ImportScripts::Base
     execute
 
     update_bumped_at
+    update_feature_topic_users
+
+    puts '', 'Done'
 
   ensure
     RateLimiter.enable
@@ -261,6 +264,19 @@ class ImportScripts::Base
     Post.exec_sql("update topics t set bumped_at = (select max(created_at) from posts where topic_id = t.id and post_type != #{Post.types[:moderator_action]})")
   end
 
+  def update_feature_topic_users
+    puts '', "updating featured topic users"
+
+    total_count = Topic.count
+    progress_count = 0
+
+    Topic.find_each do |topic|
+      topic.feature_topic_users
+      progress_count += 1
+      print_status(progress_count, total_count)
+    end
+  end
+
   def print_status(current, max)
     print "\r%9d / %d (%5.1f%%)    " % [current, max, ((current.to_f / max.to_f) * 100).round(1)]
   end
diff --git a/script/import_scripts/drupal.rb b/script/import_scripts/drupal.rb
index 57538b6da03..967bd8cf704 100644
--- a/script/import_scripts/drupal.rb
+++ b/script/import_scripts/drupal.rb
@@ -38,8 +38,6 @@ class ImportScripts::Drupal < ImportScripts::Base
       puts '', "Failed to create admin user"
       puts e.message
     end
-
-    puts '', 'Done'
   end
 
   def create_blog_topics
diff --git a/script/import_scripts/kunena.rb b/script/import_scripts/kunena.rb
index 75f569434b3..6d562054a2f 100644
--- a/script/import_scripts/kunena.rb
+++ b/script/import_scripts/kunena.rb
@@ -63,8 +63,6 @@ class ImportScripts::Kunena < ImportScripts::Base
       puts '', "Failed to create admin user"
       puts e.message
     end
-
-    puts '', 'Done'
   end
 
   def check_files_exist