[1.2.X] Fixed #11707 - limit_choices_to on a ForeignKey can render duplicate options in formfield

Thanks to Chris Wesseling for the report and patch.

Backport of [15607] from trunk.

git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@15610 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Luke Plant 2011-02-21 14:08:06 +00:00
parent fee8a49866
commit 4df0e5ff84
3 changed files with 24 additions and 2 deletions

View File

@ -904,7 +904,7 @@ class ForeignKey(RelatedField, Field):
db = kwargs.pop('using', None)
defaults = {
'form_class': forms.ModelChoiceField,
'queryset': self.rel.to._default_manager.using(db).complex_filter(self.rel.limit_choices_to),
'queryset': self.rel.to._default_manager.using(db).complex_filter(self.rel.limit_choices_to).distinct(),
'to_field_name': self.rel.field_name,
}
defaults.update(kwargs)

View File

@ -29,6 +29,11 @@ class Bar(models.Model):
b = models.CharField(max_length=10)
a = models.ForeignKey(Foo, default=get_foo)
class Baz(models.Model):
a = models.CharField(max_length=5)
#Only Foos related to Bars starting with 'a'
foo = models.ForeignKey(Foo, limit_choices_to=models.Q(bar__b__startswith='a'))
class Whiz(models.Model):
CHOICES = (
('Group 1', (

View File

@ -1,13 +1,14 @@
import datetime
import unittest
from decimal import Decimal
import re
import django.test
from django import forms
from django.db import models
from django.core.exceptions import ValidationError
from models import Foo, Bar, Whiz, BigD, BigS, Image, BigInt, Post, NullBooleanModel, BooleanModel
from models import Foo, Bar, Baz, Whiz, BigD, BigS, Image, BigInt, Post, NullBooleanModel, BooleanModel
# If PIL available, do these tests.
if Image:
@ -94,6 +95,10 @@ class DecimalFieldTests(django.test.TestCase):
# This should not crash. That counts as a win for our purposes.
Foo.objects.filter(d__gte=100000000000)
class BazForm(forms.ModelForm):
class Meta:
model = Baz
class ForeignKeyTests(django.test.TestCase):
def test_callable_default(self):
"""Test the use of a lazy callable for ForeignKey.default"""
@ -101,6 +106,18 @@ class ForeignKeyTests(django.test.TestCase):
b = Bar.objects.create(b="bcd")
self.assertEqual(b.a, a)
def test_distinct_choice_limit(self):
"""Doesn't make sense to offer the same ForeignKey multiple times in a form"""
a = Foo.objects.create(a='a', d=Decimal("-1"))
b = Foo.objects.create(a='b', d=Decimal("1"))
bar_a = Bar.objects.create(b='ah', a=a)
bar_b = Bar.objects.create(b='aha', a=a)
bar_b = Bar.objects.create(b='bla', a=b)
form = BazForm()
fk_field = str(form['foo'])
self.assertEqual(len(re.findall(r'value="2"', fk_field)), 0)
self.assertEqual(len(re.findall(r'value="1"', fk_field)), 1)
class DateTimeFieldTests(unittest.TestCase):
def test_datetimefield_to_python_usecs(self):
"""DateTimeField.to_python should support usecs"""