QuerySet.only() supports subfields + tests

2010-10-28 01:03:57 +02:00
parent 69989365c7
commit a3830be4c9
2 changed files with 46 additions and 5 deletions
--- a/mongoengine/queryset.py
+++ b/mongoengine/queryset.py
@@ -812,11 +812,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