Mercurial > public > sg101

comparison gpp/forums/search_indexes.py @ 467:b910cc1460c8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add the ability to conditionally add model instances to the search index on update. This is not perfect, as some instances should be deleted from the index if they are updated such that they should not be in the index anymore. Will think about and address that later.
author Brian Neal <bgneal@gmail.com>
date Sun, 24 Jul 2011 18:12:20 +0000
parents b1f939b1fb01
children 3b30286adba5
comparison
equal deleted inserted replaced
466:c78c6e007e61 467:b910cc1460c8
1 """Haystack search index for the weblinks application.""" 1 """Haystack search index for the weblinks application."""
2 from haystack.indexes import * 2 from haystack.indexes import *
3 from haystack import site 3 from haystack import site
4 from queued_search.indexes import QueuedSearchIndex 4 from custom_search import CondQueuedSearchIndex
5 5
6 from forums.models import Forum, Topic, Post 6 from forums.models import Forum, Topic, Post
7 7
8 8
9 class TopicIndex(QueuedSearchIndex): 9 class TopicIndex(CondQueuedSearchIndex):
10 text = CharField(document=True, use_template=True) 10 text = CharField(document=True, use_template=True)
11 author = CharField(model_attr='user') 11 author = CharField(model_attr='user')
12 pub_date = DateTimeField(model_attr='creation_date') 12 pub_date = DateTimeField(model_attr='creation_date')
13 13
14 def get_queryset(self): 14 def get_queryset(self):
15 return Topic.objects.filter(forum__in=Forum.objects.public_forum_ids()) 15 return Topic.objects.filter(forum__in=Forum.objects.public_forum_ids())
16 16
17 def get_updated_field(self): 17 def get_updated_field(self):
18 return 'update_date' 18 return 'update_date'
19 19
20 def can_index(self, instance):
21 return instance.forum.id in Forum.objects.public_forum_ids()
20 22
21 class PostIndex(QueuedSearchIndex): 23
24 class PostIndex(CondQueuedSearchIndex):
22 text = CharField(document=True, use_template=True) 25 text = CharField(document=True, use_template=True)
23 author = CharField(model_attr='user') 26 author = CharField(model_attr='user')
24 pub_date = DateTimeField(model_attr='creation_date') 27 pub_date = DateTimeField(model_attr='creation_date')
25 28
26 def get_queryset(self): 29 def get_queryset(self):
28 topic__forum__in=Forum.objects.public_forum_ids()) 31 topic__forum__in=Forum.objects.public_forum_ids())
29 32
30 def get_updated_field(self): 33 def get_updated_field(self):
31 return 'update_date' 34 return 'update_date'
32 35
36 def can_index(self, instance):
37 return instance.topic.forum.id in Forum.objects.public_forum_ids()
38
33 39
34 site.register(Topic, TopicIndex) 40 site.register(Topic, TopicIndex)
35 site.register(Post, PostIndex) 41 site.register(Post, PostIndex)