Merge branch 'master' of github.com:MongoEngine/mongoengine

This commit is contained in:
Ross Lawley
2013-06-05 09:30:56 +00:00
2 changed files with 32 additions and 1 deletions

View File

@@ -71,7 +71,10 @@ class QuerySet(object):
# If inheritance is allowed, only return instances and instances of
# subclasses of the class being used
if document._meta.get('allow_inheritance') is True:
self._initial_query = {"_cls": {"$in": self._document._subclasses}}
if len(self._document._subclasses) == 1:
self._initial_query = {"_cls": self._document._subclasses[0]}
else:
self._initial_query = {"_cls": {"$in": self._document._subclasses}}
self._loaded_fields = QueryFieldList(always_include=['_cls'])
self._cursor_obj = None
self._limit = None