AlgoliaIndex supports multiple geolocations

Author: m-vdb

README.md

@@ -134,6 +134,31 @@ class ContactIndex(AlgoliaIndex):
     geo_field = 'location'
 
 
+algoliasearch.register(Contact, ContactIndex)
+```
+Multiple geolocations are supported, for instance:
+
+```python
+class Location(models.Model):
+    lat = models.FloatField()
+    lng = models.FloatField()
+
+class Contact(models.Model):
+    name = models.CharField(max_lenght=20)
+    locations = models.ManyToManyField('Location')
+
+    def geolocations(self):
+        return [
+            {'lat': location.lat, 'lng': location.lng}
+            for location in self.locations.all()
+        ]
+
+
+class ContactIndex(AlgoliaIndex):
+    fields = 'name'
+    geo_field = 'geolocations'
+
+
 algoliasearch.register(Contact, ContactIndex)
 ```
 

src/models.py

@@ -136,6 +136,18 @@ def __get_objectID(self, instance):
         else:
             return instance.pk
 
+    @staticmethod
+    def _validate_geolocation(geolocation):
+        '''
+        Make sure we have the proper geolocation format.
+        '''
+        if set(geolocation) != {'lat', 'lng'}:
+            raise AlgoliaIndexError(
+                'Invalid geolocation format, requires "lat" and "lng" keys only got {}'.format(
+                    geolocation
+                )
+            )
+
     def _build_object(self, instance):
         '''Build the JSON object.'''
         tmp = {'objectID': self.__get_objectID(instance)}
@@ -155,8 +167,14 @@ def _build_object(self, instance):
         if self.geo_field:
             loc = self.geo_field(instance)
 
-            if loc:
+            if isinstance(loc, tuple):
                 tmp['_geoloc'] = {'lat': loc[0], 'lng': loc[1]}
+            elif isinstance(loc, dict):
+                self._validate_geolocation(loc)
+                tmp['_geoloc'] = loc
+            elif isinstance(loc, list):
+                [self._validate_geolocation(geo) for geo in loc]
+                tmp['_geoloc'] = loc
 
         if self.tags:
             attr = getattr(instance, self.tags)