Module: ClassyEnum::Attributes
- Defined in:
- lib/classy_enum/attributes.rb
Instance Method Summary collapse
-
#classy_enum_attr(*args) ⇒ Object
Class macro used to associate an enum with an attribute on an ActiveRecord model.
Instance Method Details
#classy_enum_attr(*args) ⇒ Object
Class macro used to associate an enum with an attribute on an ActiveRecord model. This method is automatically added to all ActiveRecord models when the classy_enum gem is installed. Accepts an argument for the enum class to be associated with the model. If the enum class name is different than the field name, then an optional field name can be passed. ActiveRecord validation is automatically added to ensure that a value is one of its pre-defined enum members.
Example
# Associate an enum Priority with Alarm model's priority attribute
class Alarm < ActiveRecord::Base
classy_enum_attr :priority
end
# Associate an enum Priority with Alarm model's alarm_priority attribute
class Alarm < ActiveRecord::Base
classy_enum_attr :alarm_priority, :enum => :priority
end
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 62 63 64 65 66 67 68 69 |
# File 'lib/classy_enum/attributes.rb', line 21 def classy_enum_attr(*args) = args. attribute = args[0] enum = [:enum] || attribute allow_blank = [:allow_blank] || false allow_nil = [:allow_nil] || false serialize_as_json = [:serialize_as_json] || false reader_method = attribute.to_s if .has_key? :suffix ActiveSupport::Deprecation.warn(':suffix option for classy_enum_attr is deprecated, and will be removed in ClassyEnum 3.0.', caller) reader_method += "_#{[:suffix]}" end klass = enum.to_s.camelize.constantize valid_attributes = reader_method == attribute.to_s ? klass.all : klass.all.map(&:to_s) self.instance_eval do # Add ActiveRecord validation to ensure it won't be saved unless it's an option validates_inclusion_of attribute, :in => valid_attributes, :message => "must be one of #{klass.}", :allow_blank => allow_blank, :allow_nil => allow_nil # Define getter method that returns a ClassyEnum instance define_method reader_method do klass.build(read_attribute(attribute), :owner => self, :serialize_as_json => serialize_as_json) end # Define setter method that accepts either string or symbol for member define_method "#{attribute}=" do |value| value = value.to_s unless value.nil? super(value) end # Store the enum options so it can be later retrieved by Formtastic define_method "#{attribute}_options" do {:enum => enum, :allow_blank => allow_blank} end end end |