Fixed #35690 -- Errored nicely when using in_bulk() with a values() or values_list() queryset.

This commit is contained in:
John Parton 2024-08-18 21:52:58 -05:00 committed by Sarah Boyce
parent fd1dd76778
commit e4a2e22ddb
2 changed files with 9 additions and 0 deletions

View File

@ -1101,6 +1101,8 @@ class QuerySet(AltersData):
"""
if self.query.is_sliced:
raise TypeError("Cannot use 'limit' or 'offset' with in_bulk().")
if not issubclass(self._iterable_class, ModelIterable):
raise TypeError("in_bulk() cannot be used with values() or values_list().")
opts = self.model._meta
unique_fields = [
constraint.fields[0]

View File

@ -327,6 +327,13 @@ class LookupTests(TestCase):
with self.assertRaisesMessage(TypeError, msg):
Article.objects.all()[0:5].in_bulk([self.a1.id, self.a2.id])
def test_in_bulk_not_model_iterable(self):
msg = "in_bulk() cannot be used with values() or values_list()."
with self.assertRaisesMessage(TypeError, msg):
Author.objects.values().in_bulk()
with self.assertRaisesMessage(TypeError, msg):
Author.objects.values_list().in_bulk()
def test_values(self):
# values() returns a list of dictionaries instead of object instances --
# and you can specify which fields you want to retrieve.