Fixed #32511 -- Corrected handling prefetched nested reverse relationships.

When prefetching a set of child objects related to a set of parent
objects, we usually want to populate the relationship back from the
child to the parent to avoid a query when accessing that relationship
attribute. However, there's an edge case where the child queryset
itself specifies a prefetch back to the parent. In that case, we want
to use the prefetched relationship rather than populating the reverse
relationship from the parent.
This commit is contained in:
Jamie Matthews 2022-01-04 11:10:49 +00:00 committed by Mariusz Felisiak
parent 973fa56652
commit f5233dce30
3 changed files with 30 additions and 2 deletions

View File

@ -423,6 +423,7 @@ answer newbie questions, and generally made Django that much better:
James Timmins <jameshtimmins@gmail.com>
James Turk <dev@jamesturk.net>
James Wheare <django@sparemint.com>
Jamie Matthews <jamie@mtth.org>
Jannis Leidel <jannis@leidel.info>
Janos Guljas
Jan Pazdziora

View File

@ -646,6 +646,7 @@ def create_reverse_many_to_one_manager(superclass, rel):
# Since we just bypassed this class' get_queryset(), we must manage
# the reverse relation manually.
for rel_obj in queryset:
if not self.field.is_cached(rel_obj):
instance = instances_dict[rel_obj_attr(rel_obj)]
setattr(rel_obj, self.field.name, instance)
cache_name = self.field.remote_field.get_cache_name()

View File

@ -1614,3 +1614,29 @@ class ReadPrefetchedObjectsCacheTests(TestCase):
with self.assertNumQueries(4):
# AuthorWithAge -> Author -> FavoriteAuthors, Book
self.assertSequenceEqual(authors, [self.author1, self.author2])
class NestedPrefetchTests(TestCase):
@classmethod
def setUpTestData(cls):
house = House.objects.create(name='Big house', address='123 Main St')
cls.room = Room.objects.create(name='Kitchen', house=house)
def test_nested_prefetch_is_not_overwritten_by_related_object(self):
"""
The prefetched relationship is used rather than populating the reverse
relationship from the parent, when prefetching a set of child objects
related to a set of parent objects and the child queryset itself
specifies a prefetch back to the parent.
"""
queryset = House.objects.only('name').prefetch_related(
Prefetch('rooms', queryset=Room.objects.prefetch_related(
Prefetch('house', queryset=House.objects.only('address')),
)),
)
with self.assertNumQueries(3):
house = queryset.first()
self.assertIs(Room.house.is_cached(self.room), True)
with self.assertNumQueries(0):
house.rooms.first().house.address