Module: Percent::ActiveRecord::Percentizable::ClassMethods

Defined in:
lib/percent/active_record/percentizable.rb

Instance Method Summary collapse

Instance Method Details

#percentize(*columns) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/percent/active_record/percentizable.rb', line 7

def percentize(*columns)
  # initialize options
  options = columns.extract_options!

  columns.each do |column|
    column_name = column.to_s

    # determine attribute name
    if options[:as]
      attribute_name = options[:as]
    elsif column_name =~ /_fraction$/
      attribute_name = column_name.sub /_fraction$/, ""
    else
      attribute_name = column_name + '_percent'
    end

    # validation
    unless options[:disable_validation]
      allow_nil = options.fetch :allow_nil, false
      numericality = options[:numericality]
      fraction = options[:fraction_numericality]

      if numericality || !options.key?(:numericality)
        numericality = numericality_range 0, 100 unless fraction.is_a? Hash

        validates attribute_name, {
          allow_nil: allow_nil,
          'percent/active_model/percentage': numericality
        }
      end

      if fraction || !options.key?(:fraction_numericality)
        unless numericality.is_a?(Hash) && options.key?(:numericality)
          fraction = numericality_range 0, 1
        end

        validates column_name, allow_nil: allow_nil, numericality: fraction
      end
    end

    # percent attribute getter
    define_method attribute_name do
      value = public_send column_name
      value.nil? ? value : Percentage.from_fraction(value, options)
    end

    # percent attribute setter
    define_method "#{attribute_name}=" do |value|
      unless value.nil? || value.is_a?(Percentage)
        value = Percentage.from_amount value, options
      end
      public_send "#{column_name}=", (value.nil? ? value : value.to_d)
    end
  end
end