fix(reindex): support replica settings (#205)

Author: PLNech

algoliasearch_django/models.py

@@ -394,7 +394,20 @@ def reindex_all(self, batch_size=1000):
         """
         try:
             if self.settings:
-                self.__tmp_index.set_settings(self.settings)
+                replicas = self.settings.get('replicas', None)
+                slaves = self.settings.get('slaves', None)
+
+                should_keep_replicas = replicas is not None
+                should_keep_slaves = slaves is not None
+
+                if should_keep_replicas:
+                    self.settings['replicas'] = []
+                    logger.debug("REMOVE REPLICAS FROM SETTINGS")
+                if should_keep_slaves:
+                    self.settings['slaves'] = []
+                    logger.debug("REMOVE SLAVES FROM SETTINGS")
+
+                self.__tmp_index.wait_task(self.__tmp_index.set_settings(self.settings)['taskID'])
                 logger.debug('APPLY SETTINGS ON %s_tmp', self.index_name)
             self.__tmp_index.clear_index()
             logger.debug('CLEAR INDEX %s_tmp', self.index_name)
@@ -428,6 +441,16 @@ def reindex_all(self, batch_size=1000):
                                      self.__index.index_name)
             logger.info('MOVE INDEX %s_tmp TO %s', self.index_name,
                         self.index_name)
+
+            if self.settings:
+                if should_keep_replicas:
+                    self.settings['replicas'] = replicas
+                    logger.debug("RESTORE REPLICAS")
+                if should_keep_slaves:
+                    self.settings['slaves'] = slaves
+                    logger.debug("RESTORE SLAVES")
+                if should_keep_replicas or should_keep_slaves:
+                    self.__index.set_settings(self.settings)
             return counts
         except AlgoliaException as e:
             if DEBUG:

tests/test_index.py

@@ -59,6 +59,18 @@ def test_index_name_settings(self):
             except AttributeError:
                 self.assertRegexpMatches(index.index_name, regex)
 
+    def test_reindex_with_replicas(self):
+        class WebsiteIndex(AlgoliaIndex):
+            settings = {
+                'replicas': [
+                    'django_contact_name_asc',
+                    'django_contact_name_desc'
+                ]
+            }
+
+        index = WebsiteIndex(Website, self.client, settings.ALGOLIA)
+        index.reindex_all()
+
     def test_custom_objectID(self):
         class UserIndex(AlgoliaIndex):
             custom_objectID = 'username'