Merge pull request #2670 from idoshr/fix_no_dereference_distinct

Fix no dereference distinct

Author: bagerard

mongoengine/queryset/base.py

@@ -953,7 +953,8 @@ def distinct(self, field):
             field = self._fields_to_dbfields([field]).pop()
         except LookUpError:
             pass
-
+        if not self._auto_dereference:
+            return queryset._cursor.distinct(field)
         distinct = self._dereference(
             queryset._cursor.distinct(field), 1, name=field, instance=self._document
         )

tests/queryset/test_queryset.py

@@ -3505,6 +3505,7 @@ class Bar(Document):
         foo.save()
 
         assert Foo.objects.distinct("bar") == [bar]
+        assert Foo.objects.no_dereference().distinct("bar") == [bar.pk]
 
     def test_text_indexes(self):
         class News(Document):
@@ -3711,6 +3712,7 @@ class Foo(Document):
         foo.save()
 
         assert Foo.objects.distinct("bar_lst") == [bar_1, bar_2]
+        assert Foo.objects.no_dereference().distinct("bar_lst") == [bar_1.pk, bar_2.pk]
 
     def test_custom_manager(self):
         """Ensure that custom QuerySetManager instances work as expected."""