Hopiu/xapian-haystack
1
0
Fork 0
mirror of https://github.com/Hopiu/xapian-haystack.git synced 2026-05-11 16:53:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added custom ExpandDecider. To be used in 'more_like_this' calls

Browse source
This commit is contained in:
David Sauve 2009-08-18 09:49:20 -04:00
parent 6d2f4a53a7
commit c7c6b21e20
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
xapian_backend.py
View file

@ -84,6 +84,19 @@ class XHValueRangeProcessor(xapian.ValueRangeProcessor):
return field_dict['column'], str(begin), str(end)
class XHExpandDecider(xapian.ExpandDecider):
def __call__(self, term):
"""
Return True if the term should be used for expanding the search
query, False otherwise.
Currently, we only want to ignore terms beginning with `DOCUMENT_CT_TERM_PREFIX`
"""
if term.startswith(DOCUMENT_CT_TERM_PREFIX):
return False
return True
class SearchBackend(BaseSearchBackend):
"""
`SearchBackend` defines the Xapian search backend for use with the Haystack
@ -236,7 +249,7 @@ class SearchBackend(BaseSearchBackend):
query, __unused__ = self._query(database, '*')
enquire = self._enquire(database, query)
for match in enquire.get_mset(0, DEFAULT_MAX_RESULTS):
database.delete_document(match.get_docid())
database.delete_document(match.docid)
else:
for model in models:
database.delete_document(
@ -402,7 +415,7 @@ class SearchBackend(BaseSearchBackend):
for match in enquire.get_mset(0, DEFAULT_MAX_RESULTS):
rset.add_document(match.docid)
query = xapian.Query(xapian.Query.OP_OR,
[expand.term for expand in enquire.get_eset(DEFAULT_MAX_RESULTS, rset)]
[expand.term for expand in enquire.get_eset(DEFAULT_MAX_RESULTS, rset, XHExpandDecider())]
)
query = xapian.Query(
xapian.Query.OP_AND_NOT, [query, self.get_identifier(model_instance)]
@ -415,7 +428,7 @@ class SearchBackend(BaseSearchBackend):
xapian.Query.OP_AND, query, additional_query
)
enquire.set_query(query)
results = []
matches = enquire.get_mset(start_offset, end_offset)