Mercurial > public > sg101
diff forums/search_indexes.py @ 581:ee87ea74d46b
For Django 1.4, rearranged project structure for new manage.py.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Sat, 05 May 2012 17:10:48 -0500 |
parents | gpp/forums/search_indexes.py@387d46abcb95 |
children | ad53d929281a |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/forums/search_indexes.py Sat May 05 17:10:48 2012 -0500 @@ -0,0 +1,60 @@ +"""Haystack search index for the weblinks application.""" +from haystack.indexes import * +from haystack import site +from custom_search.indexes import CondQueuedSearchIndex + +from forums.models import Forum, Topic, Post +from forums.signals import topic_content_update, post_content_update + + +class TopicIndex(CondQueuedSearchIndex): + text = CharField(document=True, use_template=True) + author = CharField(model_attr='user') + pub_date = DateTimeField(model_attr='creation_date') + + def index_queryset(self): + return Topic.objects.filter(forum__in=Forum.objects.public_forum_ids()) + + def get_updated_field(self): + return 'update_date' + + def _setup_save(self, model): + topic_content_update.connect(self.enqueue_save) + + def _teardown_save(self, model): + topic_content_update.disconnect(self.enqueue_save) + + def enqueue_save(self, sender, **kwargs): + return self.enqueue('update', sender) + + def can_index(self, instance): + return instance.forum.id in Forum.objects.public_forum_ids() + + +class PostIndex(CondQueuedSearchIndex): + text = CharField(document=True, use_template=True) + author = CharField(model_attr='user') + pub_date = DateTimeField(model_attr='creation_date') + + def index_queryset(self): + return Post.objects.filter( + topic__forum__in=Forum.objects.public_forum_ids()) + + def get_updated_field(self): + return 'update_date' + + def _setup_save(self, model): + post_content_update.connect(self.enqueue_save) + + def _teardown_save(self, model): + post_content_update.disconnect(self.enqueue_save) + + def enqueue_save(self, sender, **kwargs): + return self.enqueue('update', sender) + + def can_index(self, instance): + return instance.topic.forum.id in Forum.objects.public_forum_ids() + + +site.register(Topic, TopicIndex) +site.register(Post, PostIndex)