Fixed #31190 -- Fixed prefetch_related() crash for GenericForeignKey with custom ContentType foreign key.

Regression in dffa3e1992.
This commit is contained in:
Eugene Hatsko 2020-01-21 10:12:25 +03:00 committed by Mariusz Felisiak
parent b753e0e750
commit 0b013564ef
3 changed files with 19 additions and 1 deletions

View File

@ -583,11 +583,12 @@ def create_generic_related_manager(superclass, rel):
# We (possibly) need to convert object IDs to the type of the
# instances' PK in order to match up instances:
object_id_converter = instances[0]._meta.pk.to_python
content_type_id_field_name = '%s_id' % self.content_type_field_name
return (
queryset.filter(query),
lambda relobj: (
object_id_converter(getattr(relobj, self.object_id_field_name)),
relobj.content_type_id
getattr(relobj, content_type_id_field_name),
),
lambda obj: (obj.pk, self.get_content_type(obj).pk),
False,

View File

@ -35,3 +35,7 @@ Bugfixes
* Fixed a regression in Django 2.2.7 that caused
:meth:`~django.db.models.Model.get_FOO_display` to work incorrectly when
overriding inherited choices (:ticket:`31124`).
* Fixed a regression in Django 3.0 that caused a crash of
``QuerySet.prefetch_related()`` for ``GenericForeignKey`` with a custom
``ContentType`` foreign key (:ticket:`31190`).

View File

@ -564,6 +564,19 @@ class GenericRelationsTests(TestCase):
for tag in tags:
self.assertSequenceEqual(tag.content_object.tags.all(), [tag])
def test_prefetch_related_custom_object_id(self):
tiger = Animal.objects.create(common_name='tiger')
cheetah = Animal.objects.create(common_name='cheetah')
Comparison.objects.create(
first_obj=cheetah, other_obj=tiger, comparative='faster',
)
Comparison.objects.create(
first_obj=tiger, other_obj=cheetah, comparative='cooler',
)
qs = Comparison.objects.prefetch_related('first_obj__comparisons')
for comparison in qs:
self.assertSequenceEqual(comparison.first_obj.comparisons.all(), [comparison])
class ProxyRelatedModelTest(TestCase):
def test_default_behavior(self):