citrusbyte-milton 0.1.9 → 0.2.0

data/lib/milton/attachment.rb

@@ -3,7 +3,7 @@ require 'fileutils'
 
 module Citrusbyte
   module Milton
-    module Attachment
+    module Attachment      
       def self.included(base)
         base.class_inheritable_accessor :milton_options
         base.milton_options = {}
@@ -11,12 +11,16 @@ module Citrusbyte
       end
       
       module AttachmentMethods
-        def has_attachment_methods(options={})
+        def require_column(column, message)
           begin
-            raise "Milton requires a filename column on #{class_name} table" unless column_names.include?("filename")
+            raise message unless column_names.include?(column)
           rescue ActiveRecord::StatementInvalid => i
             # table doesn't exist yet, i.e. hasn't been migrated in...
           end
+        end
+        
+        def has_attachment_methods(options={})
+          require_column 'filename', "Milton requires a filename column on #{class_name} table"
           
           # character used to seperate a filename from its derivative options, this
           # character will be stripped from all incoming filenames and replaced by

data/lib/milton/is_resizeable.rb

@@ -7,9 +7,9 @@ module Citrusbyte
 
       module IsMethods
         def is_resizeable(options={})
-          raise "is_resizeable requires a content_type column on #{class_name} table" unless column_names.include?("content_type")
-          
           ensure_attachment_methods options
+          
+          require_column 'content_type', "Milton's is_resizeable requires a content_type column on #{class_name} table"
 
           self.milton_options.merge!(options)
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: citrusbyte-milton
 version: !ruby/object:Gem::Version 
-  version: 0.1.9
+  version: 0.2.0
 platform: ruby
 authors: 
 - Ben Alavi