Shorten migration roll-back also

lindagcaba · Apr 20, 2011 · e63549c · e63549c
1 parent c0b4626
commit e63549c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/paperclip/railtie.rb b/lib/paperclip/railtie.rb
@@ -20,6 +20,7 @@ class Railtie
     def self.insert
       ActiveRecord::Base.send(:include, Paperclip::Glue)
       File.send(:include, Paperclip::Upfile)
+      ActiveRecord::ConnectionAdapters::AbstractAdapter.send(:include, Paperclip::Schema)
       ActiveRecord::ConnectionAdapters::Table.send(:include, Paperclip::Schema)
       ActiveRecord::ConnectionAdapters::TableDefinition.send(:include, Paperclip::Schema)
     end

diff --git a/lib/paperclip/schema.rb b/lib/paperclip/schema.rb
@@ -6,5 +6,12 @@ def has_attached_file(name)
       column :"#{name}_file_size",    :integer
       column :"#{name}_updated_at",   :datetime
     end
+
+    def drop_attached_file(table_name, name)
+      remove_column table_name, :"#{name}_file_name"
+      remove_column table_name, :"#{name}_content_type"
+      remove_column table_name, :"#{name}_file_size"
+      remove_column table_name, :"#{name}_updated_at"
+    end
   end
 end