X-Git-Url: https://git.mdrn.pl/django-migdal.git/blobdiff_plain/734cd58d3f12c5397b42c1ae122c947500386014..f41fb09cefa2fca8d5c1e9314bcaa3591590a74c:/migdal/search_indexes.py

diff --git a/migdal/search_indexes.py b/migdal/search_indexes.py
index 27169f3..7c2ba7f 100644
--- a/migdal/search_indexes.py
+++ b/migdal/search_indexes.py
@@ -1,11 +1,18 @@
-import datetime
+# -*- coding: utf-8 -*-
+# This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 from django.conf import settings
 from haystack import indexes
-from fnpdjango.utils.models.translation import add_translatable_index
+from fnpdjango.utils.models.translation import add_translatable_index, localize_field
 from migdal.models import Entry
 
 
-class EntryIndex(indexes.RealTimeSearchIndex, indexes.Indexable):
+class EntryIndex(indexes.SearchIndex, indexes.Indexable):
+    text = indexes.CharField(
+        null=True,
+        model_attr=localize_field('body', settings.LANGUAGE_CODE),
+        document=True)
     date = indexes.DateTimeField(indexed=True, model_attr="date")
     author = indexes.CharField(model_attr="author")
 
@@ -14,14 +21,18 @@ class EntryIndex(indexes.RealTimeSearchIndex, indexes.Indexable):
 
     def index_queryset(self):
         """Used when the entire index for model is updated."""
-        return self.get_model().objects.all()  # filter(date__lte=datetime.datetime.now())
+        return self.get_model().objects.all()
 
 
 add_translatable_index(EntryIndex, {
     'title': indexes.CharField(null=True),
     'lead': indexes.CharField(null=True),
-    'body': indexes.CharField(null=True)
     })
 
-
-getattr(EntryIndex, "body_%s" % settings.LANGUAGE_CODE).document = True
+add_translatable_index(EntryIndex, {
+    'title': indexes.CharField(null=True),
+    'lead': indexes.CharField(null=True),
+    'body': indexes.CharField(null=True)
+    }, 
+    (lang for lang in settings.LANGUAGES if lang[0] != settings.LANGUAGE_CODE)
+)