feat: iter_synonyms

Author: LeoErcolanelli

algoliasearch/index.py

@@ -70,37 +70,6 @@ def _load_next_page(self):
         self.pos = 0
         self.cursor = self.answer.get('cursor', None)
 
-class SynonymIterator:
-    """Iterator on the synonyms of an index"""
-
-    def __init__(self, index, hits_per_page=1000):
-        self.index = index
-        self.hits_per_page = hits_per_page
-        self.page = 0
-
-    def __iter__(self):
-        self._load_next_page()
-        return self
-
-    def __next__(self):
-        return self.next()
-
-    def next(self):
-        if self.pos >= len(self.response['hits']):
-            self._load_next_page()
-        if self.pos < len(self.response['hits']):
-            result = self.response['hits'][self.pos]
-            self.pos += 1
-            # Remove highlighting.
-            if '_highlightResult' in result: del result['_highlightResult']
-            return result
-        else:
-            raise StopIteration
-
-    def _load_next_page(self):
-        self.response = self.index.search_synonyms('', page=self.page, hits_per_page=self.hits_per_page)
-        self.page += 1
-        self.pos = 0
 
 class RuleIterator:
     """Iterator on the rules of an index"""
@@ -714,6 +683,21 @@ def search_synonyms(self, query, types=[], page=0, hits_per_page=100, request_op
 
         return self._req(True, '/synonyms/search', 'POST', request_options, data=data)
 
+    def iter_synonyms(self, hits_per_page=1000):
+        page = 0
+        response = self.search_synonyms('', page=page, hits_per_page=hits_per_page)
+
+        while response['hits']:
+            print(response['hits'])
+            for hit in response['hits']:
+                if '_highlightResult' in hit:
+                    del hit['_highlightResult']
+
+                yield hit
+
+            page += 1
+            response = self.search_synonyms('', page=page, hits_per_page=hits_per_page)
+
     @deprecated
     def waitTask(self, task_id, time_before_retry=100):
         return self.wait_task(task_id, time_before_retry)

tests/__init__.py

@@ -0,0 +1,5 @@
+import sys
+
+# Use local algoliasearch package.
+sys.path.append('..')
+

tests/conftest.py

@@ -1,7 +1,7 @@
 import pytest
 
-from helpers import Factory
-from helpers import create_client, create_index, IndexWithData
+from .helpers import Factory
+from .helpers import create_client, create_index, IndexWithData
 
 
 @pytest.fixture

tests/helpers.py

@@ -17,7 +17,7 @@ def check_credentials():
 
     for credential in credentials:
         if credential not in os.environ:
-            print('environement variable {} not defined')
+            print('environement variable {} not defined'.format(credential))
             assert False
 
 

tests/test_index.py

@@ -7,7 +7,7 @@
 from algoliasearch.client import MAX_API_KEY_LENGTH
 from algoliasearch.helpers import AlgoliaException
 
-from helpers import Factory, rule_stub
+from .helpers import Factory, rule_stub
 
 
 def test_add_object(index):
@@ -113,18 +113,25 @@ def test_synonyms(index):
     assert int(task['nbHits']) == 0
 
 
-# def tests_synonym_iterator(index):
-#     synonyms = [
-#         {'objectID': 'city', 'type': 'synonym',
-#          'synonyms': ['San Francisco', 'SF']},
-#         {'objectID': 'street', 'type': 'altCorrection1',
-#          'word': 'Street', 'corrections': ['St']}
-#     ]
-#     task = index.batch_synonyms(synonyms)
-#     index.wait_task(task['taskID'])
-#     it = SynonymIterator(index)
-#     for got, expected in zip(it, synonyms):
-#         self.assertEqual(got, expected)
+def test_iter_synonyms(index):
+    synonyms = [{
+        'objectID': 'city',
+        'type': 'synonym',
+        'synonyms': ['San Francisco', 'SF']
+    }, {
+        'objectID': 'street',
+        'type': 'altCorrection1',
+        'word': 'Street', 'corrections': ['St']
+    }]
+
+    task = index.batch_synonyms(synonyms)
+    index.wait_task(task['taskID'])
+
+    res = list(index.iter_synonyms(hits_per_page=1))
+    assert len(res) == 2
+
+    for synonym in synonyms:
+        assert synonym in res
 
 
 def test_facet_search(index):