active_enum 0.3.0 → 0.4.0

data/README.rdoc

@@ -63,6 +63,12 @@ Define enum class implicitly in enumerate block
     end
   end
 
+Multiple attributes with same enum
+
+  class Patient < ActiveRecord::Base
+    enumerate :to, :from, :with => Sex
+  end
+
 Access the enum values and the enum class using the attribute method with a symbol for the enum component you want
 
   user = User.new

data/lib/active_enum/extensions.rb

@@ -21,25 +21,32 @@ module ActiveEnum
       #     enumerate :sex do
       #       value :id => 1, :name => 'Male'
       #     end
+			#
+			#     # Multiple attributes with same enum
+      #     enumerate :to, :from, :with => Sex
       #
-      def enumerate(attribute, options={}, &block)
-        attribute = attribute.to_sym
-        if block_given?
-          enum_name = "#{self.to_s.underscore}_#{attribute}"
-          ActiveEnum.define { enum(enum_name, &block) }
-          enum = enum_name.classify.constantize
-        else
-          enum = options[:with] || attribute.to_s.classify.constantize
-        end
-
-        self.enumerated_attributes[attribute] = enum
-
-        define_active_enum_read_method(attribute)
-        define_active_enum_write_method(attribute)
-        define_active_enum_question_method(attribute)
-      rescue NameError => e
-        raise e unless e.message =~ /uninitialized constant/
-        raise ActiveEnum::EnumNotFound, "Enum class could not be found for attribute '#{attribute}' in class #{self}. Specify the enum class using the :with option."
+      def enumerate(*attributes, &block)
+				options = attributes.extract_options!
+				attributes.each do |attribute|
+					begin
+						if block_given?
+							enum_name = "#{self.to_s.underscore}_#{attribute}"
+							ActiveEnum.define { enum(enum_name, &block) }
+							enum = enum_name.classify.constantize
+						else
+							enum = options[:with] || attribute.to_s.classify.constantize
+						end
+						attribute = attribute.to_sym
+						self.enumerated_attributes[attribute] = enum
+
+						define_active_enum_read_method(attribute)
+						define_active_enum_write_method(attribute)
+						define_active_enum_question_method(attribute)
+					rescue NameError => e
+						raise e unless e.message =~ /uninitialized constant/
+						raise ActiveEnum::EnumNotFound, "Enum class could not be found for attribute '#{attribute}' in class #{self}. Specify the enum class using the :with option."
+					end
+				end
       end
 
       def enum_for(attribute)

data/lib/active_enum/version.rb

@@ -1,3 +1,3 @@
 module ActiveEnum
-  VERSION = '0.3.0'
+  VERSION = '0.4.0'
 end

data/spec/active_enum/extensions_spec.rb

@@ -5,6 +5,12 @@ class Sex < ActiveEnum::Base
   value :id => 2, :name => 'Female'
 end
 
+class Accepted < ActiveEnum::Base
+  value :id => 0, :name => 'No'
+  value :id => 1, :name => 'Definitely'
+  value :id => 2, :name => 'Maybe'
+end
+
 describe ActiveEnum::Extensions do
   before do
     Person.class_eval do
@@ -21,6 +27,14 @@ describe ActiveEnum::Extensions do
     ActiveRecord::Base.should respond_to(:enum_for)
   end
 
+  it 'should allow multiple attributes to be enumerated with same enum' do
+    Person.class_eval do
+      enumerate :attending, :staying, :with => Accepted
+    end
+    Person.enum_for(:attending).should == Accepted
+    Person.enum_for(:staying).should == Accepted
+  end
+
   it 'should allow implicit enumeration class from attribute name' do
     Person.class_eval do
       enumerate :sex

data/spec/schema.rb

@@ -4,6 +4,8 @@ ActiveRecord::Schema.define(:version => 1) do
     t.string   :first_name
     t.string   :last_name
     t.integer  :sex
+    t.integer  :attending
+    t.integer  :staying
   end
 
   create_table :sexes, :force => true do |t|

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: active_enum
 version: !ruby/object:Gem::Version 
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Adam Meehan
@@ -9,7 +9,7 @@ autorequire: active_enum
 bindir: bin
 cert_chain: []
 
-date: 2009-11-05 00:00:00 +11:00
+date: 2009-11-06 00:00:00 +11:00
 default_executable: 
 dependencies: []