From 96a964a18332b13e5ba4ecd11ee246cf80a8a8da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wilson=20J=C3=BAnior?= <wilsonpjunior@gmail.com>
Date: Thu, 9 May 2013 13:18:58 -0300
Subject: [PATCH] added .disable_inheritance method for the simple fetch
 exclusives classes

---
 mongoengine/queryset/queryset.py |  9 +++++++++
 tests/queryset/queryset.py       | 20 ++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/mongoengine/queryset/queryset.py b/mongoengine/queryset/queryset.py
index 191afdd2..407bf2fd 100644
--- a/mongoengine/queryset/queryset.py
+++ b/mongoengine/queryset/queryset.py
@@ -520,6 +520,15 @@ class QuerySet(object):
         queryset._none = True
         return queryset
 
+    def disable_inheritance(self):
+        """
+        Disable inheritance query, fetch only objects for the query class
+        """
+        if self._document._meta.get('allow_inheritance') is True:
+            self._initial_query = {"_cls": self._document._class_name}
+
+        return self
+
     def clone(self):
         """Creates a copy of the current
           :class:`~mongoengine.queryset.QuerySet`
diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index b9db297b..49ed36c5 100644
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -3272,5 +3272,25 @@ class QuerySetTest(unittest.TestCase):
         self.assertEqual(outer_count, 7)  # outer loop should be executed seven times total
         self.assertEqual(inner_total_count, 7 * 7)  # inner loop should be executed fourtynine times total
 
+    def test_disable_inheritance_queryset(self):
+        class A(Document):
+            x = IntField()
+            y = IntField()
+
+            meta = {'allow_inheritance': True}
+
+        class B(A):
+            z = IntField()
+
+        A.drop_collection()
+
+        A(x=10, y=20).save()
+        A(x=15, y=30).save()
+        B(x=20, y=40).save()
+        B(x=30, y=50).save()
+
+        for obj in A.objects.disable_inheritance():
+            self.assertEqual(obj.__class__, A)
+
 if __name__ == '__main__':
     unittest.main()