Add GeoDjango support

Author: timgraham

django_mongodb_backend_gis/__init__.py

@@ -0,0 +1,3 @@
+from .functions import register_functions
+
+register_functions()

django_mongodb_backend_gis/adapter.py

@@ -0,0 +1,12 @@
+import collections
+
+
+class Adapter(collections.UserDict):
+    def __init__(self, obj, geography=False):
+        """
+        Initialize on the spatial object.
+        """
+        self.data = {
+            "type": obj.__class__.__name__,
+            "coordinates": obj.coords,
+        }

django_mongodb_backend_gis/base.py

@@ -0,0 +1,11 @@
+from django_mongodb_backend.base import DatabaseWrapper as BaseDatabaseWrapper
+
+from .features import DatabaseFeatures
+from .operations import DatabaseOperations
+from .schema import DatabaseSchemaEditor
+
+
+class DatabaseWrapper(BaseDatabaseWrapper):
+    SchemaEditorClass = DatabaseSchemaEditor
+    features_class = DatabaseFeatures
+    ops_class = DatabaseOperations

django_mongodb_backend_gis/features.py

@@ -0,0 +1,59 @@
+from django.contrib.gis.db.backends.base.features import BaseSpatialFeatures
+from django.utils.functional import cached_property
+
+from django_mongodb_backend.features import DatabaseFeatures as MongoFeatures
+
+
+class DatabaseFeatures(BaseSpatialFeatures, MongoFeatures):
+    has_spatialrefsys_table = False
+    supports_transform = False
+
+    @cached_property
+    def django_test_expected_failures(self):
+        expected_failures = super().django_test_expected_failures
+        expected_failures.update(
+            {
+                # SRIDs aren't supported.
+                "gis_tests.geogapp.tests.GeographyTest.test05_geography_layermapping",
+                # 'WithinLookup' object has no attribute 'as_mql'
+                # "gis_tests.relatedapp.tests.RelatedGeoModelTest.test10_combine",
+                # GEOSException: Calling transform() with no SRID set is not supported
+                "gis_tests.relatedapp.tests.RelatedGeoModelTest.test06_f_expressions",
+                # To triage:
+                "gis_tests.geoapp.test_expressions.GeoExpressionsTests.test_geometry_value_annotation",
+                "gis_tests.geoapp.test_expressions.GeoExpressionsTests.test_multiple_annotation",
+                "gis_tests.geoapp.test_regress.GeoRegressionTests.test_empty_count",
+                "gis_tests.geoapp.test_serializers.GeoJSONSerializerTests.test_fields_option",
+                "gis_tests.geoapp.test_serializers.GeoJSONSerializerTests.test_geometry_field_option",
+                "gis_tests.geoapp.test_serializers.GeoJSONSerializerTests.test_id_field_option",
+                "gis_tests.geoapp.test_serializers.GeoJSONSerializerTests.test_serialization_base",
+                "gis_tests.geoapp.test_serializers.GeoJSONSerializerTests.test_srid_option",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_contains_contained_lookups",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_disjoint_lookup",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_equals_lookups",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_null_geometries",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_subquery_annotation",
+                "gis_tests.geoapp.tests.GeoModelTest.test_geometryfield",
+                "gis_tests.geoapp.tests.GeoModelTest.test_gis_query_as_string",
+                "gis_tests.geoapp.tests.GeoQuerySetTest.test_within_subquery",
+                "gis_tests.geoapp.test_expressions.GeoExpressionsTests.test_update_from_other_field",
+                "gis_tests.geoapp.tests.GeoLookupTest.test_gis_lookups_with_complex_expressions",
+                "gis_tests.geoapp.tests.GeoModelTest.test_proxy",
+            }
+        )
+        return expected_failures
+
+    @cached_property
+    def django_test_skips(self):
+        skips = super().django_test_skips
+        skips.update(
+            {
+                "inspectdb not supported.": {
+                    "gis_tests.inspectapp.tests.InspectDbTests",
+                },
+                "Raw SQL not supported": {
+                    "gis_tests.geoapp.tests.GeoModelTest.test_raw_sql_query",
+                },
+            },
+        )
+        return skips

django_mongodb_backend_gis/functions.py

@@ -0,0 +1,7 @@
+from django.contrib.gis.db.models.functions import Distance, Length, Perimeter
+
+
+def register_functions():
+    Distance.as_cockroachdb = Distance.as_postgresql
+    Length.as_cockroachdb = Length.as_postgresql
+    Perimeter.as_cockroachdb = Perimeter.as_postgresql

django_mongodb_backend_gis/operations.py

@@ -0,0 +1,85 @@
+from django.contrib.gis import geos
+from django.contrib.gis.db import models
+from django.contrib.gis.db.backends.base.operations import BaseSpatialOperations
+
+from django_mongodb_backend.operations import (
+    DatabaseOperations as MongoOperations,
+)
+
+from .adapter import Adapter
+
+
+class DatabaseOperations(BaseSpatialOperations, MongoOperations):
+    Adapter = Adapter
+
+    disallowed_aggregates = (
+        models.Collect,
+        models.Extent,
+        models.Extent3D,
+        models.MakeLine,
+        models.Union,
+    )
+
+    @property
+    def gis_operators(self):
+        return {}
+
+    unsupported_functions = {
+        "Area",
+        "AsGeoJSON",
+        "AsGML",
+        "AsKML",
+        "AsSVG",
+        "AsWKB",
+        "AsWKT",
+        "Azimuth",
+        "BoundingCircle",
+        "Centroid",
+        "ClosestPoint",
+        "Difference",
+        "Distance",
+        "Envelope",
+        "ForcePolygonCW",
+        "FromWKB",
+        "FromWKT",
+        "GeoHash",
+        "GeometryDistance",
+        "Intersection",
+        "IsEmpty",
+        "IsValid",
+        "Length",
+        "LineLocatePoint",
+        "MakeValid",
+        "MemSize",
+        "NumGeometries",
+        "NumPoints",
+        "Perimeter",
+        "PointOnSurface",
+        "Reverse",
+        "Scale",
+        "SnapToGrid",
+        "SymDifference",
+        "Transform",
+        "Translate",
+        "Union",
+    }
+
+    def geo_db_type(self, f):
+        return "object"
+
+    def get_geometry_converter(self, expression):
+        geom_class = expression.output_field.geom_class
+
+        def converter(value, expression, connection):  # noqa: ARG001
+            if value is None:
+                return None
+            if issubclass(geom_class, geos.GeometryCollection):
+                init_val = [
+                    geom_class._allowed(value["coordinates"][x][0])
+                    for x in range(len(value["coordinates"]))
+                ]
+            else:
+                init_val = value["coordinates"]
+            return geom_class(init_val)
+
+        return converter

django_mongodb_backend_gis/schema.py

@@ -0,0 +1,5 @@
+from django_mongodb_backend.schema import DatabaseSchemaEditor as BaseSchemaEditor
+
+
+class DatabaseSchemaEditor(BaseSchemaEditor):
+    pass