Merge branch 'master' of http://github.com/aleszoulek/mongoengine into dev

2010-11-02 13:43:33 +00:00
parent e868f37c60 a3830be4c9
commit 11950eabea
2 changed files with 46 additions and 5 deletions
--- a/mongoengine/queryset.py
+++ b/mongoengine/queryset.py
@@ -818,11 +818,7 @@ class QuerySet(object):
        """
        self._loaded_fields = []
        for field in fields:
-            if '.' in field:
-                raise InvalidQueryError('Subfields cannot be used as '
-                                        'arguments to QuerySet.only')
-            # Translate field name
-            field = QuerySet._lookup_field(self._document, field)[-1].db_field
+            field = ".".join(f.db_field for f in QuerySet._lookup_field(self._document, field.split('.')))
            self._loaded_fields.append(field)

        # _cls is needed for polymorphism