settings: ne pas définir SITE_ID

[auf_framonde.git] / project / framonde / search_indexes.py

# encoding: utf-8
import datetime

from django.core.exceptions import ObjectDoesNotExist
from haystack import indexes

from .models import Communication, Contribution, Offre

REGIONS = {
    '0': 'Internationale',
    '1': 'Afrique Centrale et des Grands-Lacs',
    '2': 'Afrique de l\'ouest',
    '3': 'Amériques',
    '4': 'Asie-Pacifique',
    '5': 'Caraïbe',
    '6': 'Europe centrale et orientale',
    '7': 'Moyen-Orient',
    '8': 'Océan Indien',
    '9': 'Maghreb',
    '10': 'Europe de l\'Ouest'
}


class AufIndex(indexes.SearchIndex):
    text = indexes.NgramField(document=True, use_template=True)
    title = indexes.NgramField(model_attr='titre', stored=True)
    region = indexes.FacetField(stored=True)
    section = indexes.FacetField(stored=True)
#    annee_evenement = indexes.FacetField(stored=True)
    annee_limite = indexes.FacetField(stored=True)
    date_pub = indexes.DateField(model_attr='date_pub')

    def prepare_region(self, obj):
        return REGIONS[obj.region]

#    def prepare_annee_evenement(self, obj):
#        if obj.date_event is not None:
#            return str(obj.date_event.year)

    def prepare_annee_limite(self, obj):
        if obj.date_limite is not None:
            return str(obj.date_limite.year)


class CommunicationIndex(AufIndex, indexes.Indexable):

    def prepare_section(self, obj):
        return u"Communication"

    def get_model(self):
        return Communication

    def index_queryset(self, using=None):
        return Communication.objects.filter(status=3).order_by('-date_pub')


class ContributionIndex(AufIndex, indexes.Indexable):

    def prepare_section(self, obj):
        return u"Contribution"

    def get_model(self):
        return Contribution

    def index_queryset(self, using=None):
        return Contribution.objects.filter(status=3).order_by('-date_pub')


class OffreIndex(AufIndex, indexes.Indexable):

    def prepare_section(self, obj):
        return u"Offre"

    def get_model(self):
        return Offre

    def index_queryset(self, using=None):
        return Offre.objects.filter(status=3).order_by('-date_pub')