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)