From 5a769c7c48d73c93b47e2800a9aa12184b083aea Mon Sep 17 00:00:00 2001
From: Gerhard Schlager <mail@gerhard-schlager.at>
Date: Sun, 31 Aug 2014 18:09:21 +0200
Subject: [PATCH] FIX: Reset changed site settings after import is complete

After an import has finished the site settings should be reset to their
previous values. For example, since 'disable_emails' was set to false
before any import, it wasn't possible to login after the import
(sending email for password reset didn't work).
---
 script/import_scripts/base.rb | 41 +++++++++++++++++++++++++----------
 1 file changed, 30 insertions(+), 11 deletions(-)

diff --git a/script/import_scripts/base.rb b/script/import_scripts/base.rb
index ab0e89ac3..f15e96fc6 100644
--- a/script/import_scripts/base.rb
+++ b/script/import_scripts/base.rb
@@ -27,6 +27,7 @@ class ImportScripts::Base
     @categories_lookup = {}
     @existing_posts = {}
     @topic_lookup = {}
+    @old_site_settings = {}
 
     puts "loading existing groups..."
     GroupCustomField.where(name: 'import_id').pluck(:group_id, :value).each do |group_id, import_id|
@@ -66,17 +67,7 @@ class ImportScripts::Base
   def perform
     Rails.logger.level = 3 # :error, so that we don't create log files that are many GB
 
-    SiteSetting.email_domains_blacklist = ''
-    SiteSetting.min_topic_title_length = 1
-    SiteSetting.min_post_length = 1
-    SiteSetting.min_private_message_post_length = 1
-    SiteSetting.min_private_message_title_length = 1
-    SiteSetting.allow_duplicate_topic_titles = true
-    SiteSetting.default_digest_email_frequency = ''
-    SiteSetting.disable_emails = true
-
-    RateLimiter.disable
-
+    change_site_settings
     execute
 
     puts ""
@@ -89,6 +80,34 @@ class ImportScripts::Base
     puts "", "Done"
 
   ensure
+    reset_site_settings
+  end
+
+  def change_site_settings
+    new_settings = {
+      email_domains_blacklist: '',
+      min_topic_title_length: 1,
+      min_post_length: 1,
+      min_private_message_post_length: 1,
+      min_private_message_title_length: 1,
+      allow_duplicate_topic_titles: true,
+      default_digest_email_frequency: '',
+      disable_emails: true
+    }
+
+    new_settings.each do |key, value|
+      @old_site_settings[key] = SiteSetting.send(key)
+      SiteSetting.set(key, value)
+    end
+
+    RateLimiter.disable
+  end
+
+  def reset_site_settings
+    @old_site_settings.each do |key, value|
+      SiteSetting.set(key, value)
+    end
+
     RateLimiter.enable
   end