Merge pull request #79 from smacker/master

Distinct support

Author: flaper87

django_mongodb_engine/contrib/__init__.py

@@ -131,6 +131,10 @@ def inline_map_reduce(self, *args, **kwargs):
     def _get_query(self):
         return _compiler_for_queryset(self).build_query()
 
+    def distinct(self, *args, **kwargs):
+        query = self._get_query()
+        return query.collection.distinct(*args, **kwargs)
+
 class MongoDBManager(models.Manager, RawQueryMixin):
     """
     Lets you use Map/Reduce and raw query/update with your models::
@@ -147,3 +151,6 @@ def inline_map_reduce(self, *args, **kwargs):
 
     def get_query_set(self):
         return MongoDBQuerySet(self.model, using=self._db)
+
+    def distinct(self, *args, **kwargs):
+        return self.get_query_set().distinct(*args, **kwargs)

tests/contrib/tests.py

@@ -194,3 +194,13 @@ def test_for_wrong_lookups(self):
         with self.assertRaises(DatabaseError):
             Post.objects.get(content_tokenized__iexact="django mongodb")
             Post.objects.get(content_tokenized__icontains="django mongodb")
+
+
+class DistinctTests(TestCase):
+    def test_distinct(self):
+        for i in xrange(10):
+            for j in xrange(i):
+                MapReduceModel.objects.create(n=i, m=i*2)
+
+        self.assertEqual(MapReduceModel.objects.distinct('m'),
+                        [2, 4, 6, 8, 10, 12, 14, 16, 18])