nilify_blanks.rb in nilify_blanks-1.3.0

- old
+ new

@@ -2,11 +2,11 @@
   def self.included(base)
     base.extend ClassMethods
   end
 
   module ClassMethods
-    DEFAULT_TYPES = [:string, :text]
+    DEFAULT_TYPES = [:string, :text, :citext]
 
     @@define_nilify_blank_methods_lock = Mutex.new
 
     # This overrides the underlying rails method that defines attribute methods.
     # This must be thread safe, just like the underlying method.
@@ -62,11 +62,13 @@
 
         cattr_accessor :nilify_blanks_columns
 
         if options[:only]
           self.nilify_blanks_columns = options[:only].clone
+        elsif options[:nullables_only] == false
+          self.nilify_blanks_columns = self.columns.select {|c| options[:types].include?(c.type) }.map(&:name).map(&:to_s)
         else
-          self.nilify_blanks_columns = self.content_columns.select(&:null).select {|c| options[:types].include?(c.type) }.map(&:name).map(&:to_s)
+          self.nilify_blanks_columns = self.columns.select(&:null).select {|c| options[:types].include?(c.type) }.map(&:name).map(&:to_s)
         end
 
         self.nilify_blanks_columns -= options[:except] if options[:except]
         self.nilify_blanks_columns = self.nilify_blanks_columns.map(&:to_s)
 