from haystack import indexes
from models import Webinar


class WebinarIndex(indexes.SearchIndex, indexes.Indexable):
    text = indexes.CharField(document=True, use_template=True)
    where = indexes.MultiValueField()
    def prepare_where(self, obj):
        country = [tr.name for tr in obj.country.translations.all()]
        city = [tr.name for tr in obj.city.translations.all()]

        return country + city

    def get_model(self):
        return Webinar

    def index_queryset(self, using=None):

        return self.get_model().objects.filter()