Added text search convenience methods.

a1gucis · Jun 13, 2013 · 49d7d1b · 49d7d1b
1 parent c557360
commit 49d7d1b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/mongokit/collection.py b/mongokit/collection.py
@@ -124,3 +124,9 @@ def find_random(self):
             num = random.randint(0, max-1)
             return self.find().skip(num).next()
 
+    def text(self, search, **kwargs):
+        """
+        Executes a full-text search. Additional parameters may be passed as keyword arguments.
+        """
+        return self.database.command("text", self.name, search=search, **kwargs)
+
diff --git a/mongokit/document.py b/mongokit/document.py
@@ -328,6 +328,16 @@ def find_random(self):
             num = random.randint(0, max-1)
             return self.find().skip(num).next()
 
+    def text(self, search, **kwargs):
+        """
+        Executes a full-text search. Additional parameters may be passed as keyword arguments.
+        """
+        rv = self.collection.database.command("text", self.collection.name, search=search, **kwargs)
+        if 'results' in rv:
+            for res in rv['results']:
+                res['obj'] = self._obj_class(res['obj'])
+        return rv
+
     def get_from_id(self, id):
         """
         return the document which has the id