Add dwithin

Author: NoahStapp

django_mongodb_backend/gis/operations.py

@@ -95,6 +95,10 @@ def _gis_distance_operator(field, value, op=None, params=None):
     return cmd
 
 
+def _gis_dwithin_operator(field, value, op=None, params=None):  # noqa: ARG001
+    return {field: {"$geoWithin": {"$centerSphere": [value["coordinates"], params[0]]}}}
+
+
 class GISOperations(BaseSpatialOperations, BaseDatabaseOperations):
     Adapter = Adapter
 
@@ -117,6 +121,7 @@ def gis_operators(self):
             "distance_gte": SpatialOperator("distance_gte", _gis_distance_operator),
             "distance_lt": SpatialOperator("distance_lt", _gis_distance_operator),
             "distance_lte": SpatialOperator("distance_lte", _gis_distance_operator),
+            "dwithin": SpatialOperator("dwithin", _gis_dwithin_operator),
         }
 
     unsupported_functions = {

tests/gis_tests_/tests.py

@@ -51,3 +51,8 @@ def test_distance_lte_lookup(self):
         city = City.objects.create(point=Point(40.7589, -73.9851))
         qs = City.objects.filter(point__distance_lt=(Point(40.7670, -73.9820), 1000))
         self.assertIn(city, qs)
+
+    def test_dwithin_lookup(self):
+        city = City.objects.create(point=Point(40.7589, -73.9851))
+        qs = City.objects.filter(point__dwithin=(Point(40.7670, -73.9820), 1000))
+        self.assertIn(city, qs)