lib/traco/class_methods.rb in traco-0.3.1 vs lib/traco/class_methods.rb in traco-0.4.2

@@ -1,13 +1,21 @@
 module Traco
   module ClassMethods
-    def locales_for_column(column)
-      column_names.grep(/\A#{column}_([a-z]{2})\z/) {
-        $1.to_sym
-      }.sort_by { |locale| locale_sort_value(locale) }
+    def locales_for_attribute(attribute)
+      re = /\A#{attribute}_([a-z]{2})\z/
+
+      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}"
+      }
+    end
+
     def human_attribute_name(attribute, options = {})
       default = super(attribute, options.merge(:default => ""))
       if default.blank? && attribute.to_s.match(/\A(\w+)_([a-z]{2})\z/)
         column, locale = $1, $2.to_sym
         if translates?(column)
@@ -17,21 +25,23 @@
       super
     end
 
     private
 
-    def locale_sort_value(locale)
-      if locale == I18n.default_locale
-        # Sort the default locale first.
-        "0"
-      else
-        # Sort the rest alphabetically.
-        locale.to_s
-      end
+    def locale_sort_value
+      lambda { |locale|
+        if locale == I18n.default_locale
+          # Sort the default locale first.
+          "0"
+        else
+          # Sort the rest alphabetically.
+          locale.to_s
+        end
+      }
     end
 
-    def translates?(column)
-      translatable_columns.include?(column.to_sym)
+    def translates?(attribute)
+      translatable_attributes.include?(attribute.to_sym)
     end
 
     def locale_name(locale)
       I18n.t(locale, :scope => :"i18n.languages", :default => locale.to_s.upcase)
     end