You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.2 KiB
46 lines
1.2 KiB
from haystack import indexes
|
|
from models import Company
|
|
"""
|
|
class CompanyIndex(indexes.SearchIndex, indexes.Indexable):
|
|
text = indexes.CharField(document=True, use_template=True)
|
|
where = indexes.MultiValueField()
|
|
theme = indexes.MultiValueField()
|
|
tag = indexes.MultiValueField()
|
|
country = indexes.CharField(model_attr='country', null=True)
|
|
city = indexes.CharField(model_attr='city', null=True)
|
|
|
|
def prepare_country(self, obj):
|
|
if obj.country:
|
|
return '%s'%obj.country.id
|
|
return ''
|
|
|
|
def prepare_city(self, obj):
|
|
if obj.city:
|
|
return '%s'%obj.city.id
|
|
return ''
|
|
|
|
def prepare_theme(self, obj):
|
|
|
|
return [str(th.id) for th in obj.theme.filter()]
|
|
|
|
def prepare_tag(self, obj):
|
|
return [str(tag.id) for tag in obj.tag.filter()]
|
|
|
|
def prepare_where(self, obj):
|
|
country = []
|
|
city = []
|
|
if obj.country:
|
|
country = [tr.name for tr in obj.country.translations.all()]
|
|
if obj.city:
|
|
city = [tr.name for tr in obj.city.translations.all()]
|
|
|
|
return country + city
|
|
|
|
def get_model(self):
|
|
return Company
|
|
|
|
def index_queryset(self, using=None):
|
|
|
|
return self.get_model().objects.filter()
|
|
|
|
""" |