Added data kwarg and few others to the search for #109

documentcloud/__init__.py

@@ -189,15 +189,25 @@ def is_url(self, string):
         except:
             return False
 
-    def _get_search_page(self, query, page, per_page):
+    def _get_search_page(
+        self,
+        query,
+        page,
+        per_page=1000,
+        mentions=3,
+        data=False,
+    ):
         """
         Retrieve one page of search results from the DocumentCloud API.
         """
+        if mentions > 10:
+            raise ValueError("You cannot search for more than 10 mentions")
         params = {
             'q': query,
             'page': page,
             'per_page': per_page,
-            'mentions': 3,
+            'mentions': mentions,
+            'data': data,
         }
         data = self.fetch('search.json', params)
         return data.get("documents")

tests/test_all.py

@@ -198,6 +198,22 @@ def test_search(self):
         )
         self.assertEqual(len(one_page), 1)
 
+        data = one_page = self.public_client.documents.search(
+            self.test_search,
+            page=1,
+            per_page=1,
+            data=True,
+        )
+        data[0].data
+
+        with self.assertRaises(ValueError):
+            self.public_client.documents.search(
+                self.test_search,
+                page=1,
+                per_page=1,
+                mentions=11,
+            )
+
 
 class DocumentTest(BaseTest):
     """