Consistent use of languages settings, and Django 1.10 fixes.
[django-migdal.git] / migdal / search_indexes.py
1 # -*- coding: utf-8 -*-
2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from django.conf import settings
6 from haystack import indexes
7 from fnpdjango.utils.models.translation import add_translatable_index, localize_field
8 from migdal.models import Entry
9
10
11 class EntryIndex(indexes.SearchIndex, indexes.Indexable):
12     text = indexes.CharField(
13         null=True,
14         model_attr=localize_field('body', settings.LANGUAGE_CODE),
15         document=True)
16     date = indexes.DateTimeField(indexed=True, model_attr="date")
17     author = indexes.CharField(model_attr="author")
18
19     def prepare_date(self, obj):
20         date = u''
21         if 'date' in self.prepared_data:
22             date = self.prepared_data['date'].strftime('%Y-%m-%dT%H:%M:%SZ')
23         return date
24
25     def get_model(self):
26         return Entry
27
28
29 add_translatable_index(EntryIndex, {
30     'title': indexes.CharField(null=True),
31     'lead': indexes.CharField(null=True),
32     })
33
34 add_translatable_index(EntryIndex, {
35     'title': indexes.CharField(null=True),
36     'lead': indexes.CharField(null=True),
37     'body': indexes.CharField(null=True)
38     }, 
39     (lang for lang in settings.LANGUAGES if lang[0] != settings.LANGUAGE_CODE)
40 )