From 4620ce815b7989371820ff283a0a2e17bfad5025 Mon Sep 17 00:00:00 2001
From: Dave Hall <dave@etianen.com>
Date: Wed, 24 Aug 2011 13:36:26 +0100
Subject: [PATCH] Fixing filter on PostgresBackend and multi-word search on
 basic backend.

---
 src/watson/backends.py | 34 ++++++++++++++++++++--------------
 src/watson/tests.py    |  2 +-
 2 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/watson/backends.py b/src/watson/backends.py
index 53ffb42..20f70f0 100644
--- a/src/watson/backends.py
+++ b/src/watson/backends.py
@@ -1,6 +1,6 @@
 """Search backends used by django-watson."""
 
-import re
+import re, operator
 
 from django.conf import settings
 from django.core.exceptions import ImproperlyConfigured
@@ -11,14 +11,10 @@ from django.db.models import Q
 from watson.models import SearchEntry, has_int_pk
 
 
-def regex_from_search_text(search_text):
-    """Generates a regext from the given search text"""
-    words = search_text.split()
-    return u"|".join(
-        u"(\s{word}\s)|(^{word}\s)|(\s{word}$)|(^{word}$)".format(
-            word = re.escape(word),
-        )
-        for word in words
+def regex_from_word(word):
+    """Generates a regext from the given search word."""
+    return u"(\s{word}\s)|(^{word}\s)|(\s{word}$)|(^{word}$)".format(
+        word = re.escape(word),
     )
 
 
@@ -32,16 +28,24 @@ class SearchBackend(object):
         
     def do_search(self, queryset, search_text):
         """Filters the given queryset according the the search logic for this backend."""
-        regex = regex_from_search_text(search_text)
+        word_queries = []
+        for word in search_text.split():
+            regex = regex_from_word(word)
+            word_queries.append(Q(title__iregex=regex) | Q(content__iregex=regex) | Q(content__iregex=regex))
+        word_query = reduce(operator.and_, word_queries)
         return queryset.filter(
-            Q(title__iregex=regex) | Q(content__iregex=regex) | Q(content__iregex=regex),
+            word_query
         )
         
     def do_filter(self, queryset, search_text):
         """Filters the given queryset according the the search logic for this backend."""
-        regex = regex_from_search_text(search_text)
+        word_queries = []
+        for word in search_text.split():
+            regex = regex_from_word(word)
+            word_queries.append(Q(searchentry_set__title__iregex=regex) | Q(searchentry_set__content__iregex=regex) | Q(searchentry_set__content__iregex=regex))
+        word_query = reduce(operator.and_, word_queries)
         return queryset.filter(
-            Q(searchentry_set__title__iregex=regex) | Q(searchentry_set__content__iregex=regex) | Q(searchentry_set__content__iregex=regex),
+            word_query
         )
     
     def save_search_entry(self, search_entry, obj, adapter):
@@ -105,6 +109,7 @@ class PostgresSearchBackend(SearchBackend):
     def do_filter(self, queryset, search_text):
         """Performs the full text filter."""
         model = queryset.model
+        content_type = ContentType.objects.get_for_model(model)
         if has_int_pk(model):
             ref_name = "object_id_int"
         else:
@@ -122,8 +127,9 @@ class PostgresSearchBackend(SearchBackend):
                     table_name = connection.ops.quote_name(model._meta.db_table),
                     pk_name = connection.ops.quote_name(model._meta.pk.name),
                 ),
+                "watson_searchentry.content_type_id = %s"
             ),
-            params = (search_text,),
+            params = (search_text, content_type.id),
             order_by = ("-rank",),
         )
         
diff --git a/src/watson/tests.py b/src/watson/tests.py
index 263f0f0..6083ba8 100644
--- a/src/watson/tests.py
+++ b/src/watson/tests.py
@@ -222,7 +222,7 @@ class SearchTest(SearchTestBase):
     def testFilter(self):
         for model in (TestModel1, TestModel2):
             # Test can find all.
-            self.assertEqual(watson.filter(model, "title model1").count(), 2)
+            self.assertEqual(watson.filter(model, "title").count(), 2)
         # Test can find a specific one.
         obj = watson.filter(TestModel1, "12").get()
         self.assertTrue(isinstance(obj, TestModel1))