create spatial indexes

Author: timgraham

django_mongodb_backend_gis/adapter.py

@@ -17,5 +17,6 @@ def __init__(self, obj, geography=False):
     def get_data(self, obj):
         return {
             "type": obj.__class__.__name__,
-            "coordinates": obj.coords[0] if obj.__class__.__name__ == "Polygon" else obj.coords,
+            "coordinates": obj.coords,
+            # obj.coords[0] if obj.__class__.__name__ == "Polygon" else obj.coords,
         }

django_mongodb_backend_gis/features.py

@@ -29,6 +29,12 @@ def django_test_expected_failures(self):
                 "gis_tests.geoapp.tests.GeoModelTest.test_gis_query_as_string",
                 # No lookups are supported (yet?)
                 "gis_tests.geoapp.tests.GeoLookupTest.test_gis_lookups_with_complex_expressions",
+                # LinearRing requires at least 4 points, got 1.
+                "gis_tests.geoapp.tests.GeoLookupTest.test_null_geometries",
+                "gis_tests.geoapp.tests.GeoModelTest.test_geometryfield",
+                # Trying to remove spatial index fails:
+                # "index not found with name [gis_neighborhood_geom_id]"
+                "gis_tests.gis_migrations.test_operations.OperationTests.test_alter_field_remove_spatial_index",
             }
         )
         return expected_failures
@@ -44,6 +50,21 @@ def django_test_skips(self):
                 "Raw SQL not supported": {
                     "gis_tests.geoapp.tests.GeoModelTest.test_raw_sql_query",
                 },
+                "MongoDB doesn't support the SRID(s) used in this test.": {
+                    # Error messages:
+                    # - Can't extract geo keys
+                    # - Longitude/latitude is out of bounds
+                    "gis_tests.geoapp.test_expressions.GeoExpressionsTests.test_update_from_other_field",
+                    "gis_tests.layermap.tests.LayerMapTest.test_encoded_name",
+                    # SouthTexasCity fixture objects use SRID 2278 which is ignored
+                    # by the patched version of loaddata in the Django fork.
+                    "gis_tests.distapp.tests.DistanceTest.test_init",
+                },
+                "ImproperlyConfigured isn't raised when using RasterField": {
+                    # Normally RasterField.db_type() raises an error, but MongoDB
+                    # migrations don't need to call it, so the check doesn't happen.
+                    "gis_tests.gis_migrations.test_operations.NoRasterSupportTests",
+                },
             },
         )
         return skips

django_mongodb_backend_gis/schema.py

@@ -6,16 +6,16 @@
 
 
 class DatabaseSchemaEditor(BaseSchemaEditor):
-    #    def _field_should_be_indexed(self, model, field):
-    #        if getattr(field, "spatial_index", False):
-    #            return True
-    #        return super()._field_should_be_indexed(model, field)
+    def _field_should_be_indexed(self, model, field):
+        if getattr(field, "spatial_index", False):
+            return True
+        return super()._field_should_be_indexed(model, field)
 
-    #    def _add_field_index(self, model, field, *, column_prefix=""):
-    #        if hasattr(field, "geodetic"):
-    #            self._add_spatial_index(model, field)
-    #        else:
-    #            super()._add_field_index(model, field, column_prefix=column_prefix)
+    def _add_field_index(self, model, field, *, column_prefix=""):
+        if hasattr(field, "geodetic"):
+            self._add_spatial_index(model, field)
+        else:
+            super()._add_field_index(model, field, column_prefix=column_prefix)
 
     def _alter_field(
         self,