fix events search index

[prawokultury.git] / events / search_indexes.py

diff --git a/events/search_indexes.py b/events/search_indexes.py
index 1321dcd..ec33c99 100644 (file)
--- a/events/search_indexes.py
+++ b/events/search_indexes.py
@@ -1,27 +1,30 @@
-from django.conf import settings
 import datetime
+from django.conf import settings
 from haystack import indexes
+from fnpdjango.utils.models.translation import add_translatable_index, localize_field
 from events.models import Event
-from migdal.helpers import add_translatable_index
 
 
 class EventIndex(indexes.RealTimeSearchIndex, indexes.Indexable):
+    text = indexes.CharField(null=True,
+        model_attr=localize_field('title', settings.LANGUAGE_CODE),
+        document=True)
+
     def get_model(self):
         return Event
 
     def index_queryset(self):
         """Used when the entire index for model is updated."""
         return self.get_model().objects.all()
-        
 
-add_translatable_index(EventIndex, {
-# Haystack needs a main field to be the same across all indexes
-# so we treat title of the event as this main content, named 'body'
-   'body': indexes.CharField(model_attr='title'), 
-   'organizer': indexes.CharField(), 
-   'place': indexes.CharField()
-   })
 
+add_translatable_index(EventIndex, {
+    'organizer': indexes.CharField(null=True),
+    'place': indexes.CharField(null=True)
+    })
 
-getattr(EventIndex, "body_%s" % settings.LANGUAGE_CODE).document = True
-
+add_translatable_index(EventIndex, {
+    'title': indexes.CharField(null=True),
+    }, 
+    (lang for lang in settings.LANGUAGES if lang[0] != settings.LANGUAGE_CODE)
+)