import datetime

from haystack import indexes
from snipts.models import Snipt


class SniptIndex(indexes.SearchIndex, indexes.Indexable):
    text = indexes.CharField(document=True, use_template=True)
    author = indexes.CharField(model_attr="user")
    pub_date = indexes.DateTimeField(model_attr="created")
    public = indexes.BooleanField(model_attr="public")
    typ = indexes.CharField(model_attr="lexer")

    def get_model(self):
        return Snipt

    def index_queryset(self, **kwargs):
        """Used when the entire index for model is updated."""
        return self.get_model().objects.filter(created__lte=datetime.datetime.now())