IMPORT: create category definitions in `import:ensure_consistency` task

IMPORT: create category definitions in import:ensure_consistency task

diff --git a/lib/tasks/import.rake b/lib/tasks/import.rake
index b313014..936232e 100644
--- a/lib/tasks/import.rake
+++ b/lib/tasks/import.rake
@@ -20,6 +20,7 @@ task "import:ensure_consistency" => :environment do
   update_users
   update_groups
   update_tag_stats
+  create_category_definitions
 
   log "Done!"
 end
@@ -425,6 +426,11 @@ def update_tag_stats
   Tag.ensure_consistency!
 end
 
+def create_category_definitions
+  log "Creating category definitions"
+  Category.where(topic_id: nil).each(&:create_category_definition)
+end
+
 def log(message)
   puts "[#{DateTime.now.strftime("%Y-%m-%d %H:%M:%S")}] #{message}"
 end
diff --git a/script/bulk_import/base.rb b/script/bulk_import/base.rb
index 89cc948..3212702 100644
--- a/script/bulk_import/base.rb
+++ b/script/bulk_import/base.rb
@@ -303,14 +303,6 @@ class BulkImport::Base
     end
   end
 
-  def create_categories(rows, &block)
-    create_records(rows, "category", CATEGORY_COLUMNS, &block)
-
-    Category.where(topic_id: nil).each do |c|
-      c.create_category_definition
-    end
-  end
-
   def create_user_emails(rows, &block)
     create_records(rows, "user_email", USER_EMAIL_COLUMNS, &block)
   end
@@ -323,6 +315,9 @@ class BulkImport::Base
   def create_group_users(rows, &block)
     create_records(rows, "group_user", GROUP_USER_COLUMNS, &block)
   end
+  def create_categories(rows, &block)
+    create_records(rows, "category", CATEGORY_COLUMNS, &block)
+  end
   def create_topics(rows, &block)
     create_records(rows, "topic", TOPIC_COLUMNS, &block)
   end

GitHub sha: a20f5855

1 Like