From 9a27c76021f934201cccf12215514a3091325ec8 Mon Sep 17 00:00:00 2001 From: sobolevn Date: Fri, 3 May 2024 11:40:50 +0300 Subject: [PATCH] Fixed #35426 -- Updated querysets to be a required argument of GenericPrefetch. --- django/contrib/contenttypes/prefetch.py | 2 +- docs/ref/contrib/contenttypes.txt | 2 +- docs/releases/5.0.5.txt | 3 +++ tests/contenttypes_tests/test_models.py | 8 ++++++++ 4 files changed, 13 insertions(+), 2 deletions(-) diff --git a/django/contrib/contenttypes/prefetch.py b/django/contrib/contenttypes/prefetch.py index b02ed3bae5..97ac60295b 100644 --- a/django/contrib/contenttypes/prefetch.py +++ b/django/contrib/contenttypes/prefetch.py @@ -3,7 +3,7 @@ from django.db.models.query import ModelIterable, RawQuerySet class GenericPrefetch(Prefetch): - def __init__(self, lookup, querysets=None, to_attr=None): + def __init__(self, lookup, querysets, to_attr=None): for queryset in querysets: if queryset is not None and ( isinstance(queryset, RawQuerySet) diff --git a/docs/ref/contrib/contenttypes.txt b/docs/ref/contrib/contenttypes.txt index fa44679659..5aad2a4ec7 100644 --- a/docs/ref/contrib/contenttypes.txt +++ b/docs/ref/contrib/contenttypes.txt @@ -603,7 +603,7 @@ information. .. versionadded:: 5.0 -.. class:: GenericPrefetch(lookup, querysets=None, to_attr=None) +.. class:: GenericPrefetch(lookup, querysets, to_attr=None) This lookup is similar to ``Prefetch()`` and it should only be used on ``GenericForeignKey``. The ``querysets`` argument accepts a list of querysets, diff --git a/docs/releases/5.0.5.txt b/docs/releases/5.0.5.txt index bb1d732167..7e7963e707 100644 --- a/docs/releases/5.0.5.txt +++ b/docs/releases/5.0.5.txt @@ -30,3 +30,6 @@ Bugfixes * Fixed a bug in Django 5.0 that caused a migration crash when altering a ``GeneratedField`` referencing a renamed field (:ticket:`35422`). + +* Fixed a bug in Django 5.0 where the ``querysets`` argument of + ``GenericPrefetch`` was not required (:ticket:`35426`). diff --git a/tests/contenttypes_tests/test_models.py b/tests/contenttypes_tests/test_models.py index 88f715ceff..799f1cc58c 100644 --- a/tests/contenttypes_tests/test_models.py +++ b/tests/contenttypes_tests/test_models.py @@ -331,6 +331,14 @@ class ContentTypesMultidbTests(TestCase): class GenericPrefetchTests(TestCase): + def test_querysets_required(self): + msg = ( + "GenericPrefetch.__init__() missing 1 required " + "positional argument: 'querysets'" + ) + with self.assertRaisesMessage(TypeError, msg): + GenericPrefetch("question") + def test_values_queryset(self): msg = "Prefetch querysets cannot use raw(), values(), and values_list()." with self.assertRaisesMessage(ValueError, msg):