search_indexes.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536
  1. from haystack import indexes
  2. from misago.models import Post
  3. class PostIndex(indexes.SearchIndex, indexes.Indexable):
  4. text = indexes.CharField(document=True, use_template=True)
  5. forum = indexes.IntegerField(model_attr='forum_id')
  6. thread = indexes.IntegerField(model_attr='thread_id')
  7. thread_name = indexes.CharField(model_attr='thread')
  8. thread_starter = indexes.IntegerField(model_attr='thread_id')
  9. username = indexes.CharField(model_attr='user_name')
  10. date = indexes.DateTimeField(model_attr='date')
  11. def get_model(self):
  12. return Post
  13. def prepare_thread_name(self, obj):
  14. return obj.thread.name
  15. def prepare_thread_starter(self, obj):
  16. return obj.thread.start_poster_id
  17. def get_updated_field(self):
  18. return 'current_date'
  19. def should_update(self, instance, **kwargs):
  20. if (instance.deleted or instance.moderated
  21. or instance.thread.deletedor or instance.thread.moderated):
  22. self.remove_object(instance, **kwargs)
  23. return False
  24. return True
  25. def read_queryset(self, using=None):
  26. return Post.objects.all().select_related('forum', 'thread', 'user')
  27. def index_queryset(self, using=None):
  28. return self.get_model().objects.all().select_related('thread')