Fixed #25972 -- Restored support for the isnull lookup with ForeignObject.
This commit is contained in:
parent
91cd4d8e8c
commit
8b6974a685
|
@ -25,7 +25,7 @@ from .related_descriptors import (
|
||||||
)
|
)
|
||||||
from .related_lookups import (
|
from .related_lookups import (
|
||||||
RelatedExact, RelatedGreaterThan, RelatedGreaterThanOrEqual, RelatedIn,
|
RelatedExact, RelatedGreaterThan, RelatedGreaterThanOrEqual, RelatedIn,
|
||||||
RelatedLessThan, RelatedLessThanOrEqual,
|
RelatedIsNull, RelatedLessThan, RelatedLessThanOrEqual,
|
||||||
)
|
)
|
||||||
from .reverse_related import (
|
from .reverse_related import (
|
||||||
ForeignObjectRel, ManyToManyRel, ManyToOneRel, OneToOneRel,
|
ForeignObjectRel, ManyToManyRel, ManyToOneRel, OneToOneRel,
|
||||||
|
@ -672,9 +672,10 @@ class ForeignObject(RelatedField):
|
||||||
return RelatedLessThan
|
return RelatedLessThan
|
||||||
elif lookup_name == 'lte':
|
elif lookup_name == 'lte':
|
||||||
return RelatedLessThanOrEqual
|
return RelatedLessThanOrEqual
|
||||||
elif lookup_name != 'isnull':
|
elif lookup_name == 'isnull':
|
||||||
|
return RelatedIsNull
|
||||||
|
else:
|
||||||
raise TypeError('Related Field got invalid lookup: %s' % lookup_name)
|
raise TypeError('Related Field got invalid lookup: %s' % lookup_name)
|
||||||
return super(ForeignObject, self).get_lookup(lookup_name)
|
|
||||||
|
|
||||||
def get_transform(self, *args, **kwargs):
|
def get_transform(self, *args, **kwargs):
|
||||||
raise NotImplementedError('Relational fields do not support transforms.')
|
raise NotImplementedError('Relational fields do not support transforms.')
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
from django.db.models.lookups import (
|
from django.db.models.lookups import (
|
||||||
Exact, GreaterThan, GreaterThanOrEqual, In, LessThan, LessThanOrEqual,
|
Exact, GreaterThan, GreaterThanOrEqual, In, IsNull, LessThan,
|
||||||
|
LessThanOrEqual,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -131,3 +132,7 @@ class RelatedGreaterThanOrEqual(RelatedLookupMixin, GreaterThanOrEqual):
|
||||||
|
|
||||||
class RelatedLessThanOrEqual(RelatedLookupMixin, LessThanOrEqual):
|
class RelatedLessThanOrEqual(RelatedLookupMixin, LessThanOrEqual):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class RelatedIsNull(RelatedLookupMixin, IsNull):
|
||||||
|
pass
|
||||||
|
|
|
@ -64,3 +64,6 @@ Bugfixes
|
||||||
|
|
||||||
* Restored the functionality of the admin's ``list_editable`` add and change
|
* Restored the functionality of the admin's ``list_editable`` add and change
|
||||||
buttons (:ticket:`25903`).
|
buttons (:ticket:`25903`).
|
||||||
|
|
||||||
|
* Fixed ``isnull`` query lookup for ``ForeignObject``
|
||||||
|
(:ticket:`25972`).
|
||||||
|
|
|
@ -56,19 +56,19 @@ class Membership(models.Model):
|
||||||
date_joined = models.DateTimeField(default=datetime.datetime.now)
|
date_joined = models.DateTimeField(default=datetime.datetime.now)
|
||||||
invite_reason = models.CharField(max_length=64, null=True)
|
invite_reason = models.CharField(max_length=64, null=True)
|
||||||
person_id = models.IntegerField()
|
person_id = models.IntegerField()
|
||||||
group_id = models.IntegerField()
|
group_id = models.IntegerField(blank=True, null=True)
|
||||||
|
|
||||||
# Relation Fields
|
# Relation Fields
|
||||||
person = models.ForeignObject(
|
person = models.ForeignObject(
|
||||||
Person,
|
Person,
|
||||||
from_fields=['membership_country', 'person_id'],
|
from_fields=['person_id', 'membership_country'],
|
||||||
to_fields=['person_country_id', 'id'],
|
to_fields=['id', 'person_country_id'],
|
||||||
on_delete=models.CASCADE,
|
on_delete=models.CASCADE,
|
||||||
)
|
)
|
||||||
group = models.ForeignObject(
|
group = models.ForeignObject(
|
||||||
Group,
|
Group,
|
||||||
from_fields=['membership_country', 'group_id'],
|
from_fields=['group_id', 'membership_country'],
|
||||||
to_fields=['group_country', 'id'],
|
to_fields=['id', 'group_country'],
|
||||||
on_delete=models.CASCADE,
|
on_delete=models.CASCADE,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -76,7 +76,8 @@ class Membership(models.Model):
|
||||||
ordering = ('date_joined', 'invite_reason')
|
ordering = ('date_joined', 'invite_reason')
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "%s is a member of %s" % (self.person.name, self.group.name)
|
group_name = self.group.name if self.group_id else 'NULL'
|
||||||
|
return "%s is a member of %s" % (self.person.name, group_name)
|
||||||
|
|
||||||
|
|
||||||
class Friendship(models.Model):
|
class Friendship(models.Model):
|
||||||
|
|
|
@ -395,6 +395,18 @@ class MultiColumnFKTests(TestCase):
|
||||||
objs = [Person(name="abcd_%s" % i, person_country=self.usa) for i in range(0, 5)]
|
objs = [Person(name="abcd_%s" % i, person_country=self.usa) for i in range(0, 5)]
|
||||||
Person.objects.bulk_create(objs, 10)
|
Person.objects.bulk_create(objs, 10)
|
||||||
|
|
||||||
|
def test_isnull_lookup(self):
|
||||||
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group_id=None)
|
||||||
|
Membership.objects.create(membership_country=self.usa, person=self.bob, group=self.cia)
|
||||||
|
self.assertQuerysetEqual(
|
||||||
|
Membership.objects.filter(group__isnull=True),
|
||||||
|
['<Membership: Bob is a member of NULL>']
|
||||||
|
)
|
||||||
|
self.assertQuerysetEqual(
|
||||||
|
Membership.objects.filter(group__isnull=False),
|
||||||
|
['<Membership: Bob is a member of CIA>']
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class TestModelCheckTests(SimpleTestCase):
|
class TestModelCheckTests(SimpleTestCase):
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue