# -*- coding: utf-8 -*-

#

# Copyright Institut de Recherche et d'Innovation © 2014

#

# contact@iri.centrepompidou.fr

#

# Ce code a été développé pour un premier usage dans JocondeLab, projet du 

# ministère de la culture et de la communication visant à expérimenter la

# recherche sémantique dans la base Joconde

# (http://jocondelab.iri-research.org/).

#

# Ce logiciel est régi par la licence CeCILL-C soumise au droit français et

# respectant les principes de diffusion des logiciels libres. Vous pouvez

# utiliser, modifier et/ou redistribuer ce programme sous les conditions

# de la licence CeCILL-C telle que diffusée par le CEA, le CNRS et l'INRIA 

# sur le site "http://www.cecill.info".

#

# En contrepartie de l'accessibilité au code source et des droits de copie,

# de modification et de redistribution accordés par cette licence, il n'est

# offert aux utilisateurs qu'une garantie limitée.  Pour les mêmes raisons,

# seule une responsabilité restreinte pèse sur l'auteur du programme,  le

# titulaire des droits patrimoniaux et les concédants successifs.

#

# A cet égard  l'attention de l'utilisateur est attirée sur les risques

# associés au chargement,  à l'utilisation,  à la modification et/ou au

# développement et à la reproduction du logiciel par l'utilisateur étant 

# donné sa spécificité de logiciel libre, qui peut le rendre complexe à 

# manipuler et qui le réserve donc à des développeurs et des professionnels

# avertis possédant  des  connaissances  informatiques approfondies.  Les

# utilisateurs sont donc invités à charger  et  tester  l'adéquation  du

# logiciel à leurs besoins dans des conditions permettant d'assurer la

# sécurité de leurs systèmes et ou de leurs données et, plus généralement, 

# à l'utiliser et l'exploiter dans les mêmes conditions de sécurité. 

#

# Le fait que vous puissiez accéder à cet en-tête signifie que vous avez 

# pris connaissance de la licence CeCILL-C, et que vous en avez accepté les

# termes.

#

from unidecode import unidecode

'''

@author: ymh

'''

from haystack import indexes

from jocondelab.models import DbpediaFields

class DbpediaFieldsIndex(indexes.SearchIndex, indexes.Indexable):

    text = indexes.CharField(document=True, use_template=True)

    label = indexes.CharField(model_attr='label', null=True)

    label_trans = indexes.CharField(model_attr='label', null=True, indexed=True, stored=False)

    label_ngram = indexes.NgramField(model_attr='label', null=True)

    label_edge = indexes.EdgeNgramField(model_attr='label', null=True)

    language_code = indexes.CharField(model_attr='language_code', null=True)

    dbpedia_uri = indexes.CharField(model_attr='dbpedia_uri', null=True, indexed=False, stored=True)

    def get_model(self):

        return DbpediaFields

    def index_queryset(self, using=None):

        """Used when the entire index for model is updated."""

        return self.get_model().objects.filter(term__nb_illustrated_notice__gt=0)

    def prepare_label_trans(self, obj):

        if obj.label is not None:

            return unidecode(obj.label.lower())

        else:

            return None