lib/traco/class_methods.rb in traco-1.0.0 vs lib/traco/class_methods.rb in traco-1.1.0

@@ -6,12 +6,14 @@
       column_names.
         grep(re) { $1.to_sym }.
         sort_by(&locale_sort_value)
     end
 
-    def locale_columns(attribute)
-      locales_for_attribute(attribute).map { |locale|
-        :"#{attribute}_#{locale}"
+    def locale_columns(*attributes)
+      attributes.inject([]) { |memo, attribute|
+        memo += locales_for_attribute(attribute).map { |locale|
+          :"#{attribute}_#{locale}"
+        }
       }
     end
 
     def human_attribute_name(attribute, options = {})
       default = super(attribute, options.merge(:default => ""))