Regs #6133; Define file's extensions whitelist on a per-class basis6133-new-exports

3 files changed, 9 insertions, 1 deletions

diff --git a/app/models/export/base.rb b/app/models/export/base.rb
index 58a0ec22b..6085e0ffb 100644
--- a/app/models/export/base.rb
+++ b/app/models/export/base.rb
@@ -15,6 +15,10 @@ class Export::Base < ActiveRecord::Base
     I18n.t("export.#{self.name.demodulize.underscore}")
   end
 
+  def self.file_extension_whitelist
+    %w(zip csv json)
+  end
+
   if Rails.env.development?
     def self.force_load_descendants
       path = Rails.root.join 'app/models/export'
diff --git a/app/models/import/base.rb b/app/models/import/base.rb
index 73d06d968..1dd9c4195 100644
--- a/app/models/import/base.rb
+++ b/app/models/import/base.rb
@@ -10,6 +10,10 @@ class Import::Base < ActiveRecord::Base
     "Import::Resource"
   end
 
+  def self.file_extension_whitelist
+    %w(zip)
+  end
+
   include IevInterfaces::Task
 
   def self.model_name
diff --git a/app/uploaders/import_uploader.rb b/app/uploaders/import_uploader.rb
index 6ef52a984..3491768f6 100644
--- a/app/uploaders/import_uploader.rb
+++ b/app/uploaders/import_uploader.rb
@@ -37,7 +37,7 @@ class ImportUploader < CarrierWave::Uploader::Base
   # Add a white list of extensions which are allowed to be uploaded.
   # For images you might use something like this:
   def extension_whitelist
-    %w(zip csv json)
+    model.class.try(:file_extension_whitelist) || %w(zip)
   end
 
   # Override the filename of the uploaded files: