1
0
mirror of https://github.com/django/django.git synced 2025-10-25 14:46:09 +00:00

Fixed #26196 -- Made sure __in lookups use to_field as default.

Thanks Simon Charette for the test.
This commit is contained in:
Anssi Kääriäinen
2016-02-11 08:39:37 +02:00
committed by Tim Graham
parent 04e13c8913
commit 46ecfb9b3a
6 changed files with 25 additions and 4 deletions

View File

@@ -210,7 +210,7 @@ class BaseExpression(object):
])
return c
def _prepare(self):
def _prepare(self, field):
"""
Hook used by Field.get_prep_lookup() to do custom preparation.
"""

View File

@@ -740,7 +740,7 @@ class Field(RegisterLookupMixin):
Perform preliminary non-db specific lookup checks and conversions
"""
if hasattr(value, '_prepare'):
return value._prepare()
return value._prepare(self)
if lookup_type in {
'iexact', 'contains', 'icontains',

View File

@@ -1108,12 +1108,18 @@ class QuerySet(object):
for field, objects in other._known_related_objects.items():
self._known_related_objects.setdefault(field, {}).update(objects)
def _prepare(self):
def _prepare(self, field):
if self._fields is not None:
# values() queryset can only be used as nested queries
# if they are set up to select only a single field.
if len(self._fields or self.model._meta.concrete_fields) > 1:
raise TypeError('Cannot use multi-field values as a filter value.')
else:
# If the query is used as a subquery for a ForeignKey with non-pk
# target field, make sure to select the target field in the subquery.
foreign_fields = getattr(field, 'foreign_related_fields', ())
if len(foreign_fields) == 1 and not foreign_fields[0].primary_key:
return self.values(foreign_fields[0].name)
return self
def _as_sql(self, connection):

View File

@@ -235,7 +235,7 @@ class Query(object):
memo[id(self)] = result
return result
def _prepare(self):
def _prepare(self, field):
return self
def get_compiler(self, using=None, connection=None):