Fixed geo index creation bug

fixes MongoEngine/mongoengine#36

docs/changelog.rst

@@ -4,6 +4,7 @@ Changelog
 
 Changes in 0.6.15
 =================
+- Fixed geo index creation through reference fields
 - Added support for args / kwargs when using @queryset_manager
 - Deref list custom id fix
 

mongoengine/base.py

@@ -1113,7 +1113,11 @@ Invalid data to create a `%s` instance.\n%s""".strip() % (cls._class_name, error
         inspected = inspected or []
         geo_indices = []
         inspected.append(cls)
+
+        from fields import EmbeddedDocumentField, GeoPointField
         for field in cls._fields.values():
+            if not isinstance(field, (EmbeddedDocumentField, GeoPointField)):
+                continue
             if hasattr(field, 'document_type'):
                 field_cls = field.document_type
                 if field_cls in inspected:

mongoengine/queryset.py

@@ -483,7 +483,6 @@ class QuerySet(object):
             self._collection.ensure_index(index_spec,
                 background=background, **index_opts)
 
-
     @classmethod
     def _build_index_spec(cls, doc_cls, spec):
         """Build a PyMongo index spec from a MongoEngine index spec.

tests/test_document.py

@@ -872,15 +872,26 @@ class DocumentTest(unittest.TestCase):
 
     def test_geo_indexes_recursion(self):
 
-        class User(Document):
+        class Location(Document):
-            channel = ReferenceField('Channel')
+            name = StringField()
             location = GeoPointField()
 
-        class Channel(Document):
+        class Parent(Document):
-            user = ReferenceField('User')
+            name = StringField()
-            location = GeoPointField()
+            location = ReferenceField(Location)
 
-        self.assertEquals(len(User._geo_indices()), 2)
+        Location.drop_collection()
+        Parent.drop_collection()
+
+        list(Parent.objects)
+
+        collection = Parent._get_collection()
+        info = collection.index_information()
+
+        self.assertFalse('location_2d' in info)
+
+        self.assertEquals(len(Parent._geo_indices()), 0)
+        self.assertEquals(len(Location._geo_indices()), 1)
 
     def test_covered_index(self):
         """Ensure that covered indexes can be used
@@ -3170,7 +3181,7 @@ name: Field is required ("name")"""
 
         class Person(BasePerson):
             name = StringField(required=True)
-        
+
 
         p = Person(age=15)
         self.assertRaises(ValidationError, p.validate)