atomx.Atomx.search takes index parameter to only search specific models.

Author: dakra

CHANGES.txt

@@ -2,7 +2,7 @@
 ------------------
 
 - :meth:`atomx.Atomx.login` takes ``totp`` parameter for users that have 2-factor auth enabled.
-
+- :meth:`atomx.Atomx.search` takes ``index`` parameter to only search specific models.
 
 1.5
 ---

atomx/__init__.py

@@ -89,7 +89,7 @@ def logout(self):
         self.auth_token = None
         self.user = None
 
-    def search(self, query):
+    def search(self, query, index=None):
         """Search for ``query``.
 
         Returns a `dict` with all found results for:
@@ -114,10 +114,17 @@ def search(self, query):
             >>> campaign.reload()
 
         :param str query: keyword to search for.
+        :param list index: :class:`str` or :class:`list` of the indexes you want to get returned.
+            E.g. ``index=['campaigns', 'domains']``.
         :return: dict with list of :mod:`.models` as values
         """
+        params = {'q': query}
+        if index:
+            if isinstance(index, list):
+                index = ','.join(index)
+            params['index'] = index
         r = requests.get(self.api_endpoint + 'search',
-                         params={'q': query},
+                         params=params,
                          headers=self._auth_header)
         r_json = r.json()
         if not r.ok:
@@ -133,7 +140,7 @@ def search(self, query):
         for m in search_result.keys():
             model_name = get_model_name(m)
             if model_name:
-                search_result[m] = [getattr(models, model_name)(self, **v)
+                search_result[m] = [getattr(models, model_name)(session=self, **v)
                                     for v in search_result[m]]
         return search_result