search_indexes.py 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  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()
  8. start_post = indexes.IntegerField()
  9. thread_starter = indexes.IntegerField(default=0)
  10. username = indexes.CharField(model_attr='user_name')
  11. date = indexes.DateTimeField(model_attr='date')
  12. def get_model(self):
  13. return Post
  14. def prepare_thread_name(self, obj):
  15. return obj.thread.name
  16. def prepare_start_post(self, obj):
  17. return 1 if obj.thread.start_post_id == obj.pk else 0
  18. def prepare_thread_starter(self, obj):
  19. return obj.thread.start_poster_id or 0
  20. def get_updated_field(self):
  21. return 'current_date'
  22. def should_update(self, instance, **kwargs):
  23. if (instance.deleted or instance.moderated
  24. or instance.thread.deleted or instance.thread.moderated):
  25. self.remove_object(instance, **kwargs)
  26. return False
  27. return True
  28. def read_queryset(self, using=None):
  29. return Post.objects.all().select_related('forum', 'thread', 'user')
  30. def index_queryset(self, using=None):
  31. return self.get_model().objects.all().select_related('thread')