lib/schematic/generator/column.rb in schematic-0.4.2 vs lib/schematic/generator/column.rb in schematic-0.4.3

@@ -6,15 +6,16 @@
         def restriction_classes
           @restriction_classes ||= [Restrictions::Length, Restrictions::Enumeration, Restrictions::Numericality, Restrictions::Pattern, Restrictions::Custom]
         end
       end
 
-      def initialize(klass, column, additional_methods = {}, ignored_methods = {})
+      def initialize(klass, column, additional_methods = {}, ignored_methods = [], required_methods = [])
         @klass = klass
         @column = column
         @additional_methods = additional_methods
         @ignored_methods = ignored_methods
+        @required_methods = required_methods
       end
 
       def generate(builder)
         return if skip_generation?
 
@@ -30,9 +31,10 @@
           end
         end
       end
 
       def minimum_occurrences_for_column
+        return "1" if @required_methods.include?(@column.name.to_sym)
         @klass._validators[@column.name.to_sym].each do |column_validation|
           next unless column_validation.is_a?  ActiveModel::Validations::PresenceValidator
           return "1" if column_validation.options[:allow_blank] != true && column_validation.options[:if].nil?
         end
         "0"