Manage search key translations

1 files changed, 10 insertions, 3 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index 4549fdc69..2bc02288b 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -24,7 +24,8 @@ from django.contrib.gis.db import models
 from django.db.models import Max, Q
 from django.db.models.signals import post_save, post_delete, pre_delete
 from django.template.defaultfilters import slugify
-from django.utils.translation import ugettext_lazy as _, pgettext_lazy
+from django.utils.translation import ugettext_lazy as _, pgettext_lazy, \
+    activate, deactivate
 
 from archaeological_finds.models_finds import Find, FindBasket, TreatmentType
 from archaeological_operations.models import ClosedItem, Operation
@@ -98,7 +99,10 @@ class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
         ),
     }
     for v in ALT_NAMES.values():
-        EXTRA_REQUEST_KEYS[v[0]] = v[1]
+        for language_code, language_lbl in settings.LANGUAGES:
+            activate(language_code)
+            EXTRA_REQUEST_KEYS[unicode(v[0])] = v[1]
+            deactivate()
 
     BASE_SEARCH_VECTORS = [
         'treatment_types__label', 'treatment_state__label', 'label', 'goal',
@@ -562,7 +566,10 @@ class TreatmentFile(DashboardFormItem, ClosedItem, BaseHistorizedItem,
         ),
     }
     for v in ALT_NAMES.values():
-        EXTRA_REQUEST_KEYS[v[0]] = v[1]
+        for language_code, language_lbl in settings.LANGUAGES:
+            activate(language_code)
+            EXTRA_REQUEST_KEYS[unicode(v[0])] = v[1]
+            deactivate()
 
     # fields
     year = models.IntegerField(_(u"Year"), default=get_current_year)