From 328e062ae96b31b3c4095370ddf11e261dd367b2 Mon Sep 17 00:00:00 2001
From: Aleksandr Sorokoumov <aleksandr.sorokoumov@gmail.com>
Date: Mon, 24 Feb 2014 18:52:07 +0400
Subject: [PATCH 1/2] Distinct method bugfix

Creation of instances is executed now only for EmbeddedDocumentField
and GenericEmbeddedDocumentField in distinct method
---
 mongoengine/queryset/base.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mongoengine/queryset/base.py b/mongoengine/queryset/base.py
index c2ad027e..485872d7 100644
--- a/mongoengine/queryset/base.py
+++ b/mongoengine/queryset/base.py
@@ -630,7 +630,10 @@ class BaseQuerySet(object):
             # We may need to cast to the correct type eg. ListField(EmbeddedDocumentField)
             doc_field = getattr(self._document._fields.get(field), "field", None)
             instance = getattr(doc_field, "document_type", False)
-            if instance:
+            EmbeddedDocumentField = _import_class('EmbeddedDocumentField')
+            GenericEmbeddedDocumentField = _import_class('GenericEmbeddedDocumentField')
+            if instance and isinstance(doc_field, (EmbeddedDocumentField,
+                                                   GenericEmbeddedDocumentField)):
                 distinct = [instance(**doc) for doc in distinct]
             return distinct
 

From e60d56f060c03c50aeccc9159abc7591489ec0c6 Mon Sep 17 00:00:00 2001
From: Aleksandr Sorokoumov <aleksandr.sorokoumov@gmail.com>
Date: Mon, 24 Feb 2014 19:22:36 +0400
Subject: [PATCH 2/2] test implemented

---
 tests/queryset/queryset.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index 7ff2965d..8858e7d1 100644
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -2540,6 +2540,27 @@ class QuerySetTest(unittest.TestCase):
 
         self.assertEqual(authors, [mark_twain, john_tolkien])
 
+    def test_distinct_ListField_ReferenceField(self):
+        class Foo(Document):
+            bar_lst = ListField(ReferenceField('Bar'))
+
+        class Bar(Document):
+            text = StringField()
+
+        Bar.drop_collection()
+        Foo.drop_collection()
+
+        bar_1 = Bar(text="hi")
+        bar_1.save()
+
+        bar_2 = Bar(text="bye")
+        bar_2.save()
+
+        foo = Foo(bar=bar_1, bar_lst=[bar_1, bar_2])
+        foo.save()
+
+        self.assertEqual(Foo.objects.distinct("bar_lst"), [bar_1, bar_2])
+
     def test_custom_manager(self):
         """Ensure that custom QuerySetManager instances work as expected.
         """