Override attributes method using anon module

shioyama · Jul 16, 2017 · d8fe42f · d8fe42f
1 parent 049a5f9
commit d8fe42f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/mobility/active_record/attribute_methods.rb b/lib/mobility/active_record/attribute_methods.rb
@@ -13,13 +13,15 @@ module AttributeMethods
       # @return [Array<String>] Model attributes
       # @!method attributes
       def self.included(model)
-        model.class_eval do
-          alias_method :untranslated_attributes, :attributes
-
+        attributes_method = Module.new do
           def attributes
             super.merge(translated_attributes)
           end
         end
+        model.class_eval do
+          alias_method :untranslated_attributes, :attributes
+          include attributes_method
+        end
       end
 
       # Translated attributes defined on model.